feat: 增加筛选条件

This commit is contained in:
caoqianming 2023-05-29 14:42:56 +08:00
parent 4f58ed67e0
commit d533b1c93c
1 changed files with 7 additions and 0 deletions

View File

@ -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)