##// END OF EJS Templates
fold
fold

File last commit:

r5042:4c5af799 default
r5236:673ed320 default
Show More
112_version_4_23_0.py
78 lines | 2.1 KiB | text/x-python | PythonLexer
import logging
from sqlalchemy import *
from alembic.migration import MigrationContext
from alembic.operations import Operations
from rhodecode.lib.dbmigrate.versions import _reset_base
from rhodecode.model import meta, 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_20_0_0 as db
init_model_encryption(db)
context = MigrationContext.configure(migrate_engine.connect())
op = Operations(context)
table = db.RepoReviewRule.__table__
with op.batch_alter_table(table.name) as batch_op:
new_column = Column('pr_author', UnicodeText().with_variant(UnicodeText(255), 'mysql'), nullable=True)
batch_op.add_column(new_column)
new_column = Column('commit_author', UnicodeText().with_variant(UnicodeText(255), 'mysql'), nullable=True)
batch_op.add_column(new_column)
_migrate_review_flags_to_new_cols(op, meta.Session)
def downgrade(migrate_engine):
meta = MetaData()
meta.bind = migrate_engine
def fixups(models, _SESSION):
pass
def _migrate_review_flags_to_new_cols(op, session):
# set defaults for pr_author
query = text(
'UPDATE repo_review_rules SET pr_author = :val'
).bindparams(val='no_rule')
op.execute(query)
# set defaults for commit_author
query = text(
'UPDATE repo_review_rules SET commit_author = :val'
).bindparams(val='no_rule')
op.execute(query)
session().commit()
# now change the flags to forbid based on
# forbid_author_to_review, forbid_commit_author_to_review
query = text(
'UPDATE repo_review_rules SET pr_author = :val WHERE forbid_author_to_review = TRUE'
).bindparams(val='forbid_pr_author')
op.execute(query)
query = text(
'UPDATE repo_review_rules SET commit_author = :val WHERE forbid_commit_author_to_review = TRUE'
).bindparams(val='forbid_commit_author')
op.execute(query)
session().commit()