from sqlalchemy import create_engine, event from sqlalchemy.orm import sessionmaker, declarative_base from .config import get_settings settings = get_settings() engine = create_engine( settings.db_url, pool_pre_ping=True, pool_size=5, max_overflow=10, connect_args={"options": f"-csearch_path={settings.db_schema},public"} ) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) Base = declarative_base() def get_db(): db = SessionLocal() try: yield db finally: db.close()