from fastapi import FastAPI
from contextlib import asynccontextmanager

from .core.database_loader import init_database, load_database_pool, close_database_pool
from app.api.image_data import router as image_data_router
import os

from .core.config import settings
from .core.logger import setup_logging, get_logger

setup_logging()
# 获取一个用于 main 模块的日志记录器
logger = get_logger(__name__)

@asynccontextmanager
async def lifespan(main_app: FastAPI):
    print("--- 应用启动 ---")
    # --- 文件和目录准备 ---
    os.makedirs(settings.DATA_DIR, exist_ok=True)

    # --- 数据库初始化 ---
    init_database()
    load_database_pool()

    yield

    print("--- 应用关闭 ---")
    close_database_pool()


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

app.include_router(image_data_router, prefix=settings.API_ImageData_prefix, tags=["Image Data"])