diff options
| -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(): |