feat: 车间库存支持传入count_all

2025-12-24 11:09:12 +08:00 · 2025-12-24 11:09:12 +08:00 · f84ceaa95e
parent da1f587036
commit f84ceaa95e
1 changed files with 5 additions and 0 deletions
--- a/apps/wpm/views.py
+++ b/apps/wpm/views.py
@ -174,6 +174,11 @@ class WMaterialViewSet(CustomListModelMixin, CustomGenericViewSet):
    filterset_class = WMaterialFilter
    ordering_fields = ["update_time", "state", "count", "count_xtest"]

+    def get_queryset(self):
+        if self.request.query_params.get("count_all"):
+            self.queryset = WMaterial.objects.all()
+        return super().get_queryset()
+
    def filter_queryset(self, queryset):
        queryset = super().filter_queryset(queryset)
        if self.request.query_params.get("state_all"):