Warning
This document is for an old release of Galaxy. You can alternatively view this page in the latest release if it exists or view the top of the latest release's documentation.
Source code for tool_shed.webapp.model.migrate.versions.0002_add_tool_suite_column
"""
Migration script to add the suite column to the tool table.
"""
import logging
import sys
from sqlalchemy import (
Boolean,
Column,
MetaData,
Table,
)
log = logging.getLogger(__name__)
log.setLevel(logging.DEBUG)
handler = logging.StreamHandler(sys.stdout)
format = "%(name)s %(levelname)s %(asctime)s %(message)s"
formatter = logging.Formatter(format)
handler.setFormatter(formatter)
log.addHandler(handler)
metadata = MetaData()
[docs]def upgrade(migrate_engine):
metadata.bind = migrate_engine
print(__doc__)
metadata.reflect()
# Create and initialize imported column in job table.
Tool_table = Table("tool", metadata, autoload=True)
c = Column("suite", Boolean, default=False, index=True)
try:
# Create
c.create(Tool_table, index_name="ix_tool_suite")
assert c is Tool_table.c.suite
# Initialize.
if migrate_engine.name == "mysql" or migrate_engine.name == "sqlite":
default_false = "0"
elif migrate_engine.name in ["postgresql", "postgres"]:
default_false = "false"
migrate_engine.execute(f"UPDATE tool SET suite={default_false}")
except Exception:
log.exception("Adding suite column to the tool table failed.")
[docs]def downgrade(migrate_engine):
metadata.bind = migrate_engine
metadata.reflect()
# Drop suite column from tool table.
Tool_table = Table("tool", metadata, autoload=True)
try:
Tool_table.c.suite.drop()
except Exception:
log.exception("Dropping column suite from the tool table failed.")