46 lines
2.4 KiB
Python
46 lines
2.4 KiB
Python
from ninja_extra import NinjaExtraAPI
|
||
from django.http import HttpRequest, HttpResponse
|
||
from typing import Any
|
||
from utils.codes import HTTP_USER_PASSWORD_ERROR_CODE
|
||
|
||
# 重写ninja返回 - 全局统一视图函数返回，如果None则如下返回
|
||
class ChenNinjaAPI(NinjaExtraAPI):
|
||
def create_response(
|
||
self, request: HttpRequest, data: Any, *, status: int = 200, code: int = 200, message: str = "请求成功",
|
||
temporal_response: HttpResponse = None,
|
||
) -> HttpResponse:
|
||
std_data = {
|
||
"code": code,
|
||
"data": data,
|
||
"message": message,
|
||
"success": True
|
||
}
|
||
# 因为extra的APIException会添加到data['detail']里面的，所以做一致性处理转换
|
||
if std_data['data'] is not None:
|
||
if 'detail' in std_data['data']:
|
||
std_data['message'] = std_data['data']['detail']
|
||
# ~~~~~~~~~~正常异常,status进行通用处理:TODO:后续规划~~~~~~~~~~
|
||
# 1.当状态码为403时候，给前端提示message
|
||
if status == 403:
|
||
std_data['message'] = '您没有权限这样做'
|
||
# 2.当状态码为401时候，根据APIException处理
|
||
elif status == 401:
|
||
if (std_data['data']['detail'] == '找不到指定凭据对应的有效用户' or
|
||
std_data['data']['detail'] == 'No active account found with the given credentials'):
|
||
std_data['message'] = '账号或密码错误，如果是内网登录检查密码是否过期...'
|
||
std_data['data']['code'] = HTTP_USER_PASSWORD_ERROR_CODE # TODO:后续单独以枚举方式定义code
|
||
else:
|
||
std_data['message'] = '您的token已过期，请重新登录'
|
||
# 3.因为前端是获取message，所以这里处理
|
||
elif status != 200 and std_data['message'] == '请求成功':
|
||
std_data['message'] = '请求失败，请检查'
|
||
elif status == 404:
|
||
std_data['message'] = '未找到相应的内容，请检查参数'
|
||
elif status == 422:
|
||
std_data['message'] = '请求的参数或响应数据未通过验证'
|
||
content = self.renderer.render(request, std_data, response_status=status)
|
||
content_type = "{}; charset={}".format(
|
||
self.renderer.media_type, self.renderer.charset
|
||
)
|
||
return HttpResponse(content, status=status, content_type=content_type)
|