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

怪兽AI数字人

数字人短视频创作，数字人直播，实时驱动数字人

44

查看详情

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)