Django+DRF RESTful API OR条件过滤问题（非get_queryset修改方案）

不用修改get_queryset的话，django-filter是DRF生态里最常用的过滤扩展包，完美支持OR逻辑过滤，步骤如下：

1. 安装包 #

pip install django-filter

2. 配置项目 #

在settings.py里添加app并配置DRF的过滤后端：

INSTALLED_APPS = [
    # 你的其他应用
    'django_filters',
]

REST_FRAMEWORK = {
    'DEFAULT_FILTER_BACKENDS': [
        'django_filters.rest_framework.DjangoFilterBackend',
    ],
}

3. 定义FilterSet实现OR过滤 #

假设你有一个Book模型，需要实现标题包含关键词 或 作者包含关键词的OR过滤，创建filters.py：

import django_filters
from .models import Book
from django.db.models import Q

class BookFilter(django_filters.FilterSet):
    # 自定义一个搜索参数，用method实现OR逻辑
    search = django_filters.CharFilter(method='filter_or_conditions')

    def filter_or_conditions(self, queryset, name, value):
        # 这里用Q对象组合OR条件
        return queryset.filter(
            Q(title__icontains=value) | Q(author__icontains=value)
        )

    class Meta:
        model = Book
        fields = ['search']

4. 在视图中使用FilterSet #

直接在ViewSet里指定filterset_class即可，不需要改动get_queryset：

from rest_framework import viewsets
from .models import Book
from .serializers import BookSerializer
from .filters import BookFilter

class BookViewSet(viewsets.ModelViewSet):
    queryset = Book.objects.all()
    serializer_class = BookSerializer
    filterset_class = BookFilter

现在前端请求时，只需传入?search=xxx，就能返回标题或作者包含xxx的所有对象，完全满足OR过滤需求。如果需要针对多个独立参数做OR组合（比如?title=xxx&author=xxx时返回满足任一条件的结果），也可以在FilterSet里调整逻辑，django-filter的灵活性足够覆盖大部分场景。

内容的提问来源于stack exchange，提问作者Maximo Arceci