lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: create warehouse

Browse Source
This commit is contained in:
otxtan@gmail.com 2024-05-29 23:26:36 +07:00
parent 54a1856f15
commit 05a928b2bb
1 changed files with 45 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
src/documents/views.py
View File

@ -1810,51 +1810,50 @@ class WarehouseViewSet(ModelViewSet, PermissionsAwareDocumentCountMixin):
ordering_fields = ("name", "type", "parent_warehouse", "document_count") ordering_fields = ("name", "type", "parent_warehouse", "document_count")
def create(self, request, *args, **kwargs): def create(self, request, *args, **kwargs):
try: # try:
print(request.data)
serializer = WarehouseSerializer(data=request.data) serializer = WarehouseSerializer(data=request.data)
name = "" name = None
type = Warehouse.WAREHOUSE type = None
parent_warehouse = None parent_warehouse = None
if serializer.is_valid(): if serializer.is_valid(raise_exception=True):
name = serializer.validated_data.get("name", "") name = serializer.validated_data.get("name", "")
type = serializer.validated_data.get("type", Warehouse.WAREHOUSE) type = serializer.validated_data.get("type", Warehouse.WAREHOUSE)
parent_warehouse = serializer.validated_data.get('parent_warehouse',None) parent_warehouse = serializer.validated_data.get('parent_warehouse',None)
check_warehouse = Warehouse.objects.filter( # check_warehouse = Warehouse.objects.filter(
name = name, # name = name,
type = type, # type = type,
parent_warehouse=parent_warehouse # parent_warehouse=parent_warehouse
) # )
if check_warehouse: # if check_warehouse:
return Response({'status':400, # return Response({'status':400,
'message':'created fail'},status=status.HTTP_400_BAD_REQUEST) # 'message':'created fail'},status=status.HTTP_400_BAD_REQUEST)
if type == Warehouse.SHELF and parent_warehouse == None: # if type == Warehouse.SHELF and parent_warehouse == None:
return Response({'status': 400, # return Response({'status': 400,
'message': 'parent_warehouse is required for Shelf type'}, status=status.HTTP_400_BAD_REQUEST) # 'message': 'parent_warehouse is required for Shelf type'}, status=status.HTTP_400_BAD_REQUEST)
elif type == Warehouse.BOXCASE and parent_warehouse == None: # elif type == Warehouse.BOXCASE and parent_warehouse == None:
return Response({'status': 400, # return Response({'status': 400,
'message': 'parent_warehouse is required for Boxcase type'}, status=status.HTTP_400_BAD_REQUEST) # 'message': 'parent_warehouse is required for Boxcase type'}, status=status.HTTP_400_BAD_REQUEST)
if serializer.is_valid(raise_exception=True): # if serializer.is_valid(raise_exception=True):
parent_warehouse = Warehouse.objects.filter(id=parent_warehouse.id if parent_warehouse else 0).first() parent_warehouse = Warehouse.objects.filter(id=parent_warehouse.id if parent_warehouse else 0).first()
if serializer.validated_data.get("type", "") == "" and not parent_warehouse: if serializer.validated_data.get("type") == Warehouse.WAREHOUSE and not parent_warehouse:
serializer.save(type = Warehouse.WAREHOUSE) serializer.save()
elif serializer.validated_data.get("type", "") == Warehouse.SHELF and getattr(parent_warehouse, 'type', "") == Warehouse.WAREHOUSE : elif serializer.validated_data.get("type", "") == Warehouse.SHELF and getattr(parent_warehouse, 'type', "") == Warehouse.WAREHOUSE :
serializer.save(type = Warehouse.SHELF, parent_warehouse = parent_warehouse) serializer.save(type = Warehouse.SHELF, parent_warehouse = parent_warehouse)
elif serializer.validated_data.get("type", "") == Warehouse.BOXCASE and getattr(parent_warehouse, 'type', "") == Warehouse.SHELF : elif serializer.validated_data.get("type", "") == Warehouse.BOXCASE and getattr(parent_warehouse, 'type', "") == Warehouse.SHELF :
serializer.save(type = Warehouse.BOXCASE, parent_warehouse = parent_warehouse) serializer.save(type = Warehouse.BOXCASE, parent_warehouse = parent_warehouse)
else: else:
return Response({'status':400, return Response({'status':400,
'message':'created fail'},status=status.HTTP_400_BAD_REQUEST) 'message':'misplaced'},status=status.HTTP_400_BAD_REQUEST)
return Response({'status':201, return Response({'status':201,
'message':'created successfully', 'message':'created successfully',
'data':serializer.data},status=status.HTTP_201_CREATED) 'data':serializer.data},status=status.HTTP_201_CREATED)
except Exception as e: # except Exception as e:
return Response({'status':400, # return Response({'status':400,
'message':e},status=status.HTTP_400_BAD_REQUEST) # 'message':e},status=status.HTTP_400_BAD_REQUEST)