main.py 946 B

1234567891011121314151617181920212223242526272829303132333435
  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. settings.set_config()
  15. # --- 文件和目录准备 ---
  16. os.makedirs(settings.DATA_DIR, exist_ok=True)
  17. # --- 数据库初始化 ---
  18. init_database()
  19. load_database_pool()
  20. yield
  21. print("--- 应用关闭 ---")
  22. close_database_pool()
  23. app = FastAPI(title="卡片分数数据存储服务", lifespan=lifespan)
  24. app.include_router(image_data_router, prefix=settings.API_ImageData_prefix, tags=["Image Data"])