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

feat: 增加utask查询条件

This commit is contained in:
caoqianming 2023-10-26 16:40:34 +08:00
parent 4cea056955
commit 5ba3b4e07a
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
apps/pm/filters.py
View File

@ -6,6 +6,7 @@ from django.db.models import F
class UtaskFilter(filters.FilterSet): class UtaskFilter(filters.FilterSet):
tag = filters.CharFilter(method='filter_tag', label='done, not_done')
class Meta: class Meta:
model = Utask model = Utask
@ -28,6 +29,13 @@ class UtaskFilter(filters.FilterSet):
queryset = super().filter_queryset(queryset) queryset = super().filter_queryset(queryset)
return queryset.distinct() return queryset.distinct()
def filter_tag(self, queryset, name, value):
if value == 'done':
queryset = queryset.filter(count_ok__gte=F('count'))
elif value == 'not_done':
queryset = queryset.filter(count_ok__lt=F('count'))
return queryset
class MtaskFilter(filters.FilterSet): class MtaskFilter(filters.FilterSet):
tag = filters.CharFilter(method='filter_tag') tag = filters.CharFilter(method='filter_tag')