|
|
import logging
|
|
|
import datetime
|
|
|
|
|
|
from sqlalchemy import *
|
|
|
from sqlalchemy.exc import DatabaseError
|
|
|
from sqlalchemy.orm import relation, backref, class_mapper
|
|
|
from sqlalchemy.orm.session import Session
|
|
|
|
|
|
from rhodecode.lib.dbmigrate.migrate import *
|
|
|
from rhodecode.lib.dbmigrate.migrate.changeset import *
|
|
|
|
|
|
from rhodecode.model.meta import Base
|
|
|
from rhodecode.model.db import BaseModel
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def upgrade(migrate_engine):
|
|
|
""" Upgrade operations go here.
|
|
|
Don't create your own engine; bind migrate_engine to your metadata
|
|
|
"""
|
|
|
|
|
|
#==========================================================================
|
|
|
# Add table `groups``
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import Group
|
|
|
Group().__table__.create()
|
|
|
|
|
|
#==========================================================================
|
|
|
# Add table `group_to_perm`
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import GroupToPerm
|
|
|
GroupToPerm().__table__.create()
|
|
|
|
|
|
#==========================================================================
|
|
|
# Add table `users_groups`
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import UsersGroup
|
|
|
UsersGroup().__table__.create()
|
|
|
|
|
|
#==========================================================================
|
|
|
# Add table `users_groups_members`
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import UsersGroupMember
|
|
|
UsersGroupMember().__table__.create()
|
|
|
|
|
|
#==========================================================================
|
|
|
# Add table `users_group_to_perm`
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import UsersGroupToPerm
|
|
|
UsersGroupToPerm().__table__.create()
|
|
|
|
|
|
|
|
|
#==========================================================================
|
|
|
# Upgrade of `users` table
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import User
|
|
|
|
|
|
#add column
|
|
|
ldap_dn = Column("ldap_dn", String(length=None, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)
|
|
|
ldap_dn.create(User().__table__)
|
|
|
|
|
|
|
|
|
#remove old column
|
|
|
is_ldap = Column("is_ldap", Boolean(), nullable=False, unique=None, default=False)
|
|
|
is_ldap.drop(User().__table__)
|
|
|
|
|
|
|
|
|
#==========================================================================
|
|
|
# Upgrade of `repositories` table
|
|
|
#==========================================================================
|
|
|
from rhodecode.model.db import Repository
|
|
|
|
|
|
#ADD downloads column#
|
|
|
enable_downloads = Column("downloads", Boolean(), nullable=True, unique=None, default=True)
|
|
|
enable_downloads.create(Repository().__table__)
|
|
|
|
|
|
#ADD group_id column#
|
|
|
group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'),
|
|
|
nullable=True, unique=False, default=None)
|
|
|
|
|
|
group_id.create(Repository().__table__)
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
def downgrade(migrate_engine):
|
|
|
meta = MetaData()
|
|
|
meta.bind = migrate_engine
|
|
|
|
|
|
|
|
|
|
