from sqlmodel import SQLModel, create_engine, Session
from .config import settings

engine = create_engine(settings.db_url, connect_args={"check_same_thread": False})


def init_db() -> None:
    from app.models.job import Job  # noqa: F401
    from app.models.user import User  # noqa: F401
    SQLModel.metadata.create_all(engine)


def get_session():
    with Session(engine) as session:
        yield session