如何在Django单个视图集的get_queryset中返回两个不同模型的查询集？

我来帮你解决这个问题！核心思路是只执行一次产品过滤查询，然后基于过滤后的产品批量获取关联图片，最后在视图中分别序列化两个数据集并返回，完美避免重复查询和冗余字段。

步骤1：定义专用的序列化器 #

先为产品和图片分别创建只包含所需字段的序列化器，彻底杜绝冗余内容：

from rest_framework import serializers
from .models import Product, Image

class ProductListSerializer(serializers.ModelSerializer):
    class Meta:
        model = Product
        fields = ('id', 'name', 'description')  # 只返回业务需要的三个字段

class ImageListSerializer(serializers.ModelSerializer):
    product_id = serializers.PrimaryKeyRelatedField(source='product.id', read_only=True)

    class Meta:
        model = Image
        fields = ('id', 'image', 'product_id')  # 明确返回图片关联的产品ID

步骤2：重写视图集的list方法 #

不要局限于get_queryset()返回单个模型数据，直接重写list()方法来完成一次过滤+两次序列化的逻辑：

from rest_framework import viewsets, Response
from .models import Product, Image
from .serializers import ProductListSerializer, ImageListSerializer

class ProductImageViewSet(viewsets.ViewSet):
    def list(self, request):
        # 1. 仅执行一次产品关键词过滤
        # 建议从请求参数获取关键词，更灵活：request.query_params.get('keyword', 'Fanta')
        keyword = "Fanta"
        filtered_products = Product.objects.filter(name__contains=keyword)

        # 2. 基于过滤后的产品ID批量查询图片（一次数据库操作）
        product_ids = filtered_products.values_list('id', flat=True)
        related_images = Image.objects.filter(product_id__in=product_ids)

        # 3. 分别序列化两个数据集
        product_data = ProductListSerializer(filtered_products, many=True).data
        image_data = ImageListSerializer(related_images, many=True).data

        # 4. 返回包含两个独立列表的JSON响应
        return Response({
            'products': product_data,
            'images': image_data
        })

如果你的视图原本是ModelViewSet，也可以直接重写list方法，逻辑完全一致：

class ProductImageViewSet(viewsets.ModelViewSet):
    queryset = Product.objects.all()
    serializer_class = ProductListSerializer

    def list(self, request, *args, **kwargs):
        # 从URL参数接收关键词，比如 /api/products?keyword=Fanta
        keyword = request.query_params.get('keyword', 'Fanta')
        filtered_products = self.get_queryset().filter(name__contains=keyword)

        product_ids = filtered_products.values_list('id', flat=True)
        related_images = Image.objects.filter(product_id__in=product_ids)

        return Response({
            'products': ProductListSerializer(filtered_products, many=True).data,
            'images': ImageListSerializer(related_images, many=True).data
        })

方案优势 #

• 单次过滤，高效查询：产品关键词过滤仅执行一次，图片通过product_id__in批量获取，避免了重复查询的性能损耗
• 字段精准无冗余：两个序列化器各自只返回所需字段，不会出现重复的product_description
• 结构清晰符合需求：客户端直接拿到products和images两个独立数组，完全匹配你要的JSON格式

可选优化：提前预加载关联（减少数据库查询） #

如果想进一步优化数据库查询次数，可以用prefetch_related在查询产品时就预加载关联图片，后续提取图片数据时无需再查数据库：

filtered_products = Product.objects.filter(name__contains=keyword).prefetch_related('related_product')
# 提取图片数据
related_images = [img for product in filtered_products for img in product.related_product.all()]

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