26 lines
802 B
Python
26 lines
802 B
Python
|
# exceptions.py
|
|||
|
|
from rest_framework.views import exception_handler
|
|||
|
|
from rest_framework.response import Response
|
|||
|
|
|
|||
|
|
def custom_exception_handler(exc, context):
|
|||
|
|
# 调用默认的异常处理器获取标准错误响应
|
|||
|
|
response = exception_handler(exc, context)
|
|||
|
|
|
|||
|
|
# 如果是已知的异常类型，进行统一格式化
|
|||
|
|
if response is not None:
|
|||
|
|
data = {
|
|||
|
|
'msg': str(exc) if not hasattr(exc, 'detail') else exc.detail,
|
|||
|
|
'data': None,
|
|||
|
|
'code': response.status_code
|
|||
|
|
}
|
|||
|
|
response.data = data
|
|||
|
|
else:
|
|||
|
|
# 处理未捕获的异常（比如 500）
|
|||
|
|
data = {
|
|||
|
|
'msg': 'Internal Server Error',
|
|||
|
|
'data': None,
|
|||
|
|
'code': 500
|
|||
|
|
}
|
|||
|
|
response = Response(data, status=500)
|
|||
|
|
|
|||
|
|
return response
|