使用自定义筛选器与Django ModelChoice筛选器

时间:2017-03-24 21:35:30

标签: python django django-filters

我正在尝试使用django-filters

创建过滤方法

请参阅以下模型的示例:

class Chicken(TimeStampedModel):
    eggs = ForeignKey(Egg)

class Egg(TimeStampedModel):
    hatched = BooleanField(default=False)

查看我当前为过滤器设置的示例:

class ChickenFilter(FilterSet):
    eggs__contains = ModelChoiceFilter(name="eggs", method='hatched_eggs', queryset=Eggs.objects.all())

   def hatched_eggs(self, queryset, name, value):
      print "We got eggs"
     return queryset.filter(eggs__hatched=True)

问题是该方法甚至不打印当我点击网址时我们得到了鸡蛋。它只返回一个空的查询集。

2 个答案:

答案 0 :(得分:1)

我遇到了同样的问题。我的method没有被调用。

所以得出结论,我可以使用以下方法进行解决:

Using custom methods in filter with django-rest-framework

class ChickenFilter(FilterSet):
    eggs__contains = CharFilter(action='hatched_eggs')

def hatched_eggs(queryset, value):
      print "We got eggs"
     if value:
         return queryset.filter(eggs__hatched=True)
     return queryset

答案 1 :(得分:0)

我像上面那样做

在我的网址中,我发送的邮件类似于?ids=1,2,3,4

class MyFilter(filters.FilterSet):
    ids = django_filters.CharFilter(method='ids__in')

    def ids__in(self, queryset, value, *args, **kwargs):
        try:
            if args:
                ids = args[0].split(',')
                ids = [int(_id) for _id in ids]
                queryset = queryset.filter(id__in=ids)
        except ValueError:
            pass
        return queryset