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

update bulkedit vs input parent_warehouse not null

Browse Source
This commit is contained in:
hungdztrau123 2024-05-29 09:27:47 +07:00
parent 816f16667a
commit 41cad86d15
2 changed files with 75 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/documents/serialisers.py
View File

@ -1767,17 +1767,10 @@ class WarehouseSerializer(MatchingModelSerializer, OwnedObjectSerializer):
def to_representation(self, instance):
data = super().to_representation(instance)
document_count = self.get_document_count(instance)
data['document_count'] = document_count
if instance.parent_warehouse:
parent_serializer = self.__class__(instance.parent_warehouse)
data['parent_warehouse'] = parent_serializer.data
data['parent_warehouse']['document_count'] = document_count
data['parent_warehouse'] = WarehouseSerializer(instance.parent_warehouse).data
else:
data['parent_warehouse'] = None
return data

73
src/documents/views.py
View File

@ -49,7 +49,7 @@ from django_filters.rest_framework import DjangoFilterBackend
from langdetect import detect
from packaging import version as packaging_version
from redis import Redis
from rest_framework import parsers
from rest_framework import parsers, status
from rest_framework.decorators import action
from rest_framework.exceptions import NotFound
from rest_framework.filters import OrderingFilter
@ -1520,10 +1520,28 @@ class BulkEditObjectsView(PassUserMixin):
)
elif operation == "delete" and object_type == "warehouses":
for warehouse_id in object_ids:
warehouse = Warehouse.objects.get(id=int(warehouse_id))
if warehouse.type == Warehouse.SHELF:
boxcases = Warehouse.objects.filter(parent_warehouse=warehouse)
documents = Document.objects.filter(warehouses__in=[b.id for b in boxcases])
documents.delete()
boxcases.delete()
warehouse.delete()
if warehouse.type == Warehouse.BOXCASE:
documents = Document.objects.filter(warehouses=warehouse)
documents.delete()
warehouse.delete()
if warehouse.type == Warehouse.WAREHOUSE:
shelves = Warehouse.objects.filter(parent_warehouse=warehouse)
boxcases = Warehouse.objects.filter(parent_warehouse__in=[s.id for s in shelves])
documents = Document.objects.filter(warehouses__in=[b.id for b in boxcases])
documents.delete()
boxcases.delete()
shelves.delete()
warehouse.delete()
documents = Document.objects.filter(warehouses__in=object_ids)
documents.delete()
objs.delete()
elif operation == "delete":
@ -1785,3 +1803,50 @@ class WarehouseViewSet(ModelViewSet, PermissionsAwareDocumentCountMixin):
filterset_class = WarehouseFilterSet
ordering_fields = ("name", "type", "parent_warehouse", "document_count")
def create(self, request, *args, **kwargs):
try:
print(request.data)
serializer = WarehouseSerializer(data=request.data)
name = request.data.get("name", "")
type = request.data.get("type", Warehouse.WAREHOUSE)
parent_warehouse = request.data.get("parent_warehouse", None)
check_warehouse = Warehouse.objects.filter(
name = name,
type = type,
parent_warehouse=parent_warehouse
)
print(check_warehouse)
if check_warehouse:
return Response({'status':400,
'message':'created fail'},status=status.HTTP_400_BAD_REQUEST)
if type == Warehouse.SHELF and parent_warehouse == None:
return Response({'status': 400,
'message': 'parent_warehouse is required for Shelf type'}, status=status.HTTP_400_BAD_REQUEST)
elif type == Warehouse.BOXCASE and parent_warehouse == None:
return Response({'status': 400,
'message': 'parent_warehouse is required for Boxcase type'}, status=status.HTTP_400_BAD_REQUEST)
id = request.data.get('parent_warehouse',0)
if serializer.is_valid(raise_exception=True):
parent_warehouse =Warehouse.objects.filter(id=id).first()
if request.data.get("type", "") == "" and not parent_warehouse:
serializer.save(type = Warehouse.WAREHOUSE)
elif request.data.get("type", "") == Warehouse.SHELF and getattr(parent_warehouse, 'type', "") == Warehouse.WAREHOUSE :
serializer.save(type = Warehouse.SHELF, parent_warehouse = parent_warehouse)
elif request.data.get("type", "") == Warehouse.BOXCASE and getattr(parent_warehouse, 'type', "") == Warehouse.SHELF :
serializer.save(type = Warehouse.BOXCASE, parent_warehouse = parent_warehouse)
else:
return Response({'status':400,
'message':'created fail'},status=status.HTTP_400_BAD_REQUEST)
return Response({'status':201,
'message':'created successfully',
'data':serializer.data},status=status.HTTP_201_CREATED)
except Exception as e:
return Response({'status':400,
'message':e},status=status.HTTP_400_BAD_REQUEST)