From b2780d5a3dfe3dbc43198e53d67f793408a814f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com>
Date: Wed, 13 Jul 2022 14:26:22 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=9F=A5=E8=AF=A2=E6=9D=A1?=
 =?UTF-8?q?=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/opm/views.py | 3 ++-
 apps/rpm/views.py | 2 +-
 apps/vm/views.py  | 1 +
 3 files changed, 4 insertions(+), 2 deletions(-)

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