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.0009_add_malicious_column
"""
Migration script to add the malicious column to the repository_metadata 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):
print(__doc__)
metadata.bind = migrate_engine
metadata.reflect()
# Create and initialize imported column in job table.
Repository_metadata_table = Table("repository_metadata", metadata, autoload=True)
c = Column("malicious", Boolean, default=False, index=True)
try:
# Create
c.create(Repository_metadata_table, index_name="ix_repository_metadata_malicious")
assert c is Repository_metadata_table.c.malicious
# 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 repository_metadata SET malicious={default_false}")
except Exception:
log.exception("Adding malicious column to the repository_metadata table failed.")
[docs]def downgrade(migrate_engine):
metadata.bind = migrate_engine
metadata.reflect()
# Drop malicious column from repository_metadata table.
Repository_metadata_table = Table("repository_metadata", metadata, autoload=True)
try:
Repository_metadata_table.c.malicious.drop()
except Exception:
log.exception("Dropping column malicious from the repository_metadata table failed.")