kallithea Files · rhodecode/lib/dbmigrate/versions/012_version_1_7

Files @ 244f184f5fc3

Branch filter:

Location: kallithea/rhodecode/lib/dbmigrate/versions/012_version_1_7_0.py

244f184f5fc3 2.3 KiB text/x-python Show Annotation Show as Raw Download as Raw

Marcin Kuzminski

moved around some code in changeset controllers to properly log which function was decorated.
Before it was always index, and permission were checked twice giving unnneded overhead

import logging
import datetime

from sqlalchemy import *
from sqlalchemy.exc import DatabaseError
from sqlalchemy.orm import relation, backref, class_mapper, joinedload
from sqlalchemy.orm.session import Session
from sqlalchemy.ext.declarative import declarative_base

from rhodecode.lib.dbmigrate.migrate import *
from rhodecode.lib.dbmigrate.migrate.changeset import *

from rhodecode.model.meta import Base
from rhodecode.model import meta
from rhodecode.lib.dbmigrate.versions import _reset_base

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)

    #==========================================================================
    # UserUserGroupToPerm
    #==========================================================================
    from rhodecode.lib.dbmigrate.schema.db_1_7_0 import UserUserGroupToPerm
    tbl = UserUserGroupToPerm.__table__
    tbl.create()

    #==========================================================================
    # UserGroupUserGroupToPerm
    #==========================================================================
    from rhodecode.lib.dbmigrate.schema.db_1_7_0 import UserGroupUserGroupToPerm
    tbl = UserGroupUserGroupToPerm.__table__
    tbl.create()

    #==========================================================================
    # UserGroup
    #==========================================================================
    from rhodecode.lib.dbmigrate.schema.db_1_7_0 import UserGroup
    tbl = UserGroup.__table__
    user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=False, default=None)
    # create username column
    user_id.create(table=tbl)

    #==========================================================================
    # UserGroup
    #==========================================================================
    from rhodecode.lib.dbmigrate.schema.db_1_7_0 import RepoGroup
    tbl = RepoGroup.__table__
    user_id = Column("user_id", Integer(), ForeignKey('users.user_id'), nullable=True, unique=False, default=None)
    # create username column
    user_id.create(table=tbl)


def downgrade(migrate_engine):
    meta = MetaData()
    meta.bind = migrate_engine