add: add users field for warehouse model

core/apps/wherehouse/views/wherehouse.py

@@ -8,11 +8,20 @@ from core.apps.wherehouse.serializers import wherehouse as serializers
 from core.apps.accounts.permissions.permissions import HasRolePermission
 
 
-class WhereHouseListApiView(generics.ListAPIView):
+class WhereHouseListApiView(generics.GenericAPIView):
     serializer_class = serializers.WhereHouseListSerializer
-    queryset = WhereHouse.objects.all()
+    queryset = WhereHouse.objects.prefetch_related('users')
     permission_classes = [HasRolePermission]
 
+    def get(self, request):
+        mine = request.query_params.get('mine')
+        if mine and mine == 'true':
+            self.queryset = self.queryset.filter(users=request.user)
+        page = self.paginate_queryset(self.queryset)
+        if page is not None:
+            serializer = self.serializer_class(page, many=True)
+            return self.get_paginated_response(serializer.data)
+
 
 class WhereHouseDetailApiView(generics.RetrieveAPIView):
     serializer_class = serializers.WhereHouseDetailSerializer