Django使用两个用户个人资料模型注册api

Question

我正在尝试创建用户，其中用户模型具有两个用户配置文件，例如，教师和学生。 在创建用户时，我也想在教师或学生资料中创建用户实例。

这是我的序列化器类：

class StudentSerializer(serializers.ModelSerializer):

    class Meta:
        model = Student
        fields = ('id', 'user_id')


class TeacherSerializer(serializers.ModelSerializer):

    class Meta:
        model = Teacher
        fields = ('id', 'user_id')


class UserSerializer(serializers.ModelSerializer):

    class Meta:
        model = User
        fields = (
            'id',
            'username',
            'password',
            'email',
            'user_type'
        )
        extra_kwargs = {
            'password': {'write_only': True},
        }

    def create(self, validated_data):
        user = User.objects.create_user(**validated_data)
        return user

这是自定义注册视图：

class CreateUserView(CreateAPIView):

    model = User
    permission_classes = [
        permissions.AllowAny # Or anon users can't register
    ]
    serializer_class = UserSerializer

现在，我能够创建User，但是根据user_type参数创建用户个人资料对我来说是一个挑战。

因此，如果user_type是student ，那么我也需要为StudentProfile创建User实例（studentprofile表中的新行）

Answer 1

如果条件可以，您可以添加简单吗？：

def create(self, validated_data):
    user = User.objects.create_user(**validated_data)
    if validated_data('user_type') == 'student':
        Student.objects.create(user_id=user.id)
    else:
        Teacher.objects.create(user_id=user.id)
    return user