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 = {