mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: qabot
|
||
@Author：虎
|
||
@file： handle_exception.py
|
||
@date：2023/9/5 19:29
|
||
@desc:
|
||
"""
|
||
import django.core.exceptions
|
||
from psycopg2 import IntegrityError
|
||
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
|
||
|
||
import traceback
|
||
def to_result(key, args, parent_key=None):
|
||
"""
|
||
将校验异常 args转换为统一数据
|
||
:param key: 校验key
|
||
:param args: 校验异常参数
|
||
:param parent_key 父key
|
||
:return: 接口响应对象
|
||
"""
|
||
error_detail = list(filter(
|
||
lambda d: True if isinstance(d, ErrorDetail) else True if isinstance(d, dict) and len(
|
||
d.keys()) > 0 else False,
|
||
(args[0] if len(args) > 0 else {key: [ErrorDetail('未知异常', code='unknown')]}).get(key)))[0]
|
||
|
||
if isinstance(error_detail, dict):
|
||
return list(map(lambda k: to_result(k, args=[error_detail],
|
||
parent_key=key if parent_key is None else parent_key + '.' + key),
|
||
error_detail.keys() if len(error_detail) > 0 else []))[0]
|
||
|
||
return result.Result(500 if isinstance(error_detail.code, str) else error_detail.code,
|
||
message=f"【{key if parent_key is None else parent_key + '.' + key}】为必填参数" if str(
|
||
error_detail) == "This field is required." else f"【{key if parent_key is None else parent_key + '.' + key}】" + error_detail)
|
||
|
||
|
||
def validation_error_to_result(exc: ValidationError):
|
||
"""
|
||
校验异常转响应对象
|
||
:param exc: 校验异常
|
||
:return: 接口响应对象
|
||
"""
|
||
try:
|
||
res = list(map(lambda key: to_result(key, args=exc.args),
|
||
exc.args[0].keys() if len(exc.args) > 0 else []))
|
||
except Exception as e:
|
||
return result.error(str(exc.detail))
|
||
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)
|
||
traceback.print_exc()
|
||
# 在此处补充自定义的异常处理
|
||
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)
|
||
if response is None:
|
||
return result.error(str(exc))
|
||
return response
|