admin/django-vue3-admin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(登录): 登录优化

Browse Source
This commit is contained in:
猿小天
2023-04-09 23:09:35 +08:00
committed by H0nGzA1
parent 15c4808bbb
commit eea9d320af
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/dvadmin/utils/exception.py
View File

@@ -38,11 +38,15 @@ def CustomExceptionHandler(ex, context):
# 调用默认的异常处理函数 # 调用默认的异常处理函数
response = exception_handler(ex, context) response = exception_handler(ex, context)
if isinstance(ex, AuthenticationFailed): if isinstance(ex, AuthenticationFailed):
# 如果是身份验证错误
if response and response.data.get('detail') == "Given token not valid for any token type":
code = 401 code = 401
code_type = response.data.get('detail').code msg = ex.detail
if code_type == 'no_active_account': elif response and response.data.get('detail') == "Token is blacklisted":
code=400 # token在黑名单
return ErrorResponse(status=HTTP_401_UNAUTHORIZED) return ErrorResponse(status=HTTP_401_UNAUTHORIZED)
else:
code = 401
msg = ex.detail msg = ex.detail
elif isinstance(ex,Http404): elif isinstance(ex,Http404):
code = 400 code = 400