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

15
apps/em/filters.py
View File

@ -2,24 +2,29 @@ from django_filters import rest_framework as filters
from apps.em.models import Equipment
from datetime import datetime, timedelta
class EquipFilterSet(filters.FilterSet):
tag = filters.CharFilter(method='filter_tag')
class Meta:
model = Equipment
fields = {
"keeper": ["exact"],
"type": ["exact"],
"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):
now = datetime.now()
day7_after = now + timedelta(days=7)
if value == 'near_check':
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':
queryset = queryset.filter(
next_check_date__lte = datetime.date(now))
return queryset
next_check_date__lte=datetime.date(now))
return queryset