diff --git a/core/apps/counterparty/views/counterparty.py b/core/apps/counterparty/views/counterparty.py index 68d9a94..9a3d63e 100644 --- a/core/apps/counterparty/views/counterparty.py +++ b/core/apps/counterparty/views/counterparty.py @@ -32,9 +32,13 @@ class CounterpartyCreateApiView(generics.GenericAPIView): def post(self, request): serializer = self.serializer_class(data=request.data) if serializer.is_valid(raise_exception=True): - serializer.save() + data = serializer.save() return Response( - {'success': True, 'message': 'Conterparty Created'}, + { + 'success': True, + 'message': 'Conterparty Created', + 'data': serializers.CounterpartyListSerializer(data).data + }, status=201 ) return Response( diff --git a/core/apps/orders/views/party.py b/core/apps/orders/views/party.py index cb4214b..0e4764d 100644 --- a/core/apps/orders/views/party.py +++ b/core/apps/orders/views/party.py @@ -258,4 +258,3 @@ class ChangeConfirmationPartyApiView(views.APIView): }, status=200 ) - \ No newline at end of file diff --git a/core/apps/projects/views/project.py b/core/apps/projects/views/project.py index a1538ff..1e0b268 100644 --- a/core/apps/projects/views/project.py +++ b/core/apps/projects/views/project.py @@ -175,10 +175,13 @@ class ProjectAndFolderApiView(views.APIView): def get(self, request): mine = request.query_params.get('is_mine') + wherehouse_ids = request.query_params.getlist('wherehouse') folders = ProjectFolder.objects.prefetch_related('projects') projects = Project.objects.exclude(folder__isnull=False) if mine and mine.lower() == 'true': projects = projects.filter(Q(boss=request.user) | Q(foreman=request.user) | Q(other_members=request.user)) + if wherehouse_ids: + projects = projects.filter(wherehouse__in=wherehouse_ids) projects_serializer = serializers.ProjectsSerializer(projects, many=True) folders_serializer = serializers.ProjectFoldersSerializer(folders, many=True) return Response(