分模候选方案
This commit is contained in:
+23
-12
@@ -1,16 +1,27 @@
|
||||
from fastapi import APIRouter
|
||||
import importlib
|
||||
|
||||
from api.v1.health_router import router as health_router
|
||||
from api.v1.upload_router import router as upload_router
|
||||
from api.v1.task_router import router as task_router
|
||||
from api.v1.history_router import router as history_router
|
||||
from api.v1.debug_router import router as debug_router
|
||||
from api.v1.advanced_router import router as advanced_router
|
||||
from utils.logger import get_logger
|
||||
|
||||
logger = get_logger(__name__)
|
||||
|
||||
router = APIRouter()
|
||||
router.include_router(health_router)
|
||||
router.include_router(upload_router)
|
||||
router.include_router(task_router)
|
||||
router.include_router(history_router)
|
||||
router.include_router(debug_router)
|
||||
router.include_router(advanced_router)
|
||||
|
||||
def _safe_include(module_path: str, label: str):
|
||||
try:
|
||||
module = importlib.import_module(module_path)
|
||||
router_obj = getattr(module, "router", None)
|
||||
if router_obj is None:
|
||||
raise ValueError("未找到 router 对象")
|
||||
router.include_router(router_obj)
|
||||
logger.info(f"{label} 路由加载成功")
|
||||
except Exception as exc:
|
||||
logger.warning(f"{label} 路由加载失败,已跳过: {exc}")
|
||||
|
||||
|
||||
_safe_include("api.v1.health_router", "健康检查")
|
||||
_safe_include("api.v1.upload_router", "上传")
|
||||
_safe_include("api.v1.task_router", "任务")
|
||||
_safe_include("api.v1.history_router", "历史")
|
||||
_safe_include("api.v1.debug_router", "调试")
|
||||
_safe_include("api.v1.advanced_router", "高级")
|
||||
|
||||
Reference in New Issue
Block a user