# 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