"""
数据库会话管理
"""
from sqlalchemy.engine import Engine
from sqlalchemy import create_engine
from sqlmodel import Session
from typing import Generator
from app.core.config import settings

# 创建数据库引擎
engine: Engine = create_engine(
    settings.DATABASE_URL,
    connect_args={"check_same_thread": False},  # SQLite 需要此参数
    echo=True  # 开发环境显示SQL语句，生产环境设为False
)


def get_db() -> Generator[Session, None, None]:
    """获取数据库会话（用于依赖注入）"""
    with Session(engine) as session:
        yield session


# 为了向后兼容，保留 SessionLocal 别名
SessionLocal = Session