diff --git a/alembic/versions/0027_unify_company_to_contact.py b/alembic/versions/0027_unify_company_to_contact.py index c5c6cc8..177e030 100644 --- a/alembic/versions/0027_unify_company_to_contact.py +++ b/alembic/versions/0027_unify_company_to_contact.py @@ -38,7 +38,10 @@ def upgrade(): ) # Rename company_id to contact_id in mails (if column exists) conn = op.get_bind() - if conn.dialect.has_column(conn, "mails", "company_id"): + result = conn.execute( + sa.text("SELECT 1 FROM information_schema.columns WHERE table_name='mails' AND column_name='company_id'") + ).fetchone() + if result: op.alter_column("mails", "company_id", new_column_name="contact_id") @@ -61,5 +64,8 @@ def downgrade(): ) # Rename contact_id back to company_id in mails conn = op.get_bind() - if conn.dialect.has_column(conn, "mails", "contact_id"): + result = conn.execute( + sa.text("SELECT 1 FROM information_schema.columns WHERE table_name='mails' AND column_name='contact_id'") + ).fetchone() + if result: op.alter_column("mails", "contact_id", new_column_name="company_id")