main.py 979 B

1234567891011121314151617181920212223242526272829
  1. from fastapi import FastAPI
  2. from contextlib import asynccontextmanager
  3. import os
  4. from .core.database_loader import init_database, load_database_pool, close_database_pool
  5. from app.api import cards as cards_router # 导入新路由
  6. from app.api import images as images_router # 导入新路由
  7. from .core.config import settings
  8. from .core.logger import setup_logging, get_logger
  9. setup_logging()
  10. logger = get_logger(__name__)
  11. @asynccontextmanager
  12. async def lifespan(main_app: FastAPI):
  13. print("--- 应用启动 ---")
  14. settings.set_config()
  15. os.makedirs(settings.DATA_DIR, exist_ok=True)
  16. init_database()
  17. load_database_pool()
  18. yield
  19. print("--- 应用关闭 ---")
  20. close_database_pool()
  21. app = FastAPI(title="卡片分数数据存储服务", lifespan=lifespan)
  22. app.include_router(cards_router.router, prefix=f"{settings.API_PREFIX}/cards", tags=["Cards"])
  23. app.include_router(images_router.router, prefix=f"{settings.API_PREFIX}/images", tags=["Images"])