20 lines
436 B
Python
20 lines
436 B
Python
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 |