def upgrade(migrate_engine): """ Upgrade operations go here. Don't create your own engine; bind migrate_engine to your metadata. """ meta.bind = migrate_engine table = meta.tables['USERS'] drop_column(COLUMN_N1, table) table = meta.tables['BURST_CONFIGURATIONS'] drop_column(COLUMN_N2, table) table = meta.tables['OPERATIONS'] alter_column(COLUMN_N3_OLD, table=table, name=COLUMN_N3_NEW.name) try: meta.bind = migrate_engine session = SA_SESSIONMAKER() session.execute(text("""UPDATE "DATA_TYPES" SET module='tvb.datatypes.region_mapping' WHERE "type" = 'RegionMapping' """)) session.execute(text("""UPDATE "DATA_TYPES" SET module='tvb.datatypes.local_connectivity' WHERE "type" = 'LocalConnectivity' """)) session.execute(text("""UPDATE "DATA_TYPES" SET module='tvb.datatypes.cortex' WHERE "type" = 'Cortex' """)) session.commit() session.close() except Exception: LOGGER.exception("Cold not update datatypes") raise
def downgrade(migrate_engine): """Operations to reverse the above upgrade go here.""" meta.bind = migrate_engine table = meta.tables['USERS'] create_column(COLUMN_N1, table) table = meta.tables['BURST_CONFIGURATIONS'] create_column(COLUMN_N2, table) table = meta.tables['OPERATIONS'] alter_column(COLUMN_N3_NEW, table=table, name=COLUMN_N3_OLD.name) try: meta.bind = migrate_engine session = SA_SESSIONMAKER() session.execute( text( """UPDATE "DATA_TYPES" SET module='tvb.datatypes.surfaces' WHERE "type" = 'RegionMapping' """ )) session.execute( text( """UPDATE "DATA_TYPES" SET module='tvb.datatypes.surfaces' WHERE "type" = 'LocalConnectivity' """ )) session.execute( text( """UPDATE "DATA_TYPES" SET module='tvb.datatypes.surfaces' WHERE "type" = 'Cortex' """ )) session.commit() session.close() except Exception: LOGGER.exception("Cold not update datatypes") raise
def downgrade(migrate_engine): """Operations to reverse the above upgrade go here.""" meta.bind = migrate_engine table = meta.tables['USERS'] create_column(COLUMN_N1, table) table = meta.tables['BURST_CONFIGURATIONS'] create_column(COLUMN_N2, table) table = meta.tables['OPERATIONS'] alter_column(COLUMN_N3_NEW, table=table, name=COLUMN_N3_OLD.name) try: meta.bind = migrate_engine session = SA_SESSIONMAKER() session.execute(text("""UPDATE "DATA_TYPES" SET module='tvb.datatypes.surfaces' WHERE "type" = 'RegionMapping' """)) session.execute(text("""UPDATE "DATA_TYPES" SET module='tvb.datatypes.surfaces' WHERE "type" = 'LocalConnectivity' """)) session.execute(text("""UPDATE "DATA_TYPES" SET module='tvb.datatypes.surfaces' WHERE "type" = 'Cortex' """)) session.commit() session.close() except Exception: LOGGER.exception("Cold not update datatypes") raise