如何判断DRF序列化器正在序列化单个还是多个对象？

有两种实用方式可以实现需求，直接看代码和说明：

方法一：利用序列化器的many属性 #

这是最直接的方式——序列化器初始化时，如果传入many=True（用于处理多个对象），self.many会被设为True；处理单个对象时则为False。

示例代码：

from rest_framework import serializers
from .models import YourModel

class YourModelSerializer(serializers.ModelSerializer):
    custom_field = serializers.SerializerMethodField()

    class Meta:
        model = YourModel
        fields = ['id', 'name', 'custom_field']

    def get_custom_field(self, obj):
        if self.many:
            # 调用多对象对应的API
            return your_bulk_api_call()
        else:
            # 调用单对象对应的API，可传入当前对象ID
            return your_single_api_call(obj.id)

如果是重写to_representation方法，同样可以用self.many判断：

def to_representation(self, instance):
    data = super().to_representation(instance)
    if self.many:
        data['extra_data'] = bulk_api_response()
    else:
        data['extra_data'] = single_api_response(instance.id)
    return data

方法二：检查instance的类型 #

通过判断self.instance是单个模型实例，还是查询集/列表，来区分场景：

from django.db.models.query import QuerySet

def get_custom_field(self, obj):
    # 判断instance是否为查询集或列表
    is_multiple = isinstance(self.instance, (QuerySet, list))
    if is_multiple:
        return your_bulk_api_call()
    else:
        return your_single_api_call(obj.id)

优化建议（避免多对象场景重复调用API） #

如果处理多对象时需要调用批量API，建议在序列化器初始化阶段提前获取批量数据，不要在get_custom_field中为每个对象单独调用API，避免性能浪费：

class YourModelSerializer(serializers.ModelSerializer):
    custom_field = serializers.SerializerMethodField()

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.bulk_result = None
        # 提前获取批量数据
        if self.many:
            obj_ids = [item.id for item in self.instance]
            self.bulk_result = your_bulk_api_call(obj_ids)

    def get_custom_field(self, obj):
        if self.many:
            # 从提前获取的批量数据中取当前对象对应的值
            return self.bulk_result.get(obj.id)
        else:
            return your_single_api_call(obj.id)

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