diff --git a/apps/qm/filters.py b/apps/qm/filters.py index 198f5528..31d11c8e 100644 --- a/apps/qm/filters.py +++ b/apps/qm/filters.py @@ -38,6 +38,7 @@ class TestItemFilter(filters.FilterSet): class FtestWorkFilter(filters.FilterSet): + cbatch = filters.CharFilter(label='批次号', method='filter_cbatch') class Meta: model = FtestWork fields = { @@ -50,6 +51,11 @@ class FtestWorkFilter(filters.FilterSet): "type2": ["exact"], "shift": ["exact"] } + + def filter_cbatch(self, queryset, name, value): + qs1 = queryset.filter(wm__batch=value) + qs2 = queryset.filter(mb__batch=value) + return qs1.union(qs2) class FtestFilter(filters.FilterSet): wpr = filters.CharFilter(label="wprId", method="filter_wpr")