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
|