From 562a025ac4e433c8ec9fdc0342c0c8722207e6c1 Mon Sep 17 00:00:00 2001 From: LoRiot Date: Sat, 22 Oct 2022 15:11:11 +0300 Subject: [PATCH] Using get_bind to get connection in the template --- alembic/script.py.mako | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/alembic/script.py.mako b/alembic/script.py.mako index 55df286..0ae67fc 100644 --- a/alembic/script.py.mako +++ b/alembic/script.py.mako @@ -6,7 +6,7 @@ Create Date: ${create_date} """ from alembic import op -import sqlalchemy as sa +from sqlalchemy.sql import text ${imports if imports else ""} # revision identifiers, used by Alembic. @@ -17,8 +17,8 @@ depends_on = ${repr(depends_on)} def upgrade() -> None: - ${upgrades if upgrades else "pass"} + ${upgrades if upgrades else "conn = op.get_bind()"} def downgrade() -> None: - ${downgrades if downgrades else "pass"} + ${downgrades if downgrades else "conn = op.get_bind()"}