CardScoreDataServer/app/main.py

from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
import os

from .core.database_loader import init_database, load_database_pool, close_database_pool
from app.api import cards as cards_router
from app.api import images as images_router
from app.api import labelme as labelme_router
from .core.config import settings
from .core.logger import setup_logging, get_logger

setup_logging()
logger = get_logger(__name__)

settings.set_config()
os.makedirs(settings.DATA_DIR, exist_ok=True)

@asynccontextmanager
async def lifespan(main_app: FastAPI):
    print("--- 应用启动 ---")

    init_database()
    load_database_pool()
    yield
    print("--- 应用关闭 ---")
    close_database_pool()

app = FastAPI(title="卡片分数数据存储服务", lifespan=lifespan)

app.mount("/Data", StaticFiles(directory="Data"), name="Data")
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"]
)

app.include_router(cards_router.router, prefix=f"{settings.API_PREFIX}/cards", tags=["Cards"])
app.include_router(images_router.router, prefix=f"{settings.API_PREFIX}/images", tags=["Images"])
app.include_router(labelme_router.router, prefix=f"{settings.API_PREFIX}/labelme", tags=["Labelme"])