# -*- coding: utf-8 -*- """ 统一异常处理模块 定义所有自定义异常类和错误处理函数 """ from datetime import datetime, timezone from fastapi import HTTPException, Request from fastapi.exception_handlers import ( http_exception_handler as fastapi_http_exception_handler, request_validation_exception_handler as fastapi_validation_exception_handler, ) from fastapi.exceptions import RequestValidationError from fastapi.responses import JSONResponse import logging from typing import Any, Dict, Optional logger = logging.getLogger(__name__) def _uses_meeting_legacy_response(request: Request) -> bool: """Return True for Model-Test-New compatible offline meeting/speaker APIs.""" path = request.url.path.rstrip("/") method = request.method.upper() if path.endswith("/asr/transcriptions"): return method == "POST" if "/asr/transcriptions/" in path: return method == "GET" if path.endswith("/speakers"): return method == "POST" return False def get_iso_timestamp() -> str: """获取ISO 8601格式的UTC时间戳""" return datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ") def create_error_response( error_code: str, message: str, task_id: str = "", details: Optional[Dict[str, Any]] = None, ) -> Dict[str, Any]: """ 创建标准错误响应格式 Args: error_code: 错误代码（如 INVALID_PARAMETER） message: 人类可读的错误信息 task_id: 任务ID（可选） details: 额外详细信息（可选） Returns: 标准错误响应字典 """ return { "error_code": error_code, "message": message, "task_id": task_id or "", "timestamp": get_iso_timestamp(), "details": details or {}, } def get_http_status_code(status_code: int) -> int: """Map internal API status code to HTTP status code.""" return 500 if status_code >= 50000000 else 400 class APIException(Exception): """API基础异常类""" def __init__( self, status_code: int, message: str, task_id: str = "", error_code: str = "", details: Optional[Dict[str, Any]] = None, ): self.status_code = status_code self.message = message self.task_id = task_id self.error_code = error_code or self._get_error_code(status_code) self.details = details or {} super().__init__(self.message) def _get_error_code(self, status_code: int) -> str: """根据状态码获取错误代码""" code_mapping = { 20000000: "SUCCESS", 40000000: "DEFAULT_CLIENT_ERROR", 40000001: "AUTHENTICATION_FAILED", 40000002: "INVALID_MESSAGE", 40000003: "INVALID_PARAMETER", 40000004: "IDLE_TIMEOUT", 40000005: "TOO_MANY_REQUESTS", 40000010: "TRIAL_EXPIRED", 41010101: "UNSUPPORTED_SAMPLE_RATE", 50000000: "DEFAULT_SERVER_ERROR", 50000001: "INTERNAL_GRPC_ERROR", } return code_mapping.get(status_code, "UNKNOWN_ERROR") def to_dict(self) -> Dict[str, Any]: """ 将异常转换为标准错误响应字典 Returns: 标准错误响应字典，包含 error_code, message, task_id, timestamp, details """ return create_error_response( error_code=self.error_code, message=self.message, task_id=self.task_id, details=self.details, ) # 标准异常类 class AuthenticationException(APIException): """身份认证异常""" def __init__(self, message: str, task_id: str = "", details: Optional[Dict[str, Any]] = None): super().__init__(40000001, message, task_id, details=details) class InvalidMessageException(APIException): """无效消息异常""" def __init__(self, message: str, task_id: str = "", details: Optional[Dict[str, Any]] = None): super().__init__(40000002, message, task_id, details=details) class InvalidParameterException(APIException): """无效参数异常""" def __init__(self, message: str, task_id: str = "", details: Optional[Dict[str, Any]] = None): super().__init__(40000003, message, task_id, details=details) class UnsupportedSampleRateException(APIException): """不支持的采样率异常""" def __init__(self, message: str, task_id: str = "", details: Optional[Dict[str, Any]] = None): super().__init__(41010101, message, task_id, details=details) class DefaultServerErrorException(APIException): """默认服务端错误异常""" def __init__(self, message: str, task_id: str = "", details: Optional[Dict[str, Any]] = None): super().__init__(50000000, message, task_id, details=details) # 异常处理器 async def api_exception_handler(request: Request, exc: Exception) -> JSONResponse: """API异常处理器""" # FastAPI 只会在抛出 APIException 时调用此处理器 api_exc = exc if isinstance(exc, APIException) else APIException(50000000, str(exc)) logger.error(f"[{api_exc.task_id}] API异常: {api_exc.message}") # 使用标准错误格式 response_data = api_exc.to_dict() # 确定HTTP状态码 http_status_code = get_http_status_code(api_exc.status_code) return JSONResponse( content=response_data, headers={"task_id": api_exc.task_id} if api_exc.task_id else {}, status_code=http_status_code, ) async def http_exception_handler(request: Request, exc: Exception) -> JSONResponse: """FastAPI HTTPException handler for legacy meeting/speaker endpoints.""" http_exc = exc if isinstance(exc, HTTPException) else HTTPException(500, str(exc)) if not _uses_meeting_legacy_response(request): return await fastapi_http_exception_handler(request, http_exc) detail = http_exc.detail if isinstance(detail, dict): return JSONResponse( status_code=http_exc.status_code, content={ "code": detail.get("code", 4001), "message": detail.get("message", "Request failed"), }, ) return JSONResponse( status_code=http_exc.status_code, content={"code": 4001, "message": str(detail)}, ) async def validation_exception_handler(request: Request, exc: Exception) -> JSONResponse: """FastAPI/Pydantic validation handler for legacy meeting/speaker endpoints.""" validation_exc = exc if isinstance(exc, RequestValidationError) else None if validation_exc is None: return JSONResponse( status_code=422, content={"code": 1001, "message": "Invalid request"}, ) if not _uses_meeting_legacy_response(request): return await fastapi_validation_exception_handler(request, validation_exc) errors = validation_exc.errors() message = str(errors[0]["msg"]) if errors else "Invalid request" return JSONResponse(status_code=422, content={"code": 1001, "message": message}) async def general_exception_handler(request: Request, exc: Exception) -> JSONResponse: """通用异常处理器""" logger.error(f"未处理的异常: {str(exc)}", exc_info=True) # 使用标准错误格式 response_data = create_error_response( error_code="DEFAULT_SERVER_ERROR", message=f"内部服务错误: {str(exc)}", ) return JSONResponse(content=response_data, status_code=500)