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

sqlite_file_name = "database.db"
sqlite_url = f"sqlite:///" + str(settings.OUTPUT_DIR / sqlite_file_name)

engine = create_engine(sqlite_url, echo=False)

def get_session() -> Iterator[Session]:
    with Session(engine) as session:
        yield session

def create_db_and_tables():
    SQLModel.metadata.create_all(engine)