Django FilterSet AND(?) 条件

Question

我正在创建一个应用程序，其中食物有多个成分，每个成分都有特定的价值。我希望能够根据特定成分的价值过滤食物。

这是我的模型：

class Food(models.Model):
    id = models.AutoField(primary_key=True, db_column="id")
    name = models.CharField(max_length=150)
    product_type = models.CharField(max_length=150)

class Component(models.Model):

    id = models.AutoField(primary_key=True, db_column="id")
    name = models.CharField(max_length=150)
    value = models.FloatField()
    food = models.ForeignKey(Food, related_name='food_components', related_query_name='food_components')

然后我定义过滤器集和视图

import rest_framework_filters as filters

class ComponentFilter(filters.FilterSet):

    name = filters.CharFilter(name="name")
    val = filters.NumberFilter(name="value", distinct=True)
    minval = filters.NumberFilter(name="value", lookup_type="gte", distinct=True)
    maxval = filters.NumberFilter(name="value", lookup_type="lte", distinct=True)

    class Meta:
        model = Component

class FoodFilter(filters.FilterSet):

    name = filters.CharFilter(name='name')
    product_type = filters.CharFilter(name='product_type')
    components = filters.RelatedFilter(ComponentFilter, name='food_components')

    class Meta:
        model = Food

class FoodViewSet(viewsets.ModelViewSet):

    queryset = Food.objects.all()
    serializer_class = FoodSerializer
    filter_class = FoodFilter

我希望能够使用相关模型中的多个字段（结合两个参数）来过滤视图集。像http://whatever/foods/components__name=X&components__value=1 这样的东西会让我得到所有元素 X 值为 1 的食物。我正在使用 django-rest-framework-filters (https://github.com/philipn/django-rest-framework-filters)

据我所知，由于所有食物都具有所有元素，因此 components__name 无关紧要，我将获得所有具有任何值为 1 的组件的食物。如何将两者结合起来？

提前致谢！

Answer 1

您需要创建一个新过滤器来处理与给定Component 相关的单个 Food 对象上的name、val 对：

class FoodComponentFilter(django_filters.Filter):
    def filter(self, qs, value):
        strs = value.split(',')
        if len(strs) != 2:
            raise Exception
        return qs.filter(
            food_components__in = Component.objects.filter(
                name=strs[0],
                value=float(strs[1])
            )
        )

在FoodFilter 中实例化这个：

class FoodFilter(filters.FilterSet):
   component_food_pair = FoodComponentFilter(name='dummy_field')
   [...]
   class Meta:
        model = Food
        fields = ('component_food_pair',[...])

然后，像这样使用它：

?component_food_pair=Foo,42.0