import logging from sqlalchemy import * from alembic.migration import MigrationContext from alembic.operations import Operations from sqlalchemy import BigInteger 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_18_0_1 as db init_model_encryption(db) context = MigrationContext.configure(migrate_engine.connect()) op = Operations(context) pull_requests = db.PullRequest.__table__ with op.batch_alter_table(pull_requests.name) as batch_op: new_column = Column( 'last_merge_metadata', db.JsonType(dialect_map=dict(mysql=UnicodeText(16384)))) batch_op.add_column(new_column) pull_request_version = db.PullRequestVersion.__table__ with op.batch_alter_table(pull_request_version.name) as batch_op: new_column = Column( 'last_merge_metadata', db.JsonType(dialect_map=dict(mysql=UnicodeText(16384)))) batch_op.add_column(new_column) def downgrade(migrate_engine): meta = MetaData() meta.bind = migrate_engine def fixups(models, _SESSION): pass