Source code for galaxy.model.migrations.util
import logging
from alembic import op
from sqlalchemy import inspect
log = logging.getLogger(__name__)
[docs]def drop_column(table_name, column_name):
with op.batch_alter_table(table_name) as batch_op:
batch_op.drop_column(column_name)
[docs]def column_exists(table_name, column_name):
bind = op.get_context().bind
insp = inspect(bind)
columns = insp.get_columns(table_name)
return any(c["name"] == column_name for c in columns)