From 00677d100b3251e002a2c40b666e582d54f683db Mon Sep 17 00:00:00 2001 From: caoqianming Date: Tue, 25 Jul 2023 11:33:33 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20swagger=E6=98=BE=E7=A4=BA=E9=97=AE?= =?UTF-8?q?=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/qm/filters.py | 14 ++++++++++++++ apps/qm/views.py | 4 ++-- 2 files changed, 16 insertions(+), 2 deletions(-) create mode 100644 apps/qm/filters.py diff --git a/apps/qm/filters.py b/apps/qm/filters.py new file mode 100644 index 00000000..a70bbf1d --- /dev/null +++ b/apps/qm/filters.py @@ -0,0 +1,14 @@ +from django_filters import rest_framework as filters +from apps.qm.models import QuaStat + +class QuaStatFilter(filters.FilterSet): + class Meta: + model = QuaStat + fields = { + "material": ["exact"], + "testitem": ["exact"], + "belong_dept": ["exact"], + "sflog": ["exact"], + "sflog__mgroup": ["exact"], + "sflog__end_time": ["day", "month", "year", "lt"], + } \ No newline at end of file diff --git a/apps/qm/views.py b/apps/qm/views.py index 2cb4b787..09993b96 100644 --- a/apps/qm/views.py +++ b/apps/qm/views.py @@ -10,7 +10,7 @@ import datetime from apps.utils.viewsets import CustomGenericViewSet from apps.wpm.models import SfLog - +from apps.qm.filters import QuaStatFilter # Create your views here. class TestItemViewSet(ListModelMixin, CustomGenericViewSet): """ @@ -34,7 +34,7 @@ class QuaStatViewSet(ListModelMixin, BulkUpdateModelMixin, CustomGenericViewSet) queryset = QuaStat.objects.all() serializer_class = QuaStatSerializer update_serializer_class = QuaStatUpdateSerializer - filterset_fields = ['material', 'testitem', 'belong_dept', 'sflog', 'sflog__mgroup', 'sflog__shift', 'sflog__end_time__year', 'sflog__end_time__month', 'sflog__end_time__day'] + filterset_class = QuaStatFilter select_related_fields = ['belong_dept', 'material', 'testitem'] ordering = ['belong_dept__sort', 'material__sort', 'testitem__sort']