zcdsj
/
hberp
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wmaterial filterset

This commit is contained in:
caoqianming 2021-11-22 14:59:45 +08:00
parent b80b3ac5f2
commit 0238e09d0a
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
hb_server/apps/wpm/filters.py
View File

@ -1,4 +1,5 @@
from django_filters import rest_framework as filters
from apps.mtm.models import Step
from apps.wpm.services import WpmServies
from .models import Operation, WMaterial, WProduct
@ -13,8 +14,12 @@ class WMaterialFilterSet(filters.FilterSet):
def filter_operation(self, queryset, name, value):
operation = Operation.objects.get(pk=value)
wproducts = WProduct.objects.filter(ow_wproduct__operation=value)
step = operation.step
if wproducts.exists():
subplans = WpmServies.get_subplans_queryset_from_wproducts(wproducts)
else:
subplans = WpmServies.get_subplans_queyset_from_step(operation.step)
return queryset.filter(subproduction_plan__in=subplans)
subplans = WpmServies.get_subplans_queyset_from_step(step)
queryset = queryset.filter(subproduction_plan__in=subplans)
# if step.type == Step.STEP_TYPE_NOM:
# queryset = queryset.filter()
return queryset