# -*- coding: utf-8 -*-
"""
DRF 自定义异常处理器：统一错误响应格式，并确保返回正确 HTTP 状态码。
"""
from rest_framework.views import exception_handler
from rest_framework.response import Response
from rest_framework.exceptions import APIException


def custom_exception_handler(exc, context):
    response = exception_handler(exc, context)
    if response is not None:
        # 确保使用异常自带的 HTTP 状态码（如 401 登录失效、403 无权限）
        if isinstance(exc, APIException) and getattr(exc, "status_code", None):
            response.status_code = exc.status_code
        # 统一格式：{"detail": "...", "code": 状态码}
        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)}
        response.data["code"] = response.status_code
    return response