from sqlmodel import create_engine, SQLModel, Session

from ..model.models import Base

SQLALCHEMY_DATABASE_URL = "sqlite:///./gatekeeper.db"

engine = create_engine(SQLALCHEMY_DATABASE_URL)

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

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

def add_and_refresh(db: Session, obj):
    db.add(obj)
    db.commit()
    db.refresh(obj)
    return obj