django rest框架用密码创建用户

Question

使用 django-rest-framework 3 和 django 1.8

我正在尝试使用 django-rest-framework ModelViewSerializer 创建用户。问题是 DRF 使用的默认 objects.create 方法将密码保留为纯文本。

问题是 DRF 序列化器创建方法使用 objects.create querysets/#create 方法而不是使用 objects.create_user 方法。

serializers.py 第 775 行的代码

instance = ModelClass.objects.create(**validated_data)

对此最好的解决方案是什么？我可以重写 serializer.create 方法以使用 objects.user_create 而不是 objects.create 但它感觉不是正确的解决方案。

其余代码：

from django.contrib.auth.models import User
from rest_framework import viewsets

class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = ('username', 'email','password')
        write_only_fields = ('password',)


class UserViewSet(viewsets.ModelViewSet):   
    queryset = User.objects.all()
    serializer = UserSerializer()

Answer 1

您可以在 UserSerializer 中覆盖 create：

class UserSerializer(serializers.ModelSerializer):
    # ....

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

其他解决方案可以在 ViewSet 类中覆盖 perform_create，或者您可以在视图集类中编写自己的 create 方法

class UserViewSet(viewsets.ModelViewSet): 
    def create(self, request, format=None):
        # create user here
        # do not call seriailzer.save()

更新：在@freethebees 评论后，覆盖perform_create 也可以，所以这里是代码sn-p：

class UserViewSet(viewsets.ModelViewSet, mixins.CreateModelMixin): 
    def perform_create(self, serializer):
        # use User.objects.create_user to create user
        pass

注意： 此答案提供 3 个解决方案，选择您认为更适合您的需求和项目生态系统的解决方案

注意 2 我个人更喜欢在UserViewSet（第二个代码sn-p）中覆盖create，因为在那里您可以简单地返回您的自定义Response（例如登录后返回用户配置文件）

Answer 2

除了@aliva 的回答你错过了serializers.Modelserializer.create() 中的功能（这可能会很好地保留，例如处理多对多关系），还有一种方法可以保留它。

使用user.set_password()方法，也可以正确设置密码，如：

class UserSerializer(serializers.ModelSerializer):

    def create(self, validated_data):
        user = super().create(validated_data)
        user.set_password(validated_data['password'])
        user.save()
        return user

这样做的好处是保留了超类的功能，但缺点是额外写入数据库。决定哪种权衡对您更重要:-)。

请参阅set_password 的文档。

Answer 3

在序列化程序中验证密码有更好的选择

from django.contrib.auth.hashers import make_password

class UserSerializer(serializers.ModelSerializer):
    def validate_password(self, value: str) -> str:
        return make_password(value)

Answer 4

支持POST 和PUT/PATCH 的完整示例，无需其他SQL UPDATE 语句。

class MyUserSerializer(serializers.ModelSerializer):
    class Meta:
        model = models.User
        fields = '__all__'

    def create(self, validated_data):
        if "password" in validated_data:
            from django.contrib.auth.hashers import make_password
            validated_data["password"] = make_password(validated_data["password"])
        return super().create(validated_data)

    def update(self, instance, validated_data):
        if "password" in validated_data:
            from django.contrib.auth.hashers import make_password
            validated_data["password"] = make_password(validated_data["password"])
        return super().update(instance, validated_data)