From 4177ba93f8507407c0a8f65e3f2dedf122973a9c Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Thu, 11 Jul 2024 13:13:39 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20wmaterial=E5=A2=9E=E5=8A=A0mgroupx=20?=
 =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E6=9D=A1=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/wpm/filters.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/apps/wpm/filters.py b/apps/wpm/filters.py
index e596973d..abaddda5 100644
--- a/apps/wpm/filters.py
+++ b/apps/wpm/filters.py
@@ -1,5 +1,5 @@
 from django_filters import rest_framework as filters
-from apps.wpm.models import SfLog, StLog, WMaterial, Mlog, Handover
+from apps.wpm.models import SfLog, StLog, WMaterial, Mlog, Handover, Mgroup
 
 
 class SfLogFilter(filters.FilterSet):
@@ -30,6 +30,10 @@ class StLogFilter(filters.FilterSet):
 
 
 class WMaterialFilter(filters.FilterSet):
+    mgroupx = filters.CharFilter(label='MgroupId', method='filter_mgroupx')
+
+    def filter_mgroupx(self, queryset, name, value):
+        return queryset.filter(mgroup__id=value)|queryset.filter(belong_dept=Mgroup.objects.get(id=value).belong_dept, mgroup=None)
     class Meta:
         model = WMaterial
         fields = {