Django REST：如何优雅实现注册用户时同步创建关联公司的接口？

1. 先定义基础实体序列化器 #

复用已有字段逻辑，避免重复编码：

from rest_framework import serializers
from django.contrib.auth.models import User
from .models import Company

class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = ['username', 'email', 'password']
        extra_kwargs = {'password': {'write_only': True}}

class CompanySerializer(serializers.ModelSerializer):
    class Meta:
        model = Company
        fields = ['name', 'address', 'industry']

2. 编写注册专用复合序列化器 #

整合两个实体的输入字段，重写create方法完成双实体创建与关联：

class RegistrationSerializer(serializers.Serializer):
    # 显式声明所有输入字段，确保Swagger和可浏览API能识别
    username = serializers.CharField(max_length=150)
    email = serializers.EmailField()
    password = serializers.CharField(write_only=True)
    company_name = serializers.CharField(max_length=255, source='company.name')
    company_address = serializers.CharField(max_length=255, source='company.address')
    company_industry = serializers.CharField(max_length=100, source='company.industry')

    def create(self, validated_data):
        # 拆分用户与公司数据
        company_data = validated_data.pop('company')
        # 创建用户（用create_user处理密码哈希）
        user = User.objects.create_user(**validated_data)
        # 创建关联的公司实体
        Company.objects.create(user=user, **company_data)
        return user

    def to_representation(self, instance):
        # 自定义返回格式，同时返回用户和公司信息
        user_data = UserSerializer(instance).data
        user_data['company'] = CompanySerializer(instance.company).data
        return user_data

3. 配置视图使用该序列化器 #

from rest_framework import generics

class RegistrationView(generics.CreateAPIView):
    serializer_class = RegistrationSerializer

嵌套序列化器替代方案 #

如果偏好结构化的嵌套请求格式，可使用嵌套序列化器写法：

class RegistrationNestedSerializer(serializers.Serializer):
    user = UserSerializer()
    company = CompanySerializer()

    def create(self, validated_data):
        user_data = validated_data.pop('user')
        company_data = validated_data.pop('company')
        user = User.objects.create_user(**user_data)
        Company.objects.create(user=user, **company_data)
        return {
            'user': user,
            'company': user.company
        }

这种写法会在可浏览API中生成嵌套表单，Swagger也能正确识别嵌套字段，但需要前端传递嵌套结构的JSON数据。

方案优势 #

• 解决字段显示问题：所有输入字段均显式声明，DRF可浏览API和Django Spectacular能正常识别并展示。
• 消除代码冗余：复用基础序列化器的字段验证与逻辑，避免重复定义。
• 逻辑集中：双实体创建逻辑封装在单个序列化器中，便于维护和修改。

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