main.py 919 B

123456789101112131415161718192021222324252627282930313233
  1. from fastapi import FastAPI
  2. from contextlib import asynccontextmanager
  3. from .core.database_loader import init_database, load_database_pool, close_database_pool
  4. from app.api.image_data import router as image_data_router
  5. import os
  6. from .core.config import settings
  7. from .core.logger import setup_logging, get_logger
  8. setup_logging()
  9. # 获取一个用于 main 模块的日志记录器
  10. logger = get_logger(__name__)
  11. @asynccontextmanager
  12. async def lifespan(main_app: FastAPI):
  13. print("--- 应用启动 ---")
  14. # --- 文件和目录准备 ---
  15. os.makedirs(settings.DATA_DIR, exist_ok=True)
  16. # --- 数据库初始化 ---
  17. init_database()
  18. load_database_pool()
  19. yield
  20. print("--- 应用关闭 ---")
  21. close_database_pool()
  22. app = FastAPI(title="卡片分数数据存储服务", lifespan=lifespan)
  23. app.include_router(image_data_router, prefix=settings.API_ImageData_prefix, tags=["Image Data"])