Django Rest Framework多查询集合并及django-filter实现模型多过滤接口

嘿，这两个Django Rest Framework的问题我刚好有实际开发经验，给你详细拆解下～

这个需求得分两种情况处理，取决于你要合并的是同模型还是不同模型的查询集：

情况1：合并同模型的多个查询集 #

如果是同一个模型下的不同查询结果，直接用Python的**管道符|**就能合并，而且合并后的结果依然是QuerySet类型，完全支持DRF的分页、过滤等后续操作，非常方便。示例代码：

from .models import Places

# 先定义两个不同的查询集
queryset_new = Places.objects.filter(new_place=True)
queryset_editor = Places.objects.filter(editor_choice=True)

# 合并成一个查询集
combined_queryset = queryset_new | queryset_editor

情况2：合并不同模型的多个查询集 #

如果是不同模型的查询集，需要分两种场景：

• 模型字段结构相似：可以用itertools.chain把多个查询集拼起来，但注意返回的是迭代器而非QuerySet，所以序列化的时候要分别处理再合并数据：

from itertools import chain
from rest_framework.views import APIView
from rest_framework.response import Response
from .models import Places, Events  # 假设Events是另一个结构相似的模型
from .serializers import PlacesSerializer, EventsSerializer

class CombinedView(APIView):
    def get(self, request):
        # 获取两个模型的查询集
        places = Places.objects.all()
        events = Events.objects.all()
        
        # 分别序列化后合并数据
        places_data = PlacesSerializer(places, many=True).data
        events_data = EventsSerializer(events, many=True).data
        combined_data = places_data + events_data
        
        return Response(combined_data)

• 模型字段完全一致：可以用Django的union()或union_all()方法合并，这样返回的还是QuerySet，支持后续操作，但要求两个模型的字段数量、类型完全匹配：

combined_queryset = Places.objects.all().union(Events.objects.all())

针对你说的Places模型需求，我推荐两种实用的实现方式：

方式1：创建两个独立的接口 #

直接写两个视图和路由，一个返回全部实例，一个返回new_place=True的实例，同时保留django-filter的过滤能力：

第一步：定义FilterSet #

先在filters.py里创建过滤规则：

import django_filters
from .models import Places

class PlacesFilter(django_filters.FilterSet):
    class Meta:
        model = Places
        fields = ['new_place', 'editor_choice']  # 可以添加其他需要过滤的字段

第二步：编写视图 #

在views.py里创建两个列表视图：

from rest_framework import generics
from .models import Places
from .serializers import PlacesSerializer
from .filters import PlacesFilter

class AllPlacesView(generics.ListAPIView):
    queryset = Places.objects.all()
    serializer_class = PlacesSerializer
    filterset_class = PlacesFilter  # 支持通过查询参数额外过滤，比如?editor_choice=true

class NewPlacesView(generics.ListAPIView):
    queryset = Places.objects.filter(new_place=True)
    serializer_class = PlacesSerializer
    filterset_class = PlacesFilter  # 同样可以叠加其他过滤，比如?editor_choice=true

第三步：配置路由 #

在urls.py里添加两个路由：

from django.urls import path
from .views import AllPlacesView, NewPlacesView

urlpatterns = [
    path('places/', AllPlacesView.as_view(), name='all-places'),
    path('places/new/', NewPlacesView.as_view(), name='new-places'),
]

这样用户访问/places/就能拿到全部实例，访问/places/new/就只返回标记为新地点的实例，而且两个接口都支持通过查询参数叠加其他过滤条件。

方式2：用单个接口加查询参数控制过滤 #

如果不想写两个接口，可以通过查询参数来动态切换过滤规则，比如用?new_only=true来触发new_place=True的过滤：

from rest_framework import generics
from .models import Places
from .serializers import PlacesSerializer
from .filters import PlacesFilter

class PlacesListView(generics.ListAPIView):
    serializer_class = PlacesSerializer
    filterset_class = PlacesFilter

    def get_queryset(self):
        queryset = Places.objects.all()
        # 获取查询参数，默认返回全部
        new_only = self.request.query_params.get('new_only', 'false')
        if new_only.lower() == 'true':
            queryset = queryset.filter(new_place=True)
        return queryset

然后路由只需要配置一个：

path('places/', PlacesListView.as_view(), name='places-list'),

这样访问/places/返回全部实例，访问/places/?new_only=true就返回新地点实例，灵活性更高。

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