From 784fe20e78de2b1b1a6aac0ceda71038cf70a52f Mon Sep 17 00:00:00 2001 From: hungdztrau123 Date: Fri, 7 Jun 2024 13:36:08 +0700 Subject: [PATCH] fix:filter-warehouse --- src/documents/views.py | 30 +++++++++++++++++++++++++++--- 1 file changed, 27 insertions(+), 3 deletions(-) diff --git a/src/documents/views.py b/src/documents/views.py index 08ad0cfd9..9e4f1f764 100644 --- a/src/documents/views.py +++ b/src/documents/views.py @@ -755,11 +755,10 @@ class DocumentViewSet( def get_queryset(self): queryset = self.queryset warehouse_id = self.request.query_params.get('warehouse_id', None) - # folder_id = self.request.query_param.get('folder_id', None) + if warehouse_id is not None: queryset = self.get_warehouse(warehouse_id) - # if folder_id is not None: - # queryset = self.get_folder(folder_id) + return queryset def get_warehouse(self, warehouse_id): @@ -779,7 +778,32 @@ class DocumentViewSet( else: return Document.objects.none() + # def get_queryset(self): + # queryset = self.queryset + # folder_id = self.request.query_params.get('folder_id', None) + # if folder_id is not None: + # queryset = self.get_folder(folder_id) + # return queryset + + # return queryset + + # def get_folder(self, folder_id): + # folder = Folder.objects.get(id=int(folder_id)) + # return self.get_folder_documents(folder) + + # def get_folder_documents(self, folder): + # documents = Document.objects.filter(folder=folder) + # child_folders = Folder.objects.filter(parent_folder=folder) + + # return documents + + + + + + + class SearchResultSerializer(DocumentSerializer, PassUserMixin): def to_representation(self, instance): doc = (