django注册模块逻辑代码()

最新推荐文章于 2023-02-14 15:26:13 发布

苹果一分为二

最新推荐文章于 2023-02-14 15:26:13 发布

阅读量407

点赞数

分类专栏： django drf 文章标签： python

django 同时被 2 个专栏收录

2 篇文章

订阅专栏

drf

1 篇文章

订阅专栏

本文详细介绍了使用Django REST framework实现用户注册、用户名和手机号验证的过程。通过APIView和CreateAPIView，结合自定义序列化器，确保了数据的有效性和唯一性。

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