diff --git a/apps/opm/views.py b/apps/opm/views.py index ae59f1f4..896a4d96 100644 --- a/apps/opm/views.py +++ b/apps/opm/views.py @@ -24,6 +24,7 @@ class OperationViewSet(CustomModelViewSet): create_serializer_class = OperationCreateUpdateSerializer update_serializer_class = OperationCreateUpdateSerializer serializer_class = OperationSerializer + filterset_fields = ['state', 'opl_operation__cate'] def update(self, request, *args, **kwargs): obj = self.get_object() @@ -45,7 +46,7 @@ class OplViewSet(CustomModelViewSet): update_serializer_class = OplCreateUpdateSerializer serializer_class = OplSerializer select_related_fields = ['cate'] - filterset_fields = ['operation'] + filterset_fields = ['operation', 'cate'] def destroy(self, request, *args, **kwargs): obj = self.get_object() diff --git a/apps/rpm/views.py b/apps/rpm/views.py index 44650cc6..1a543264 100644 --- a/apps/rpm/views.py +++ b/apps/rpm/views.py @@ -111,7 +111,7 @@ class RpjViewSet(CustomModelViewSet): update_serializer_class = RpjCreateUpdateSerializer serializer_class = RpjListSerializer select_related_fields = ['rparty', 'belong_dept'] - filterset_fields = ['rparty', 'belong_dept'] + filterset_fields = ['rparty', 'belong_dept', 'state'] def get_queryset(self): queryset = super().get_queryset() diff --git a/apps/vm/views.py b/apps/vm/views.py index c777b5a0..3ce06afc 100644 --- a/apps/vm/views.py +++ b/apps/vm/views.py @@ -21,6 +21,7 @@ class VisitViewSet(CustomModelViewSet): update_serializer_class = VisitCreateUpdateSerializer serializer_class = VisitSerializer retrieve_serializer_class = VisitDetailSerializer + filterset_fields = ['state'] def get_queryset(self): user = self.request.user