From 17789c1c0bbe83d369ea5df46aa674d140558618 Mon Sep 17 00:00:00 2001 From: Daniel Schadt Date: Sun, 8 Jun 2025 00:25:03 +0200 Subject: fix postgres detection in upgrade script --- fietsboek/alembic/versions/20220808_d085998b49ca.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/fietsboek/alembic/versions/20220808_d085998b49ca.py b/fietsboek/alembic/versions/20220808_d085998b49ca.py index b33b8ee..2c5b71d 100644 --- a/fietsboek/alembic/versions/20220808_d085998b49ca.py +++ b/fietsboek/alembic/versions/20220808_d085998b49ca.py @@ -6,7 +6,6 @@ Create Date: 2022-08-08 14:11:40.746008 """ import sqlalchemy as sa -import alembic.context from alembic import op # revision identifiers, used by Alembic. @@ -15,15 +14,15 @@ down_revision = '091ce24409fe' branch_labels = None depends_on = None -is_postgres = alembic.context.config.get_main_option("sqlalchemy.url", "").startswith("postgresql") +is_postgres = op.get_bind().dialect.name == "postgresql" def upgrade(): if is_postgres: - op.add_column('tracks', sa.Column('type', sa.Enum('ORGANIC', 'SYNTHETIC', name='tracktype'), nullable=True)) - else: tracktype = sa.dialects.postgresql.ENUM("ORGANIC", "SYNTHETIC", name="tracktype") tracktype.create(op.get_bind()) op.add_column("tracks", sa.Column("type", tracktype, nullable=True)) + else: + op.add_column('tracks', sa.Column('type', sa.Enum('ORGANIC', 'SYNTHETIC', name='tracktype'), nullable=True)) op.execute("UPDATE tracks SET type='ORGANIC';") def downgrade(): -- cgit v1.2.3