From d533b1c93c26ad6749bcb2e030ae0f2c3e7d331f Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 29 May 2023 14:42:56 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E7=AD=9B=E9=80=89?=
 =?UTF-8?q?=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/apps/supervision/views.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/server/apps/supervision/views.py b/server/apps/supervision/views.py
index 5f3d51e..8a775cd 100644
--- a/server/apps/supervision/views.py
+++ b/server/apps/supervision/views.py
@@ -578,6 +578,7 @@ class Domixin:
 class PgoalDeptViewSet(Domixin, CreateUpdateCustomMixin, ModelViewSet):
     queryset = PgoalDept.objects.all()
     serializer_class = PgoalDeptSerializer
+    filterset_fields = ['goal_key', 'belong_dept', 'year']
 
     def perform_update(self, serializer):
         instance = serializer.save()
@@ -632,6 +633,7 @@ class RcViewSet(ImpMixin, Domixin, CreateUpdateCustomMixin, ModelViewSet):
         "num_error": {'etype': 10},
         "num_overdue": {'etype': 20}
     }
+    filterset_fields = ['etype', 'task2do', 'task2do__belong_dept', 'task2do__year']
 
 
     def make_data(self, data, sheet, i, etype):
@@ -676,6 +678,8 @@ class ComplaintViewSet(ImpMixin, Domixin, CreateUpdateCustomMixin, ModelViewSet)
     count_dict = {
         "num_complaint": {}
     }
+    filterset_fields = ['task2do', 'task2do__belong_dept', 'task2do__year']
+
     def make_data(self, data, sheet, i, etype):
         data['complainant'] = sheet['b'+str(i)].value
         data['date_occurrence'] = sheet['c'+str(i)].value
@@ -700,6 +704,7 @@ class PtViewSet(ImpMixin, Domixin, CreateUpdateCustomMixin, ModelViewSet):
     count_dict = {
         "num_pt": {}
     }
+    filterset_fields = ['task2do', 'task2do__belong_dept', 'task2do__year']
 
     def make_data(self, data, sheet, i, etype):
         data['name'] = sheet['b'+str(i)].value
@@ -726,6 +731,7 @@ class RiskViewSet(ImpMixin, Domixin, CreateUpdateCustomMixin, ModelViewSet):
     count_dict = {
         "num_risk": {}
     }
+    filterset_fields = ['task2do', 'task2do__belong_dept', 'task2do__year']
 
     def make_data(self, data, sheet, i, etype):
         data['content'] = sheet['b'+str(i)].value
@@ -750,6 +756,7 @@ class OinspectViewSet(ImpMixin, Domixin, CreateUpdateCustomMixin, ModelViewSet):
     count_dict = {
         "num_oinspect": {}
     }
+    filterset_fields = ['task2do', 'task2do__belong_dept', 'task2do__year']
 
     def make_data(self, data, sheet, i, etype):
         data['cate'] = self.get_enum(sheet['b'+str(i)].value, Oinspect.oinspect_cate_choices, i)