Django REST Framework中嵌套数据注册的优化实践

from rest_framework import serializers
from rest_framework.validators import UniqueValidator
from django.contrib.auth.password_validation import validate_password as django_validate_password
from django.db import transaction

# 假设 CustomUser 和 Rider 模型已定义
# from .models import CustomUser, Rider

class UserSerializer(serializers.ModelSerializer):
    """
    用于RiderSerializer内部展示用户信息的嵌套序列化器（只读）。
    """
    class Meta:
        model = CustomUser
        fields = (
            "email",
            "first_name",
            "last_name",
            "phone_number",
        )

class RiderSerializer(serializers.ModelSerializer):
    """
    用于骑手注册的统一序列化器，处理用户和骑手相关的所有输入字段。
    """
    # 用于输出的嵌套用户数据（只读）
    user = UserSerializer(read_only=True)

    # 用户相关输入字段 (write_only=True 表示这些字段只用于输入，不包含在输出中)
    email = serializers.EmailField(
        write_only=True,
        validators=[UniqueValidator(queryset=CustomUser.objects.all(), message="此邮箱已被注册。")]
    )
    first_name = serializers.CharField(write_only=True, required=True)
    last_name = serializers.CharField(write_only=True, required=True)
    phone_number = serializers.CharField(write_only=True, required=True)
    password = serializers.CharField(write_only=True, required=True, style={'input_type': 'password'})
    confirm_password = serializers.CharField(write_only=True, required=True, style={'input_type': 'password'})

    # 骑手相关输入字段
    vehicle_registration_number = serializers.CharField(
        max_length=20,
        validators=[UniqueValidator(queryset=Rider.objects.all(), message="此车牌号已被注册。")]
    )
    min_capacity = serializers.IntegerField(required=False, allow_null=True)
    max_capacity = serializers.IntegerField(required=False, allow_null=True)
    fragile_item_allowed = serializers.BooleanField(default=True)
    charge_per_mile = serializers.DecimalField(
        max_digits=6, decimal_places=2, required=False, allow_null=True
    )
    # vehicle_type 字段如果需要用户输入，也应在此定义，否则使用模型默认值
    # vehicle_type = serializers.CharField(max_length=50, required=False, default="TWO_WHEELER")


    class Meta:
        model = Rider
        fields = (
            'user', 'email', 'first_name', 'last_name', 'phone_number',
            'password', 'confirm_password', 'vehicle_type', 'vehicle_registration_number',
            'is_available', 'min_capacity', 'max_capacity', 'fragile_item_allowed',
            'ratings', 'charge_per_mile',
        )
        # 确保 vehicle_type, is_available, ratings 等字段如果不需要用户输入，
        # 且希望使用模型默认值，则可以在这里或模型中设置好。
        # 如果需要用户输入，则需在上面定义。

    def validate(self, data):
        """
        执行密码匹配和Django内置密码验证。
        """
        password = data.get('password')
        confirm_password = data.pop('confirm_password') # 移除 confirm_password，因为它不需要保存到模型

        if password != confirm_password:
            raise serializers.ValidationError({"confirm_password": "两次输入的密码不匹配。"})

        try:
            # 使用Django内置的密码验证器
            django_validate_password(password=password)
        except Exception as e: # 捕获所有验证错误
            raise serializers.ValidationError({"password": list(e.messages)}) # 将错误信息转换为列表

        return data

    @transaction.atomic
    def create(self, validated_data):
        """
        创建 CustomUser 和 Rider 对象。
        """
        # 从 validated_data 中分离出 Rider 相关的字段
        rider_data = {
            'vehicle_registration_number': validated_data.pop('vehicle_registration_number', ''),
            'min_capacity': validated_data.pop('min_capacity', None),
            'max_capacity': validated_data.pop('max_capacity', None),
            'fragile_item_allowed': validated_data.pop('fragile_item_allowed', True),
            'charge_per_mile': validated_data.pop('charge_per_mile', None),
            # 如果 vehicle_type 也来自输入，需要在这里pop
            'vehicle_type': validated_data.pop('vehicle_type', 'TWO_WHEELER'), # 使用模型默认值或pop输入值
            'is_available': validated_data.pop('is_available', True), # 确保默认值被正确处理
            'ratings': validated_data.pop('ratings', None),
        }

        # 剩余的 validated_data 将是 CustomUser 相关的字段
        user = CustomUser.objects.create_user(**validated_data)

        # 创建 Rider 对象并关联到新创建的用户
        rider = Rider.objects.create(user=user, **rider_data)

        # 可以在这里添加发送验证邮件等逻辑
        # send_verification_email(user, "registration")

        return rider

from rest_framework import generics, status
from rest_framework.response import Response
# from .serializers import RiderSerializer # 假设 RiderSerializer 在同一目录

class RiderRegistrationView(generics.CreateAPIView):
    """
    骑手注册视图，使用 RiderSerializer 处理所有注册逻辑。
    """
    serializer_class = RiderSerializer

    def post(self, request, *args, **kwargs):
        """
        处理骑手注册请求。
        """
        serializer = self.serializer_class(data=request.data)
        # is_valid(raise_exception=True) 会在验证失败时自动抛出异常，并返回400响应
        serializer.is_valid(raise_exception=True)
        # serializer.save() 会调用序列化器中的 create 或 update 方法
        rider = serializer.save() # save() 方法会返回创建或更新的对象

        # 构造成功的响应数据
        response_data = {
            "message": "骑手注册成功",
            "data": serializer.data, # serializer.data 会包含嵌套的 user 信息
        }
        return Response(response_data, status=status.HTTP_201_CREATED)