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

View File

@ -6,6 +6,7 @@ from django.db.models import F
class UtaskFilter(filters.FilterSet):
tag = filters.CharFilter(method='filter_tag', label='done, not_done')
class Meta:
model = Utask
@ -28,6 +29,13 @@ class UtaskFilter(filters.FilterSet):
queryset = super().filter_queryset(queryset)
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):
tag = filters.CharFilter(method='filter_tag')