在使用Django做前端后端项目时,登陆认证方法往往使用的是jwt_token,但是想自定义登陆成功和失败的返回体。
1.当用户名和密码正确就会生成token,返回response是调用的是JWT_RESPONSE_PAYLOAD_HANDLER
,如果想自定义返回体就需要重写jwt_response_payload_handler
函数。
- 在项目目录下settings.py中,添加
JWT_RESPONSE_PAYLOAD_HANDLER
。
JWT_AUTH = {
'JWT_EXPIRATION_DELTA': datetime.timedelta(seconds=6000),
'JWT_RESPONSE_PAYLOAD_HANDLER': 'Users.views.jwt_response_payload_handler',
}
- 自定义一个视图编辑
jwt_response_payload_handler
。
def jwt_response_payload_handler(token, user=None, request=None):
"""
设置jwt登陆返回的格式
:param token:
:param user:
:param request:
:return:
"""
return {
"msg": "success",
"status": 200,
"data": [{
'token': token,
'username': user.username
}]
}
- 现在的效果如下:
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190905142319406.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI5MTc1NjA1,size_16,color_FFFFFF,t_70)
2. Django中jwt错误返回的是{"non_field_errors":["无法使用提供的认证信息登录。"]}
,当用户名或密码错误时,是不会去调用jwt_response_payload_handler
,那么失败时调用的是什么函数了,JWT_RESPONSE_PAYLOAD_ERROR_HANDLER
,但是发现这个方法并没有合到master分支,那么我们就需要手动给他合并进去。
- 我们打开
/usr/local/lib/python3.6/site-packages/rest_framework_jwt/settings.py
在IMPORT_STRINGS
中添加JWT_RESPONSE_PAYLOAD_ERROR_HANDLER
:
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190905145252262.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI5MTc1NjA1,size_16,color_FFFFFF,t_70)
/usr/local/lib/python3.6/site-packages/rest_framework_jwt/views.py
中修改:![在这里插入图片描述](https://img-blog.csdnimg.cn/20190905145911121.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI5MTc1NjA1,size_16,color_FFFFFF,t_70)
- 我们在自己的视图里编写一个
jwt_response_payload_error_handler
函数
def jwt_response_payload_error_handler(serializer, request = None):
return {
"msg": "用户名或者密码错误",
"status": 400,
"detail": serializer.errors
}
- 最后我们在我们项目的
settings.py
的JWT
中添加错误返回的函数:
JWT_AUTH = {
'JWT_EXPIRATION_DELTA': datetime.timedelta(seconds=6000),
'JWT_RESPONSE_PAYLOAD_HANDLER': 'Users.views.jwt_response_payload_handler',
'JWT_RESPONSE_PAYLOAD_ERROR_HANDLER': 'Users.views.jwt_response_payload_error_handler',
}
- 最后的样子就成为这样:
![在这里插入图片描述](https://img-blog.csdnimg.cn/20190905150722891.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI5MTc1NjA1,size_16,color_FFFFFF,t_70)