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 galaxy.model.migrate.versions.0078_add_columns_for_disk_usage_accounting

Migration script to add 'total_size' column to the dataset table, 'purged'
column to the HDA table, and 'disk_usage' column to the User and GalaxySession

import logging

from sqlalchemy import Boolean, Column, MetaData, Numeric, Table

from galaxy.model.migrate.versions.util import add_column, drop_column

log = logging.getLogger(__name__)
metadata = MetaData()

[docs]def upgrade(migrate_engine): print(__doc__) metadata.bind = migrate_engine metadata.reflect() c = Column('total_size', Numeric(15, 0)) add_column(c, 'dataset', metadata) HistoryDatasetAssociation_table = Table("history_dataset_association", metadata, autoload=True) c = Column("purged", Boolean, index=True, default=False) add_column(c, HistoryDatasetAssociation_table, metadata, index_name="ix_history_dataset_association_purged") try: migrate_engine.execute(HistoryDatasetAssociation_table.update().values(purged=False)) except Exception: log.exception("Updating column 'purged' of table 'history_dataset_association' failed.") c = Column('disk_usage', Numeric(15, 0), index=True) add_column(c, 'galaxy_user', metadata, index_name="ix_galaxy_user_disk_usage") c = Column('disk_usage', Numeric(15, 0), index=True) add_column(c, 'galaxy_session', metadata, index_name="ix_galaxy_session_disk_usage")
[docs]def downgrade(migrate_engine): metadata.bind = migrate_engine metadata.reflect() drop_column('disk_usage', 'galaxy_session', metadata) drop_column('disk_usage', 'galaxy_user', metadata) # SQLAlchemy Migrate has a bug when dropping a boolean column in SQLite if migrate_engine.name != 'sqlite': drop_column('purged', 'history_dataset_association', metadata) drop_column('total_size', 'dataset', metadata)