Django drf使用Django自帶的用戶系統(tǒng)的注冊功能

更新時間：2023年02月13日 11:18:16 作者：Alan and fis

本文主要介紹了Django drf使用Django自帶的用戶系統(tǒng)的注冊功能，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

1.閱讀Django自帶用戶系統(tǒng)源碼

1.1 閱讀User類源碼

系統(tǒng)自帶的用戶系統(tǒng)的models文件的位置\site-packages\django\contrib\auth\models.py,打開這個文件發(fā)現(xiàn)User類繼承的是一個AbstractUser類,因此我們想要使用系統(tǒng)自帶的用戶系統(tǒng),只要在models中重新寫一個User類,也繼承AbstractUser就可以了.

jpg

1.2 閱讀AbstractUser類

閱讀AbstractUser類,發(fā)現(xiàn)了其中的對username字段做的幾個處理,第一個就是他添加了一個驗證器validators,第二個增加了error_messages,當在我們往數據庫中添加用戶信息的時候,系統(tǒng)就會自動驗證是否符合要求,如果不符合要求,就會返回一個字典,將對應字段的錯誤返回,這就是我想到的與其他博主所做的不同之處.

jpg

2.創(chuàng)建自己的User類

我想我的用戶類中有mobile,username,email,password四個字段,我的需求:

mobile,username,email都是不能重復

驗證這幾個字段的格式

返回的errormessage為中文
因此我就想到了模仿系統(tǒng)自帶的AbstractUser的寫法:

2.1 創(chuàng)建驗證器

這個驗證器是其中主要是驗證電話號碼和郵箱的格式正誤,因為其他的類也可能需要用到驗證器,因此我就在項目的utils目錄下創(chuàng)建了一個validator.py文件,用于專門存取驗證器.
項目根目錄\utils\validator.py代碼:

# 驗證電話號碼是否有誤
import re
from django.core.exceptions import ValidationError

# 對手機格式進行驗證
class UnicodMobileValidator:
    def __call__(self, value):
        if not re.match(r"^1[3-9]\d{9}$", value):
             raise ValidationError("對不起,手機格式有誤!")

# 對郵箱格式進行驗證
class UnicodEmailValidator:
    def __call__(self, value):
        if not re.match(r"^[a-zA-Z0-9_.-]+@[a-zA-Z0-9-]+(\.[a-zA-Z0-9-]+)*\.[a-zA-Z0-9]{2,6}$", value):
             raise ValidationError("對不起,郵箱格式有誤!")

2.2 創(chuàng)建User類

這里我繼承了AbstractUser類,并且重寫了其中的方法和屬性.
user\models.py代碼:

from datetime import timezone

from django.contrib.auth.validators import UnicodeUsernameValidator
from django.db import models
from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin, AbstractUser

from qianxingtong_main.utils.validator import UnicodMobileValidator, UnicodEmailValidator


class User(AbstractUser):
    username_validator = UnicodeUsernameValidator()

    username = models.CharField(
        ("username"),
        max_length=150,
        unique=True,
        help_text=(
            "Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only."
        ),
        validators=[username_validator],
        error_messages={
            "unique": ("該用戶名已經存在"),
        },
    )

    email = models.EmailField(("email address"), unique=True, validators=[UnicodEmailValidator()],error_messages={"unique": ("郵箱已經存在"),})
    mobile = models.CharField(max_length=15, unique=True, validators=[UnicodMobileValidator()],
                              error_messages={"unique": ("電話號碼已經存在")}, verbose_name="手機號碼")

    class Meta:
        db_table = "drf_user"
        verbose_name = "用戶信息"
        verbose_name_plural = verbose_name

這里在遷移數據的時候可能會出現(xiàn)錯誤,其中錯誤的解決方法參考我寫的筆記:http://www.fzitv.net/article/275133.htm 的第三步

3.創(chuàng)建序列化類

user\serializers代碼:

from django.contrib.auth.hashers import make_password
from rest_framework import serializers
from user.models import User
class UserSerializer(serializers.ModelSerializer):
    rpassword = serializers.CharField(required=True, write_only=True, help_text="確認密碼")
    token = serializers.CharField(max_length=1024, read_only=True, help_text="token認證字符串")
    class Meta:
        model = User
        fields = ["id", "username", "email", "mobile", "password", "token", "rpassword"]
        # write_only:只寫入數據庫,
        extra_kwargs = {
            "mobile": {
                "write_only": True,
            },
            "email": {
                "write_only": True,
            },
            "id": {
                "read_only": True,
            },

        }

    def validate(self, attrs):
        """校驗信息"""
        password = attrs.get("password")
        rpassword = attrs.get("rpassword")
        if password != rpassword:
            # 驗證手機號碼是否已經被注冊
            raise serializers.ValidationError("對不起,確認密碼與密碼不一致!")
        return attrs

    def create(self, validated_data):
        """保存用戶信息"""
        mobile = validated_data.get("mobile")
        username = validated_data.get("username")
        email = validated_data.get("email")

        validated_data.pop("rpassword")  # 移除不需要的數據
        # 對密碼進行加密
        raw_password = validated_data.get("password")
        hash_password = make_password(raw_password)
        # 調用序列化器提供的create方法
        user = User.objects.create(
            mobile=mobile,
            username=username,
            password=hash_password,
            email=email
        )
        return user

這個代碼就是我與其他的博主不同的地方,我看網上的up主就是把驗證的代碼寫在validate中,但是這里存在的一個問題就是所有的錯誤都是統(tǒng)一一起打包返回,無法在前端中做對應字段的錯誤提示.

4.創(chuàng)建views類

user\views代碼:

from rest_framework.generics import CreateAPIView
from user.models import User
from user.serializers import UserSerializer

# Create your views here.
class UserViewSets(CreateAPIView):
    queryset = User.objects.all()
    serializer_class = UserSerializer