diff --git a/alembic/versions/0002_trigger_comment_body.py b/alembic/versions/0002_trigger_comment_body.py index 19e09d4..a3f7cb0 100644 --- a/alembic/versions/0002_trigger_comment_body.py +++ b/alembic/versions/0002_trigger_comment_body.py @@ -22,4 +22,12 @@ def upgrade() -> None: def downgrade() -> None: - op.drop_column("review_jobs", "trigger_comment_body") + bind = op.get_bind() + inspector = sa.inspect(bind) + table_names = inspector.get_table_names() + if "review_jobs" not in table_names: + return + + column_names = {column["name"] for column in inspector.get_columns("review_jobs")} + if "trigger_comment_body" in column_names: + op.drop_column("review_jobs", "trigger_comment_body")