API 端点中的多个过滤器参数 - DRF

Question

我将Django rest framework 用于我的 REST API。我有一些用于过滤的参数的端点，如下所示：

# in my view
from_date = self.request.query_params.get('from', None)
to_date = self.request.query_params.get('to', None)
category = self.request.query_params.get('category', None)
color = self.request.query_params.get('color', None)

然后我在选择数据的地方有 if/else 代码。第一次只有两个参数所以没问题，但后来我添加了更多参数进行过滤，现在我认为这段代码看起来不太好，我认为需要进行一些优化。此外，不同参数的组合现在不起作用。

实际代码（在我看来）：

if from_date is not None and to_date is not None:
    something = Something.objects.filter(send_date__range=(from_date, to_date))
elif category:
    something = Something.objects.filter(values__something__category__name__iexact=category)
elif color:
    something = Something.objects.filter(values__something__color__name__iexact=color)
else:
    something = Something.objects.all()

是否有任何选项，内置于 DRF 或 Django 中的某种类型的组合过滤，考虑到值可以为无，我如何将参数组合成一个大查询？

例如，如果用户使用日期发送 from_date 和 to_date 查询过滤数据。但用户也希望使用 from_date、to_date 和 color 或其他参数组合过滤数据。有些参数可以省略。该代码只是最小版本。我有更多参数，我不想为每个参数组合编写 if 和 else。

类似的东西（但如果其中一个值为 None 则没有此值查询将起作用）：

Something.objects.filter(send_date__range=(from_date, to_date))
    .filter(values__something__category__name__iexact=category)
    .filter(values__something__color__name__iexact=color)

Answer 1

将参数转换为查询过滤器不能完全自动化，因为只有您知道哪些参数应该以哪种方式过滤哪些字段。但是，除非你有 or 和 and 标准的疯狂组合，否则它应该非常简单，因为你可以链接 .filter() 调用而没有任何限制，只要你想 and 所有标准：

something = Something.objects.all()
if from_date and to_date:
    something = something.filter(send_date__range=(from_date, to_date))
if category:
    something = something.filter(values__something__category__name__iexact=category)
if color:
    something = something.filter(values__something__color__name__iexact=color)

这涵盖了所有组合，看起来很容易管理，不是吗？

如果您需要or（某些）您的条件，则需要Q objects，但是您可以以相同的方式链接：

from django.db.models.query_utils import Q

qq = Q()
if category:
    qq = qq | Q(values__something__category__name__iexact=category)
if color:
    qq = qq | Q(values__something__color__name__iexact=color)
something = Something.objects.filter(qq)

Answer 2

这是我用于自定义过滤的模式。如果您最终需要围绕每个过滤器参数进行更多自定义实现，您可以使用策略设计模式之类的东西，它可以很好地摆脱 if-else 链。

FILTER_PARAMS = {'from': 'from_date', 'to': 'to_date', 'category': 'values__something__category__name__iexact', 'color': 'values__something__color__name__iexact'}

def get(self):
        filter_params = self.get_filter_params(self.request.query_params)
        qs = queryset.filter(**filter_params)

def get_filter_params(self, query_params):
        fields = {}

        for k, v in query_params.items():
                if k in self.FILTER_PARAMS:
                    fields[self.FILTER_PARAMS[k]] = v
        return fields