fix(登录): 登录优化

This commit is contained in:
猿小天
2023-04-09 23:09:35 +08:00
parent b96c220f1d
commit 160c376f81

View File

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