hts-code-query-system-backend/exceptions/handle.py

# _*_ coding : UTF-8 _*_
# @Time : 2025/01/18 02:24
# @UpdateTime : 2025/01/18 02:24
# @Author : sonder
# @File : handle.py
# @Software : PyCharm
# @Comment : 本程序
from fastapi import FastAPI, Request
from fastapi.exceptions import HTTPException
from pydantic_validation_decorator import FieldValidationError
from starlette.responses import JSONResponse

from exceptions.exception import (
    AuthException,
    LoginException,
    ModelValidatorException,
    PermissionException,
    ServiceException,
    ServiceWarning,
)
from utils.log import logger
from utils.response import Response


def handle_exception(app: FastAPI):
    """
    全局异常处理拦截器。
    - 捕获并处理所有异常，返回统一的接口响应格式。
    """

    @app.exception_handler(AuthException)
    async def auth_exception_handler(request: Request, exc: AuthException):
        """
        处理自定义身份验证异常（AuthException）。
        - 返回 401 未授权响应。
        """
        logger.warning(f"身份验证异常: {exc.message}")
        return Response.unauthorized(data=exc.data, msg=exc.message)

    @app.exception_handler(LoginException)
    async def login_exception_handler(request: Request, exc: LoginException):
        """
        处理自定义登录异常（LoginException）。
        - 返回 400 失败响应。
        """
        logger.warning(f"登录异常: {exc.message}")
        return Response.failure(data=exc.data, msg=exc.message)

    @app.exception_handler(ModelValidatorException)
    async def model_validator_exception_handler(request: Request, exc: ModelValidatorException):
        """
        处理自定义模型校验异常（ModelValidatorException）。
        - 返回 400 失败响应。
        """
        logger.warning(f"模型校验异常: {exc.message}")
        return Response.failure(data=exc.data, msg=exc.message)

    @app.exception_handler(FieldValidationError)
    async def field_validation_error_handler(request: Request, exc: FieldValidationError):
        """
        处理字段校验异常（FieldValidationError）。
        - 返回 400 失败响应。
        """
        logger.warning(f"字段校验异常: {exc.message}")
        return Response.failure(msg=exc.message)

    @app.exception_handler(PermissionException)
    async def permission_exception_handler(request: Request, exc: PermissionException):
        """
        处理自定义权限异常（PermissionException）。
        - 返回 403 未授权响应。
        """
        logger.warning(f"权限异常: {exc.message}")
        return Response.forbidden(data=exc.data, msg=exc.message)

    @app.exception_handler(ServiceException)
    async def service_exception_handler(request: Request, exc: ServiceException):
        """
        处理自定义服务异常（ServiceException）。
        - 返回 500 错误响应。
        """
        logger.error(f"服务异常: {exc.message}")
        return Response.error(data=exc.data, msg=exc.message)

    @app.exception_handler(ServiceWarning)
    async def service_warning_handler(request: Request, exc: ServiceWarning):
        """
        处理自定义服务警告（ServiceWarning）。
        - 返回 400 失败响应。
        """
        logger.warning(f"服务警告: {exc.message}")
        return Response.failure(data=exc.data, msg=exc.message)

    @app.exception_handler(HTTPException)
    async def http_exception_handler(request: Request, exc: HTTPException):
        """
        处理 FastAPI 的 HTTP 异常。
        - 返回统一的错误响应格式。
        """
        logger.warning(f"HTTP 异常: {exc.detail}")
        return Response.failure(msg=exc.detail)

    @app.exception_handler(Exception)
    async def exception_handler(request: Request, exc: Exception):
        """
        处理其他未捕获的异常。
        - 返回 500 错误响应。
        """
        logger.exception(f"未捕获的异常: {str(exc)}")
        return Response.error(msg="服务器内部错误")

    @app.exception_handler(404)
    async def not_found_exception_handler(request: Request, exc: HTTPException):
        """
        处理 404 未找到资源异常。
        - 返回 404 失败响应。
        """
        logger.warning(f"404 异常: {request.url} 未找到")
        return JSONResponse(
            content={"code": 404, "msg": "无效路径！", "data": None},
            status_code=404,
        )

    @app.exception_handler(405)
    async def method_not_allowed_handler(request: Request, exc: HTTPException):
        """
        处理 405 方法不允许异常。
        - 返回 405 失败响应。
        """
        logger.warning(f"405 异常: {request.method} 方法不允许")
        return JSONResponse(
            status_code=405,
            content={"code": 405, "msg": "请求方法错误", "data": None},
        )