django-rest-framework - 将过滤器应用于任何或某个多对多记录

问题描述

考虑这些模型：

from django.db import models

class Group(models.Model):
    name = models.CharField()

class Person(models.Model):
    height = models.PositiveIntegerField()
    weight = models.PositiveIntegerField()
    gender = models.CharField()
    groups = models.ManyToManyField(Group, blank=True)

和 DRF 视图

from rest_framework import viewsets
from rest_framework.filters import SearchFilter, OrderingFilter

from django_filters import rest_framework as filters

from .serializers import GroupSerializer
from ..models import Group

class GroupViewSet(viewsets.ModelViewSet):
    queryset = Group.objects.all().distinct()
    serializer_class = GroupSerializer
    filter_backends = (filters.DjangoFilterBackend,
                       SearchFilter, OrderingFilter)
    filter_class = GroupFilter

一个组可以有 0,1,2 或多个Persons，其中 1 和 2 是最常见的，其中 1 和 2 是明确定义的。把它想象成 Facebook 的聊天：最常见的是一对一聊天，但有时也可以进行群聊。什么时候是一对一聊天，1是发送者，2是接收者。

我需要从 DRF 中过滤这些记录，在浏览GroupViewSet和过滤Person属性时，我可以将一组过滤器应用于任何Person或某个Person.

对于任何人，无论适用于哪个特定条件，都是明确的：

/api/group/?person__height__gt=100&person__weight__gt=200

但是对于某个人，如果一组条件适用于那个人，在 URL 中，我可以有类似的内容：

/api/group/?person__0__height__gt=100&person__0__weight__gt=200&person__1__height__lte=200

并将这些声明到我的自定义中FilterSet：

from django.db.models.constants import LOOKUP_SEP

class GroupFilter(filters.FilterSet):
    person__0__height = filters.NumberFilter(method='person_filter')
    person__0__height__gt = filters.NumberFilter(method='person_filter')
    person__0__height__lt = filters.NumberFilter(method='person_filter')
    # ... and so on for the rest of the possibilities 

    def person_filter(self, queryset, name, value):
        m2mfield, index, field, *comparison = name.split(LOOKUP_SEP, 3)
        # do subqueries based on the above and construct queryset filter.

但正如您可以想象的那样，这意味着我将有很多样板代码。在我的真实模型中，有很多领域，上面的“解决方案”对我来说似乎很老套。

所以问题是：是否有更简单/更清洁的方法来实现上述过滤？

也许通过动态声明person__0__height__gt属性，我还没有找到解决方案。

请注意，我预先不知道Person实体的 ID。那些 person__0, person__1 是数组索引。

标签： django-rest-frameworkdjango-querysetdjango-filters

class GroupFilter(django_filters.FilterSet): person_range = django_filters.NumericRangeFilter(field_name='person__0__height', lookup_expr='range') person = django_filters.NumberFilter(field_name='person__0__height', lookup_expr='exact') class Meta: model = Group fields = ('person_range','person',)

django-rest-framework - 将过滤器应用于任何或某个多对多记录

问题描述

解决方案

推荐阅读