datax-admin/app/__init__.py

from app.models.database import SessionLocal

# Dependency
def get_db():
    try:
        db = SessionLocal()
        yield db
    finally:
        db.close()