diff options
| author | Daniel Schadt <kingdread@gmx.de> | 2025-06-08 00:25:03 +0200 |
|---|---|---|
| committer | Daniel Schadt <kingdread@gmx.de> | 2025-06-08 00:25:03 +0200 |
| commit | 17789c1c0bbe83d369ea5df46aa674d140558618 (patch) | |
| tree | 3f19ff13237c6b1749d234d1c311147ad900d4a1 | |
| parent | 914fd263f6ad0b7009a25bb13f1bd26b7b6c8558 (diff) | |
| download | fietsboek-17789c1c0bbe83d369ea5df46aa674d140558618.tar.gz fietsboek-17789c1c0bbe83d369ea5df46aa674d140558618.tar.bz2 fietsboek-17789c1c0bbe83d369ea5df46aa674d140558618.zip | |
fix postgres detection in upgrade script
| -rw-r--r-- | fietsboek/alembic/versions/20220808_d085998b49ca.py | 7 |
1 files 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(): |