django通过使用jwt模块实现状态保持

时间:2023-03-10 07:35:54
django通过使用jwt模块实现状态保持

第一步:安装jwt

pip install djangorestframework-jwt

第二步:settings/dev的配置文件配置

REST_FRAMEWORK = {
# 认证配置
'DEFAULT_AUTHENTICATION_CLASSES': (
'rest_framework_jwt.authentication.JSONWebTokenAuthentication',
'rest_framework.authentication.SessionAuthentication',
'rest_framework.authentication.BasicAuthentication',
),
}

解释:定义drf配置全局默认的认证方案

第三步:

JWT_AUTH = {
# 重新定义jwt认证成功后返回的数据
'JWT_RESPONSE_PAYLOAD_HANDLER':'users.utils.jwt_response_payload_handler',
'JWT_EXPIRATION_DELTA': datetime.timedelta(days=1), #定义token过期时间为一天
}

解释:第一个参数,重新定义jwt认证成功后返回的数据(jwt默认只返回token,我们在users.utils 下定义该类认证成功后 增加返回字段信息)

def jwt_response_payload_handler(token, user=None, request=None):
"""
自定义jwt认证成功返回数据
"""
return {
'token': token,
'user_id': user.id,
'username': user.username
}

第四步:因为jwt 只对用户名进行认证,我们重写认证方式,增加手机号认证,上代码:

setting.py继续配置:

#在配置文件中告知Django使用我们自定义的认证后端
AUTHENTICATION_BACKENDS = [
'users.utils.UsernameMobileAuthenticate',
]

user.utils下自定义类重写认证方式:

from django.contrib.auth.backends import ModelBackend
from .models import User
from django.db.models import Q
class UsernameMobileAuthenticate(ModelBackend):
def authenticate(self, request, username=None, password=None, **kwargs):
try:
user = User.objects.get(Q(username=username)|Q(mobile=username))
if user.check_password(password):
return user
except Exception as e:
return None

第五步:登录时post请求url设置:

from rest_framework_jwt.views import obtain_jwt_token
urlpatterns = [
url(r'^login/$',obtain_jwt_token),
  ]

是不是很厉害,就这样我们就可以完成登录校验


下面上个例子,注册成功后后台生成token 并返回前端,用户直接登录即可

注册时 创建用户序列化器增加字段:


from rest_framework_jwt.settings import api_settings     #导入该模块
class CreateUserSerializer(serializers.ModelSerializer):
token = serializers.CharField(label='登录状态token', read_only=True) # 增加token字段
   username = ...
   password = ... 
''''''
user = super(CreateUserSerializer,self).create(validated_data)
user.set_password(validated_data['password'])
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) # 将token保存到user对象中,随着返回值返回给前端
user.token = token
return user

注:以上代码是获取token 生成token

我们再来看看前端接收,

LocalStorage.token=response.data.token
sessionStorage.token=response.data.token #我们将后台返回的token 保存在本地或者session即可:

然后我们请求后台是请求体携带即可:

django通过使用jwt模块实现状态保持

这样当我们登录成功后 ,我们后台 request,user 就可以输出当前用户是谁

这些都是jwt  为我们做的很多事情。