或使用django-filter过滤MultipleChoices

Question

我有一个表单字段，其中包含[1,2,3,4 +]等选项，4 +表示greater than equal，可以选择多个选项。我想使用django-filter进行过滤。

我可以为[1,2,3]进行过滤，但我不知道or如何使用gte=4。

以下工作用于过滤[1,2,3]：

class NumberInFilter(django_filters.BaseInFilter, django_filters.NumberFilter):
    pass

class PlanFilter(FilterSet):
    obj = django_filters.NumberInFilter(name='object', lookup_expr='in')
    class Meta:
        model = Plan 
        fields= ['object',]

或

choices= ( 
         (1,1),
         (2,2),
         (3,3),
         )

class PlanFilter(FilterSet):
    obj = django_filters.MultipleChoiceFilter(name='object', choices=choices) 
    ...

那么如何使用gte=4字段过滤多个选项？

Answer 1

任何一个选项都只需要对滤波器逻辑进行最小的更改。

NumberInFilter

的自定义class NumberInFilter(django_filters.BaseInFilter, django_filters.NumberFilter): def filter(self, qs, value): # CSV-based filters receive a list of values if 4 not in value: return super(NumberInFilter, self).filter(qs, value) value.remove(4) # a little bit more verbose, but doesn't require hardcoding the name. # eg, `qs.filter(Q(object__in=value) | Q(object__gte=4))` q1 = Q(**{'%s__%s' % (self.name, 'in'): value}) q2 = Q(**{'%s__%s' % (self.name, 'gte'): 4}) return qs.filter(q1 | q2) 方法

.get_filter_predicate()

覆盖MultipleChoiceFilter

中的ObjectMultipleChoiceFilter(django_filters.MultipleChoiceFilter): def get_filter_predicate(self, value): if value == '4': return {'%s__gte' % self.name: 4} return super(ObjectMultipleChoiceFilter, self).get_filter_predicate(value)

edgePadding