refactor: replace Engine.execute() with Connection.execute() (#41917)

This commit is contained in:
Hans Yu
2026-07-13 18:40:33 +02:00
committed by GitHub
parent 8f75f1a353
commit 68ebc22e1e
32 changed files with 342 additions and 266 deletions

View File

@@ -153,10 +153,11 @@ def main( # noqa: C901
)
print(f"Migration goes from {down_revision} to {revision}")
current_revision = db.engine.execute(
text("SELECT version_num FROM alembic_version")
).scalar()
print(f"Current version of the DB is {current_revision}")
with db.engine.connect() as conn:
current_revision = conn.execute(
text("SELECT version_num FROM alembic_version")
).scalar()
print(f"Current version of the DB is {current_revision}")
if current_revision != down_revision:
if not force: