from sqlmodel import Session, create_engine, select

from app import crud
from app.core.config import settings
from app.models import User, UserCreate

engine = create_engine(str(settings.SQLALCHEMY_DATABASE_URI))


# 确保在初始化数据库之前已导入所有 SQLModel 模型（app.models）
# 否则，SQLModel 可能无法正确初始化关系
# 更多详情请参考：https://github.com/fastapi/full-stack-fastapi-template/issues/28


def init_db(session: Session) -> None:
    # 数据库表应该通过 Alembic 迁移创建
    # 但如果你不想使用迁移，可以取消注释下面的代码来创建表
    # from sqlmodel import SQLModel

    # 这样可以正常工作，因为模型已经从 app.models 导入并注册
    # SQLModel.metadata.create_all(engine)

    user = session.exec(
        select(User).where(User.email == settings.FIRST_SUPERUSER)
    ).first()
    if not user:
        user_in = UserCreate(
            email=settings.FIRST_SUPERUSER,
            password=settings.FIRST_SUPERUSER_PASSWORD,
            is_superuser=True,
        )
        user = crud.create_user(session=session, user_create=user_in)