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

feat: equip增加工段等筛选条件

This commit is contained in:
caoqianming 2023-12-27 11:06:05 +08:00
parent c91533a328
commit 1134fc3809
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
apps/em/filters.py
View File

@ -2,15 +2,20 @@ from django_filters import rest_framework as filters
from apps.em.models import Equipment from apps.em.models import Equipment
from datetime import datetime, timedelta from datetime import datetime, timedelta
class EquipFilterSet(filters.FilterSet): class EquipFilterSet(filters.FilterSet):
tag = filters.CharFilter(method='filter_tag') tag = filters.CharFilter(method='filter_tag')
class Meta: class Meta:
model = Equipment model = Equipment
fields = { fields = {
"keeper": ["exact"], "keeper": ["exact"],
"type": ["exact"], "type": ["exact"],
"power_kw": ["gte"], "power_kw": ["gte"],
"belong_dept": ["exact"] "belong_dept": ["exact"],
"belong_dept__name": ["exact"],
"mgroup": ["exact"],
"mgroup__name": ["exact"]
} }
def filter_tag(self, queryset, name, value): def filter_tag(self, queryset, name, value):
@ -18,8 +23,8 @@ class EquipFilterSet(filters.FilterSet):
day7_after = now + timedelta(days=7) day7_after = now + timedelta(days=7)
if value == 'near_check': if value == 'near_check':
queryset = queryset.filter( queryset = queryset.filter(
next_check_date__lt = datetime.date(day7_after), next_check_date__gte=datetime.date(now)) next_check_date__lt=datetime.date(day7_after), next_check_date__gte=datetime.date(now))
elif value == 'out_check': elif value == 'out_check':
queryset = queryset.filter( queryset = queryset.filter(
next_check_date__lte = datetime.date(now)) next_check_date__lte=datetime.date(now))
return queryset return queryset