Django序列化多对象时如何批量调用REST接口获取关联数据

核心实现思路 #

本质是序列化场景下的N+1外部调用优化，和ORM N+1查询的优化思路完全一致：在批量序列化所有对象之前，一次性拉取所有需要的关联数据，再把数据附着到单个对象上供序列化阶段直接读取，完全避免单对象序列化时的重复接口调用。 #

批量逻辑的合理存放位置（按推荐优先级排序） #

1. 自定义Model QuerySet/Manager层（最符合Django规范） #

Django官方推荐「胖模型、瘦视图」的设计原则，所有和Account数据查询、拓展相关的逻辑都可以下沉到模型的自定义QuerySet或Manager中，视图层仅做参数接收、权限校验、结果返回的轻量化操作。
示例代码：

# models.py
from django.db import models

class AccountQuerySet(models.QuerySet):
    def with_favorite_colors(self):
        # 拿到当前查询集所有账号ID
        account_ids = list(self.values_list("id", flat=True))
        # 单次调用批量接口获取颜色映射，假设返回格式为 {account_id: color}
        color_map = batch_get_favorite_color(account_ids)
        # 把颜色属性附加到每个account对象上
        for account in self:
            account.favorite_color = color_map.get(account.id)
        return self

class Account(models.Model):
    name = models.CharField(db_index=True)
    # 绑定自定义QuerySet为管理器
    objects = AccountQuerySet.as_manager()

# views.py 视图层逻辑极简
class AccountViewSet(viewsets.ModelViewSet):
    queryset = Account.objects.all()
    serializer_class = AccountSerializer

    def list(self, request, *args, **kwargs):
        queryset = self.filter_queryset(self.get_queryset()).with_favorite_colors()
        serializer = self.get_serializer(queryset, many=True)
        return Response({"accounts": serializer.data})

# serializers.py 直接读取预加载的属性即可
class AccountSerializer(serializers.ModelSerializer):
    name = serializers.CharField()
    favorite_color = serializers.CharField()

    class Meta:
        model = Account
        fields = ["name", "favorite_color"]

2. 独立服务层（适合中大型项目） #

如果项目涉及大量第三方接口调用、复杂业务规则，不想让Model层承载过多非数据持久化相关的逻辑，可以抽离独立的服务层模块存放这类批量处理逻辑，实现业务逻辑和数据层、视图层的解耦。
示例代码：

# accounts/services.py 独立服务模块
def attach_favorite_colors(accounts):
    account_ids = [a.id for a in accounts]
    color_map = batch_get_favorite_color(account_ids)
    for account in accounts:
        account.favorite_color = color_map.get(account.id)
    return accounts

# views.py 调用服务层方法即可
def list(self, request, *args, **kwargs):
    queryset = self.filter_queryset(self.get_queryset())
    accounts = attach_favorite_colors(queryset)
    serializer = self.get_serializer(accounts, many=True)
    return Response({"accounts": serializer.data})

3. 自定义ListSerializer（适合和序列化强绑定的场景） #

如果批量获取逻辑仅在当前序列化场景下使用，不需要其他业务模块复用，可以重写DRF的ListSerializer的批量序列化钩子，在序列化全量对象前完成批量数据拉取，存入序列化上下文供单个序列化方法读取。
示例代码：

# serializers.py
class AccountListSerializer(serializers.ListSerializer):
    def to_representation(self, data):
        account_ids = [item.id for item in data]
        color_map = batch_get_favorite_color(account_ids)
        # 把批量获取的颜色映射存入上下文
        self.context["color_map"] = color_map
        return super().to_representation(data)

class AccountSerializer(serializers.ModelSerializer):
    name = serializers.CharField()
    favorite_color = serializers.SerializerMethodField()

    class Meta:
        model = Account
        fields = ["name", "favorite_color"]
        # 指定批量序列化使用自定义的ListSerializer
        list_serializer_class = AccountListSerializer

    def get_favorite_color(self, obj):
        # 直接从上下文读取预加载的颜色
        return self.context["color_map"].get(obj.id)

方案优势 #

三种方案都可以实现单次外部接口调用完成全量favorite_color的获取，完全避免N次重复请求，且View层的逻辑始终保持轻量化，不需要承载业务处理代码。 #

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