mung-bean/sceneweaver

1import os2from sqlalchemy import create_engine3from sqlalchemy.ext.declarative import declarative_base4from sqlalchemy.orm import sessionmaker, Session5 6from dotenv import load_dotenv7 8load_dotenv()9 10DB_USER = os.getenv("DB_USER")11DB_PASSWORD = os.getenv("DB_PASSWORD")12DB_HOST = os.getenv("DB_HOST")13DB_PORT = os.getenv("DB_PORT")14DB_NAME = os.getenv("DB_NAME")15 16DATABASE_URL = f"postgresql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}"17 18engine = create_engine(DATABASE_URL)19SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)20 21 22Base = declarative_base()23 24 25def get_db():26    db = SessionLocal()27    try:28        yield db29    finally:30        db.close()31    return db32