Skip to content

Commit

Permalink
Merge pull request #7103 from dannon/migration_fixes
Browse files Browse the repository at this point in the history
Fix migration 0146 column selection on downgrade
  • Loading branch information
nsoranzo committed Dec 12, 2018
2 parents d7bac46 + 586f0ca commit 58dd7e6
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 19 deletions.
Expand Up @@ -104,7 +104,7 @@ def downgrade(migrate_engine):
"FROM workflow_step_connection_predowngrade145 AS wsc JOIN workflow_step_input AS wsi ON wsc.input_step_input_id = wsi.id ORDER BY wsc.id"
migrate_engine.execute(insert_step_connections_cmd)

for table in (WorkflowStepInput_table, NewWorkflowStepConnection_table):
for table in (NewWorkflowStepConnection_table, WorkflowStepInput_table):
_drop(table)


Expand Down
22 changes: 4 additions & 18 deletions lib/galaxy/model/migrate/versions/0146_workflow_paths.py
Expand Up @@ -23,40 +23,26 @@ def upgrade(migrate_engine):
print(__doc__)
metadata.reflect()

__add_column(from_path_column, "stored_workflow", metadata)
_add_column(from_path_column, "stored_workflow", metadata)


def downgrade(migrate_engine):
metadata.bind = migrate_engine

__drop_column(from_path_column, "stored_workflow", metadata)
_drop_column("from_path", "stored_workflow", metadata)


def __add_column(column, table_name, metadata, **kwds):
def _add_column(column, table_name, metadata, **kwds):
try:
table = Table(table_name, metadata, autoload=True)
column.create(table, **kwds)
except Exception:
log.exception("Adding column %s failed.", column)


def __drop_column(column_name, table_name, metadata):
def _drop_column(column_name, table_name, metadata):
try:
table = Table(table_name, metadata, autoload=True)
getattr(table.c, column_name).drop()
except Exception:
log.exception("Dropping column %s failed.", column_name)


def _create(table):
try:
table.create()
except Exception:
log.exception("Creating %s table failed.", table.name)


def _drop(table):
try:
table.drop()
except Exception:
log.exception("Dropping %s table failed.", table.name)

0 comments on commit 58dd7e6

Please sign in to comment.