18 lines
643 B
Python
18 lines
643 B
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
DRF 自定义异常处理器：统一错误响应格式。
|
|||
|
|
"""
|
|||
|
|
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:
|
|||
|
|
# 统一格式：{"detail": "..."}
|
|||
|
|
if isinstance(response.data, list):
|
|||
|
|
response.data = {"detail": response.data[0] if response.data else str(exc)}
|
|||
|
|
elif isinstance(response.data, dict) and "detail" not in response.data:
|
|||
|
|
response.data = {"detail": str(response.data)}
|
|||
|
|
return response
|