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)