fix(登录): 登录优化

This commit is contained in:
猿小天
2023-04-09 23:09:35 +08:00
committed by H0nGzA1
parent 15c4808bbb
commit eea9d320af

View File

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