在Django REST Framework中同时兼容嵌套序列化器与PrimaryKeyRelatedField

完全可行，核心思路是自定义一个序列化字段，根据输入数据的类型自动判断是关联现有对象还是创建新对象。以下是具体实现方案：

1. 自定义通用序列化字段 #

这个字段会处理两种输入场景：整数/字符串形式的主键，以及嵌套对象的字典数据。

from rest_framework import serializers

class PrimaryKeyOrNestedSerializerField(serializers.RelatedField):
    def __init__(self, serializer_class, **kwargs):
        self.serializer_class = serializer_class
        super().__init__(**kwargs)

    def to_internal_value(self, data):
        # 处理主键关联：输入为整数或字符串主键
        if isinstance(data, (int, str)):
            try:
                return self.queryset.get(pk=data)
            except self.queryset.model.DoesNotExist:
                raise serializers.ValidationError(f"{self.queryset.model.__name__} 对象不存在")
        # 处理嵌套创建：输入为完整对象字段字典
        elif isinstance(data, dict):
            serializer = self.serializer_class(data=data)
            serializer.is_valid(raise_exception=True)
            return serializer.save()
        # 非法输入格式
        raise serializers.ValidationError("输入格式无效，请提供对象主键或完整字段字典")

    def to_representation(self, value):
        # 输出时统一用嵌套序列化器格式（也可改为输出主键，按需调整）
        return self.serializer_class(value).data

2. 定义业务序列化器 #

将自定义字段应用到你的关联字段上：

# 先定义关联模型的序列化器
class LocationSerializer(serializers.ModelSerializer):
    class Meta:
        model = Location
        fields = ['id', 'name', 'zip']

# 主模型序列化器使用自定义字段
class AddressSerializer(serializers.ModelSerializer):
    location = PrimaryKeyOrNestedSerializerField(
        queryset=Location.objects.all(),
        serializer_class=LocationSerializer
    )

    class Meta:
        model = Address
        fields = ['id', 'name', 'street', 'location', 'country']

3. 使用示例 #

• 关联现有Location对象：传入主键

{
    "name": "家庭地址",
    "street": "XX街道XX号",
    "location": 1,
    "country": "中国"
}

• 创建新Location并关联：传入嵌套字段

{
    "name": "公司地址",
    "street": "YY大道YY号",
    "location": {
        "name": "科技园",
        "zip": "518000"
    },
    "country": "中国"
}

扩展：支持更新现有关联对象 #

如果需要支持传入字典更新已有对象（而非仅创建新对象），可以修改to_internal_value方法：

elif isinstance(data, dict):
    if 'id' in data:
        # 存在id则尝试更新现有对象
        try:
            instance = self.queryset.get(pk=data['id'])
            serializer = self.serializer_class(instance, data=data, partial=True)
        except self.queryset.model.DoesNotExist:
            raise serializers.ValidationError(f"{self.queryset.model.__name__} 对象不存在")
    else:
        # 无id则创建新对象
        serializer = self.serializer_class(data=data)
    serializer.is_valid(raise_exception=True)
    return serializer.save()

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