基于jwt的登录/注册功能实现

# views.py
# 登录视图
class LoginView(ViewSet):# 登录@action(methods=['POST'], detail=False)def login(self, request, *args, **kwargs):ser_obj = ser.UserModelSerializer(data=request.data)if ser_obj.is_valid():token = ser_obj.context['token']username = ser_obj.context['user'].usernameuser = ser_obj.context['user']return response.ApiResponse(token=token, username=username, is_superuser=user.is_superuser)else:return response.ApiResponse(code=0, msg=ser_obj.errors)# 手机号验证@action(methods=['GET'], detail=False)def check_telephone(self, request):telephone = request.query_params.get('telephone')if re.match('^1[3-9][0-9]{9}$', telephone):user = models.UserInfo.objects.filter(telephone=telephone).first()if user:return response.ApiResponse(result=True)else:return response.ApiResponse(result=False)else:return response.ApiResponse(code=0, msg='手机号不合法')

# 发送验证码视图
class SmsView(ViewSet):# 如果项目要上线,最好加上下边的频率限制类,该类没有写^_^# throttle_classes = [throttlings.SmsThrotting]# 发送验证码@action(methods=['GET'], detail=False)def send_sms(self, request):telephone = request.query_params.get('telephone')if re.match('^1[3-9][0-9]{9}$', telephone):user = models.UserInfo.objects.filter(telephone=telephone).first()# print(user.username)if user:print('用户存在')code = send_sms.get_code()if send_sms.send(code, telephone):cache.set('code', code, 60)print(cache.get('code'))return response.ApiResponse(code=100, msg='发送验证码成功')else:return response.ApiResponse(msg='未知错误')else:print('用户不存在')return response.ApiResponse(code=2, msg='手机号不存在')else:return response.ApiResponse(msg='号码不符合规范,傻der')# 验证码登录@action(methods=['POST'], detail=False)def sms_login(self, request):telephone = request.data['telephone']code = request.data['code']if str(code) == str(cache.get('code')):user = models.UserInfo.objects.filter(telephone=telephone).first()if user:# 获取签发payload = jwt_payload_handler(user)# 签发encode成tokentoken = jwt_encode_handler(payload)return response.ApiResponse(token=token, username=user.username, is_superuser=user.is_superuser)else:return response.ApiResponse(code=0, msg='用户不存在')else:return response.ApiResponse(code=2, msg='验证码过期或者不正确')

# ser.py
class UserModelSerializer(ModelSerializer):username = serializers.CharField(max_length=32)class Meta:model = models.UserInfofields = ['username', 'password', 'id']extra_kwargs = {'id': {'read_only': True},'password': {'write_only': True}}def validate(self, attrs):# 验证用户user = self._get_user(attrs)# 签发tokentoken = self._get_token(user)# 将验证后的数据保存下来,给视图使用self.context['token'] = tokenself.context['user'] = userreturn attrsdef _get_user(self, attrs):username = attrs.get('username')password = attrs.get('password')import re# 后台验证账号类型if re.match('^1[3-9][0-9]{9}$', username):user = models.UserInfo.objects.filter(telephone=username).first()elif re.match('^.*@.*$', username):user = models.UserInfo.objects.filter(email=username).first()else:user = models.UserInfo.objects.filter(username=username).first()if user:ret = user.check_password(password)if ret:user.last_login = datetime.datetime.now()user.save()return userelse:raise ValidationError('密码错误')else:raise ValidationError('用户不存在')def _get_token(self, user):# 获取签发payload = jwt_payload_handler(user)# 签发encode成tokentoken = jwt_encode_handler(payload)return tokenclass RegisterSerializer(serializers.ModelSerializer):code = serializers.CharField(max_length=4)class Meta:model = models.UserInfofields = ['telephone', 'code', 'password']extra_kwargs = {'password': {'write_only': True, 'max_length': 11, 'min_length': 3}}def validate(self, attrs):code = attrs.get('code')telephone = attrs.get('telephone')password = attrs.get('password')import reif re.match('^1[3-9][0-9]{9}$', telephone):user = models.UserInfo.objects.filter(telephone=telephone).first()if user:raise ValidationError('用户已经存在')if code == cache.get('code'):return attrselse:raise ValidationError('验证码不正确')else:return ValidationError('手机号不合法')return attrsclass RegisterView(ViewSet):# 注册@action(methods=['POST'], detail=False)def register(self, request):ser_obj = ser.RegisterSerializer(data=request.data)telephone = request.data['telephone']password = request.data['password']if ser_obj.is_valid():user = models.UserInfo.objects.create_user(username=telephone, telephone=telephone, password=password)token = self._get_token(user)return response.ApiResponse(msg='注册成功', token=token, username=user.username)else:return response.ApiResponse(code=0, msg='用户已经存在或者手机号不合法或者验证码不正确')# 注册通过验证码注册@action(methods=['GET'], detail=False)def register_send_sms(self, request):telephone = request.query_params.get('telephone')if re.match('^1[3-9][0-9]{9}$', telephone):user = models.UserInfo.objects.filter(telephone=telephone).first()code = send_sms.get_code()if send_sms.send(code, telephone):# if 1:cache.set('code', code, 60)print(cache.get('code'))return response.ApiResponse(code=100, msg='发送验证码成功')else:return response.ApiResponse(msg='未知错误')else:return response.ApiResponse(msg='号码不符合规范,傻der')def _get_token(self, user):# 获取签发payload = jwt_payload_handler(user)# 签发encode成tokentoken = jwt_encode_handler(payload)return token

# urls.py
from rest_framework.routers import SimpleRouter
router = SimpleRouter()# 注册用户的登录,验证码,注册路由
router.register('',views.LoginView,'login')
router.register('',views.SmsView,'send')
router.register('',views.RegisterView,'register')

基于jwt的登录/注册功能实现相关推荐

Django练习——基于Mysql的登录注册功能界面(django模型、模板、表单简单应用)
目录 1.创建 2.配置Mysql (1)连接数据库驱动 (2)创建表格 (3)创建迁移文件并同步到数据库 3.admin后台 (1)创建超级用户 (2)注册模型login_msg 4.路由与视图 4 ...
java wed登录面代码_JavaWeb实现用户登录注册功能实例代码(基于Servlet+JSP+JavaBean模式)...
下面通过通过图文并茂的方式给大家介绍JavaWeb实现用户登录注册功能实例代码,一起看看吧. 一.Servlet+JSP+JavaBean开发模式(MVC)介绍 Servlet+JSP+JavaBea ...
php注册登录描述,基于PHP实现用户登录注册功能的详细教程
教程前先给大家看看小编的实现成果吧! 图1: 图2: 图3: 教程: 实现这个功能我们需要五个php文件: login.php (登录界面,如图2) 登录登录页面用户名: required=&qu ...
基于android校园订餐APP，简单实现登录注册功能（SharedPreferences）
基于android校园订餐APP,简单实现登录注册功能(SharedPreferences) 欢迎浏览本篇文章界面效果图页面设计代码源代码欢迎浏览本篇文章大家好! 这是小编首次在博客上写的文 ...
spring 3.x 学习笔记_spring mvc、spring jdbc 实现网站的登录注册功能
使用spring mvc.spring jdbc 实现网站的登录注册功能 1. 据业务模型创建model 一般实现序列化 2. 用spring 注解(@Repositor ...
html登陆注册功能实现,实现用户的登录注册功能
在基于Spring Boot框架上,实现用户的登录注册功能, 首先分析前期所需要的规划. 1 实现登录注册前端向后端发起post请求 2用户密码安全性密码是不推荐明文入库的,在后台采取秘钥加加不可 ...
基于Bootstrap的登录注册模板（html+css）
基于Bootstrap的登录&注册模板(html+css) 1.简介 2.功能 2.1 登录 2.2 注册 3.展示 4.下载地址 1.简介基于Bootstrap的登录&注册模板(h ...
Spring boot 搭建个人博客系统（二）——登录注册功能
Spring boot 搭建个人博客系统(二)--登录注册功能一直想用Spring boot 搭建一个属于自己的博客系统,刚好前段时间学习了叶神的牛客项目课受益匪浅,乘热打铁也主要是学习,好让自己熟 ...
java实现用户登录注册功能(用集合框架来实现)
需求:实现用户登录注册功能(用集合框架来实现) 分析: A:需求的类和接口 1.用户类 UserBean 2.用户操作方法接口和实现类 UserDao UserDaoImpl 3.测试类 UserTe ...

基于jwt的登录/注册功能实现

基于jwt的登录/注册功能实现相关推荐

最新文章

热门文章