57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: qabot
|
|||
|
|
@Author：虎
|
|||
|
|
@file： handle_exception.py
|
|||
|
|
@date：2023/9/5 19:29
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
from rest_framework.exceptions import ValidationError, ErrorDetail, APIException
|
|||
|
|
from rest_framework.views import exception_handler
|
|||
|
|
|
|||
|
|
from common.exception.app_exception import AppApiException
|
|||
|
|
from common.response import result
|
|||
|
|
|
|||
|
|
|
|||
|
|
def to_result(key, args):
|
|||
|
|
"""
|
|||
|
|
将校验异常 args转换为统一数据
|
|||
|
|
:param key: 校验key
|
|||
|
|
:param args: 校验异常参数
|
|||
|
|
:return: 接口响应对象
|
|||
|
|
"""
|
|||
|
|
error_detail = (args[0] if len(args) > 0 else {key: [ErrorDetail('未知异常', code='unknown')]}).get(key)[
|
|||
|
|
0]
|
|||
|
|
|
|||
|
|
return result.Result(500 if isinstance(error_detail.code, str) else error_detail.code,
|
|||
|
|
message=f"【{key}】为必填参数" if str(
|
|||
|
|
error_detail) == "This field is required." else error_detail)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def validation_error_to_result(exc: ValidationError):
|
|||
|
|
"""
|
|||
|
|
校验异常转响应对象
|
|||
|
|
:param exc: 校验异常
|
|||
|
|
:return: 接口响应对象
|
|||
|
|
"""
|
|||
|
|
res = list(map(lambda key: to_result(key, args=exc.args),
|
|||
|
|
exc.args[0].keys() if len(exc.args) > 0 else []))
|
|||
|
|
if len(res) > 0:
|
|||
|
|
return res[0]
|
|||
|
|
else:
|
|||
|
|
return result.error("未知异常")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def handle_exception(exc, context):
|
|||
|
|
exception_class = exc.__class__
|
|||
|
|
# 先调用REST framework默认的异常处理方法获得标准错误响应对象
|
|||
|
|
response = exception_handler(exc, context)
|
|||
|
|
# 在此处补充自定义的异常处理
|
|||
|
|
if issubclass(exception_class, ValidationError):
|
|||
|
|
return validation_error_to_result(exc)
|
|||
|
|
if issubclass(exception_class, AppApiException):
|
|||
|
|
return result.Result(exc.code, exc.message, response_status=exc.status_code)
|
|||
|
|
if issubclass(exception_class, APIException):
|
|||
|
|
return result.error(exc.detail)
|
|||
|
|
return response
|