多登陸方式與自動簽發token
views.py
1.繼承Viewset,方法里可以使用自定義login,更直觀。需要路由直接配置請方式
2. 序列化是直接對request數據處理,并從對象中獲取token
3.context可以儲存自定義數據
# 多登陸方式,自動簽發tokenfrom rest_framework.viewsets import ViewSet
from app02 import ser
class Login2View(ViewSet):# 這是登陸接口# 不用post寫 直接logindef login(self,request,*args,**kwargs):# 1 需要序列化的類login_ser=ser.LoginModelSerializer(data=request.data,context={'request':request})# 2 生成序列化對象# 3 調用序列化對象的is_validadlogin_ser.is_valid(raise_exception=True)token=login_ser.context.get('token')# 4 returnreturn Response({'status':100,'msg':'登陸成功','token':token,'username':login_ser.context.get('username')})
urls.py
指定請求方式和對應方法
urlpatterns = [path('login2view/', views.Login2View.as_view({'post':'login'})),
]
ser.py
1.因為請求數據可能是用戶名,郵箱,電話,所以這里使用自定義name來序列化該數據,然后進行判斷
2.name需要自定義的,用serializers.CharField。如果不聲明,會使用內置校驗,然后再進如?def validate會出錯。
3.fields 是請求數據的字段
4.自定義的校驗在“def validate”里面
5.使用jwt方法得到token,
payload=jwt_payload_handler(user) # 把user 傳入,得到payload
token=jwt_encode_handler(payload) # 把payload傳入,得到token
6.使用內置的self.context來儲存自定義數據,返回給視圖類。
from rest_framework import serializers
from api import models
# 正則
import re
from rest_framework.exceptions import ValidationError
from rest_framework_jwt.utils import jwt_encode_handler,jwt_payload_handlerclass LoginModelSerializer(serializers.ModelSerializer):name=serializers.CharField(max_length=32)class Meta:model=models.Userfields=['name','password']def validate(self, attrs):print(self.context)# 這里寫邏輯username=attrs.get('name') # 用戶名有三種方式password=attrs.get('password')# 通過判斷,username數據,查詢對應字段# 正則匹配,如果手機號if re.match('^1[3-9][0-9]{9}$',username):user=models.User.objects.filter(mobile=username).first()elif re.match('^.+@.+$',username):user=models.User.objects.filter(email=username).first()else:user=models.User.objects.filter(username=username).first()if user: # 存在用戶# 校驗密碼,因為是密文,要用check_passwordif user.check_password(password):payload=jwt_payload_handler(user) # 把user 傳入,得到payloadtoken=jwt_encode_handler(payload) # 把payload傳入,得到tokenself.context['token']=tokenself.context['username']=user.usernamereturn attrselse:raise ValidationError('密碼錯誤')else:raise ValidationError('用戶不存在')
配置過期時間
settings.py
import datetime
JWT_AUTH={'JWT_EXPIRATION_DELTA': datetime.timedelta(days=7),
}