feat: 增加mlog查询条件

apps/wpm/filters.py

@@ -1,5 +1,5 @@
 from django_filters import rest_framework as filters
-from apps.wpm.models import SfLog, SfLogExp, WMaterial
+from apps.wpm.models import SfLog, SfLogExp, WMaterial, Mlog
 ['mgroup', 'shift', 'team', 'leader', 'team__belong_dept']
 
 
@@ -49,3 +49,15 @@ class WMaterialFilter(filters.FilterSet):
             "batch": ["exact"],
             "count": ["gte", "lte", "exact"]
         }
+
+
+class MlogFilter(filters.FilterSet):
+    class Meta:
+        model = Mlog
+        fields = {
+            "mtask": ["exact", "in"],
+            "batch": ["exact"],
+            "handle_date": ["exact"],
+            "handle_user": ["exact"],
+            "mtask__mgroup__belong_dept__name": ["exact"]
+        }

apps/wpm/views.py

@@ -10,7 +10,7 @@ from apps.pm.models import Mtask
 from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet
 from apps.utils.mixins import BulkCreateModelMixin
 
-from .filters import SfLogExpFilter, SfLogFilter, WMaterialFilter
+from .filters import SfLogExpFilter, SfLogFilter, WMaterialFilter, MlogFilter
 from .models import SfLog, SfLogExp, StLog, WMaterial, Mlog
 from .serializers import SflogExpSerializer, SfLogSerializer, StLogSerializer, WMaterialSerializer, MlogSerializer, MlogRelatedSerializer, DeptBatchSerializer
 
@@ -121,7 +121,7 @@ class MlogViewSet(CustomModelViewSet):
     queryset = Mlog.objects.all()
     serializer_class = MlogSerializer
     select_related_fields = ['create_by', 'update_by', 'mtask']
-    filterset_fields = ['mtask', 'handle_user']
+    filterset_class = MlogFilter
 
     def perform_destroy(self, instance):
         if instance.mtask.state == Mtask.MTASK_DONE: