docker

import os from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker, declarative_base DATABASE_URL = os.getenv( "DATABASE_URL", "postgresql+psycopg2://postgres:password@localhost:5432/aegisflood_mvp", ) engine = create_engine(DATABASE_URL, pool_pre_ping=True, future=True) SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False, future=True) Base = declarative_base() def get_db(): db = SessionLocal() try: yield db finally: db.close()