from contextlib import contextmanager
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine


engine = create_engine("sqlite:///alchemy.db")
Session = sessionmaker(bind=engine)

Base.metadata.create_all(engine)

@contextmanager
def get_session():
    session = Session()
    try:
        yield session
    finally:
        session.close()