from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker, declarative_base from config.settings import DATABASE_URL # Create the SQLAlchemy engine engine = create_engine(DATABASE_URL) # Create a SessionLocal class SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) # Create a Base class for declarative models Base = declarative_base() def get_db(): db = SessionLocal() try: yield db finally: db.close()