django註冊模塊邏輯代碼()

view.py
from rest_framework.response import Response
from rest_framework.views import APIView
from rest_framework.generics import CreateAPIView
from .models import User
from .serializers import UserCreateSerializer

list===>[user,user,…]

retrieve===>pk===>user

class UsernameCountView(APIView):
def get(self, request, username):
# 查詢用戶名的個數
count = User.objects.filter(username=username).count()
# 響應
return Response({
‘username’: username,
‘count’: count
})

class MobileCountView(APIView):
def get(self, request, mobile):
# 查詢手機號的個數
count = User.objects.filter(mobile=mobile).count()
# 響應
return Response({
‘mobile’: mobile,
‘count’: count
})

class UserCreateView(CreateAPIView):
# def post(self,request):
#註冊用戶==>創建用戶
# queryset = 當前進行創建操作，不需要查詢
serializer_class = UserCreateSerializer

serializers.py
from rest_framework import serializers
from django_redis import get_redis_connection
import re
from rest_framework_jwt.settings import api_settings
from .models import User

class UserCreateSerializer(serializers.Serializer):
# 定義屬性
id=serializers.IntegerField(read_only=True)#不接收客戶端的數據，只向客戶端輸出
token=serializers.CharField(read_only=True)
username = serializers.CharField(
min_length=5,
max_length=20,
error_messages={
‘min_length’: ‘用戶名包含5-20個字符’,
‘max_length’: ‘用戶名包含5-20個字符’,
}
)
password = serializers.CharField(
min_length=8,
max_length=20,
error_messages={
‘min_length’: ‘密碼包含8-20個字符’,
‘max_length’: ‘密碼包含8-20個字符’,
},
write_only=True#只接收客戶端的數據，不向客戶端輸出數據
)
password2 = serializers.CharField(
min_length=8,
max_length=20,
error_messages={
‘min_length’: ‘密碼包含8-20個字符’,
‘max_length’: ‘密碼包含8-20個字符’,
},
write_only=True
)
sms_code = serializers.IntegerField(write_only=True)
mobile = serializers.CharField()
allow = serializers.BooleanField(write_only=True)

# 驗證
def validate_username(self, value):用戶名已經
    # 驗證用戶名是否重複
    count = User.objects.filter(username=value).count()
    if count > 0:
        raise serializers.ValidationError('用戶名已經存在')
    return value

def validate_mobile(self, value):
    # 驗證手機號格式
    if not re.match(r'^1[3-9]\d{9}$', value):
        raise serializers.ValidationError('手機號格式錯誤')
    # 驗證手機號是否重複
    count = User.objects.filter(mobile=value).count()
    if count > 0:
        raise serializers.ValidationError('手機號已經存在')
    return value

def validate_allow(self, value):
    # 是否同意協議
    if not value:
        raise serializers.ValidationError('請先閱讀協議並同意')
    return value

# 2.多屬性判斷
def validate(self, attrs):
    # 判斷兩個密碼是否一致
    password = attrs.get('password')
    password2 = attrs.get('password2')
    if password != password2:
        raise serializers.ValidationError('兩個密碼不一致')

    # 短信驗證碼是否正確
    # 1.獲取請求報文中的短信驗證碼、手機號
    sms_code_request = attrs.get('sms_code')
    mobile = attrs.get('mobile')
    # 2.獲取redis中的短信驗證碼
    redis_cli = get_redis_connection('sms_code')
    sms_code_redis = redis_cli.get('sms_code_' + mobile)  # None
    # 3判斷是否過期
    if sms_code_redis is None:
        raise serializers.ValidationError('短信驗證碼已經過期')
    # 4.強制立即過期
    redis_cli.delete('sms_code_' + mobile)
    # 5.判斷兩個驗證碼是否相等
    if int(sms_code_request) != int(sms_code_redis):
        raise serializers.ValidationError('短信驗證碼錯誤')

    return attrs

# 保存if
def create(self, validated_data):
    user = User()
    user.username = validated_data.get('username') #驗證成功，可以通過序列化器對象的validated_data屬性獲取數據
    user.set_password(validated_data.get('password'))
    user.mobile = validated_data.get('mobile')
    user.save()

    #需要生成token
    jwt_payload_handler=api_settings.JWT_PAYLOAD_HANDLER
    jwt_encode_handler=api_settings.JWT_ENCODE_HANDLER
    payload=jwt_payload_handler(user)
    token=jwt_encode_handler(payload)#header.payload.signature

    #將token輸出到客戶端
    user.token=token

    return user