Django序列化器能否关联已有Customer或创建新Customer对象？

示例请求输入 #

{
    "abc_obj": {
        "comments": "示例备注",
        "ref": "REF001",
        "customer": 1,
        "address": {
            "street": "示例街道",
            "country": "示例国家"
        }
    }
}

当前序列化器代码 #

class ABCObjSerializer(ModelSerializer[ABCObj]):
    address: AddressSerializer = AddressSerializer(required=False)
    customer: CustomerSerializer = CustomerSerializer(required=False)

    class Meta:
        model = ABCObj
        fields = "__all__"
        extra_kwargs = {
            "created_by": {"read_only": True},
        }

当前视图代码 #

class ABCObjViewSet(ModelViewSet):
    queryset = (
        ABCObj.objects.all()
        .select_related("customer", "delivery_address")
    )

    def get_serializer_class(self) -> Type[BaseSerializer[_MT_co]]:
        if hasattr(self, "action"):
            if self.action == "list":
                return SalesOrderListSerializer
            else:
                return ABCObjSerializer

需求描述 #

现有上述ABCObjSerializer，能否实现以下功能：当请求中customer字段传入ID（如1）时关联已有ID为1的Customer对象；当请求中传入Customer的完整嵌套数据（类似示例中address的结构）时，自动创建新的Customer对象，支持用户选择复用已有记录或创建新记录？

可以实现，核心是自定义customer字段的反序列化逻辑，同时兼容ID关联与嵌套数据创建两种场景，具体实现如下：

1. 自定义兼容字段 #

创建一个继承自serializers.Field的自定义字段，处理两种输入格式：

from rest_framework import serializers
from .models import Customer

class CustomerRelatedOrCreateField(serializers.Field):
    def to_representation(self, value):
        # 序列化时返回完整的Customer数据，复用现有CustomerSerializer
        return CustomerSerializer(value).data

    def to_internal_value(self, data):
        # 反序列化分支处理
        if isinstance(data, int):
            # 传入ID：验证并返回已有Customer对象
            try:
                return Customer.objects.get(id=data)
            except Customer.DoesNotExist:
                raise serializers.ValidationError(f"ID为{data}的Customer不存在")
        elif isinstance(data, dict):
            # 传入嵌套数据：验证并创建新Customer
            serializer = CustomerSerializer(data=data)
            serializer.is_valid(raise_exception=True)
            return serializer.save()
        else:
            raise serializers.ValidationError("customer字段格式无效：需传入整数ID或Customer嵌套数据字典")

2. 修改ABCObjSerializer #

用自定义字段替换原有的customer字段定义：

class ABCObjSerializer(ModelSerializer[ABCObj]):
    address: AddressSerializer = AddressSerializer(required=False)
    customer = CustomerRelatedOrCreateField(required=False)  # 替换原字段

    class Meta:
        model = ABCObj
        fields = "__all__"
        extra_kwargs = {
            "created_by": {"read_only": True},
        }

功能说明 #

• 关联已有Customer：当请求传入"customer": 1时，字段会自动查找ID为1的Customer对象，验证存在后关联到当前ABCObj实例。
• 创建新Customer：当请求传入"customer": {"name": "新客户", "phone": "123456"}（完整的Customer数据结构）时，会通过CustomerSerializer验证数据有效性，创建新的Customer对象并关联到ABCObj。
• 序列化输出：返回响应时，仍会输出Customer的完整嵌套数据，保持原有输出格式一致。

注意事项 #

• 确保CustomerSerializer已正确配置，支持创建对象（即其Meta类指定了对应的model，且字段设置允许写入）。
• 若需支持更新场景：更新时传入新ID会切换关联的Customer，传入嵌套数据会创建新Customer并替换原有关联，当前逻辑已覆盖该场景。

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