#db/__init__.py
from contextlib import contextmanager
from models import Base


from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine
import env_provider

engine = create_engine(env_provider.DB_CONNECT_STRING)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
@contextmanager
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()


def create_all():
    with engine.begin() as conn:
        Base.metadata.create_all(bind=conn)