diff --git a/apps/wpm/views.py b/apps/wpm/views.py index 1fe3843d..41f88666 100644 --- 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"):