import logging from alembic.migration import MigrationContext from alembic.operations import Operations from sqlalchemy import Column, String from rhodecode.lib.dbmigrate.versions import _reset_base from rhodecode.model import init_model_encryption log = logging.getLogger(__name__) def upgrade(migrate_engine): """ Upgrade operations go here. Don't create your own engine; bind migrate_engine to your metadata """ _reset_base(migrate_engine) from rhodecode.lib.dbmigrate.schema import db_4_16_0_2 init_model_encryption(db_4_16_0_2) context = MigrationContext.configure(migrate_engine.connect()) op = Operations(context) cache_key = db_4_16_0_2.CacheKey.__table__ with op.batch_alter_table(cache_key.name) as batch_op: batch_op.add_column( Column("cache_state_uid", String(255), nullable=True, unique=None, default=None)) def downgrade(migrate_engine): pass