103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
"""FastAPI 应用入口。
|
||
|
||
创建应用实例、配置 CORS 中间件、注册全局异常处理器、挂载 API 路由。
|
||
启动命令: uvicorn backend.app.main:app --reload
|
||
"""
|
||
|
||
import logging
|
||
import traceback
|
||
from contextlib import asynccontextmanager
|
||
|
||
logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(name)s] %(levelname)s: %(message)s")
|
||
|
||
from fastapi import FastAPI
|
||
from fastapi import Request
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
from fastapi.responses import JSONResponse
|
||
from fastapi.staticfiles import StaticFiles
|
||
from pathlib import Path
|
||
|
||
from backend.app.api.routes import api_router
|
||
from backend.app.api.ws import router as ws_router
|
||
from backend.app.core.config import get_settings
|
||
from backend.app.core.exceptions import AppException
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
settings = get_settings()
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""管理应用生命周期：启动时自动初始化数据库和调度器，关闭时清理。"""
|
||
# 自动初始化数据库（检查/创建数据库、执行迁移、初始化种子数据）
|
||
try:
|
||
from backend.app.startup.db_init import auto_init_database
|
||
auto_init_database()
|
||
except Exception as e:
|
||
logger.error("数据库自动初始化失败: %s", e)
|
||
# 不阻止启动，允许手动修复
|
||
|
||
from backend.app.scheduler import init_scheduler, shutdown_scheduler
|
||
init_scheduler()
|
||
yield
|
||
shutdown_scheduler()
|
||
|
||
|
||
app = FastAPI(title=settings.app_name, lifespan=lifespan)
|
||
|
||
# 配置 CORS 跨域，允许前端开发服务器访问
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=settings.cors_origins,
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
|
||
@app.exception_handler(AppException)
|
||
async def app_exception_handler(_: Request, exc: AppException) -> JSONResponse:
|
||
"""全局捕获 AppException，返回统一的错误 JSON 格式。
|
||
|
||
避免每个路由重复构造错误响应，所有业务异常由此处理器统一格式化。
|
||
"""
|
||
# 统一业务异常返回格式，避免每个接口重复构造错误响应。
|
||
return JSONResponse(
|
||
status_code=exc.status_code,
|
||
content={
|
||
"code": exc.code,
|
||
"message": exc.message,
|
||
"data": {},
|
||
},
|
||
)
|
||
|
||
|
||
@app.exception_handler(Exception)
|
||
async def unhandled_exception_handler(_: Request, exc: Exception) -> JSONResponse:
|
||
"""捕获所有未处理异常，记录完整 traceback 到日志并返回 500。
|
||
|
||
避免 FastAPI 默认的 bare 500 响应，让运维能从日志中定位问题。
|
||
"""
|
||
logger.error("Unhandled exception: %s\n%s", exc, traceback.format_exc())
|
||
return JSONResponse(
|
||
status_code=500,
|
||
content={
|
||
"code": 500,
|
||
"message": f"服务器内部错误: {type(exc).__name__}: {exc}",
|
||
"data": {},
|
||
},
|
||
)
|
||
|
||
|
||
# 挂载所有 API 路由，路由前缀在 router.py 中统一定义
|
||
app.include_router(api_router)
|
||
|
||
# 挂载 WebSocket 路由（不通过 api_router，直接挂载）
|
||
app.include_router(ws_router)
|
||
|
||
# 挂载本地上传文件目录为静态文件服务（开发环境用，生产环境由 Nginx 代理）
|
||
_local_upload_dir = Path(settings.local_upload_dir)
|
||
_local_upload_dir.mkdir(parents=True, exist_ok=True)
|
||
app.mount("/uploads", StaticFiles(directory=str(_local_upload_dir)), name="local-uploads")
|