003_version_1_2_0.py
119 lines
| 4.8 KiB
| text/x-python
|
PythonLexer
r900 | 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 | ||||
r907 | ||||
from rhodecode.lib.dbmigrate.migrate import * | ||||
from rhodecode.lib.dbmigrate.migrate.changeset import * | ||||
r900 | from rhodecode.model.meta import Base | |||
log = logging.getLogger(__name__) | ||||
def upgrade(migrate_engine): | ||||
r1203 | """ Upgrade operations go here. | |||
r900 | Don't create your own engine; bind migrate_engine to your metadata | |||
""" | ||||
#========================================================================== | ||||
# Add table `groups`` | ||||
#========================================================================== | ||||
r1632 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import RepoGroup as Group | |||
r1026 | Group().__table__.create() | |||
r900 | ||||
#========================================================================== | ||||
# Add table `group_to_perm` | ||||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UserRepoGroupToPerm | |||
UserRepoGroupToPerm().__table__.create() | ||||
r1023 | ||||
#========================================================================== | ||||
# Add table `users_groups` | ||||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroup | |||
r1026 | UsersGroup().__table__.create() | |||
r1023 | ||||
#========================================================================== | ||||
# Add table `users_groups_members` | ||||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroupMember | |||
r1023 | UsersGroupMember().__table__.create() | |||
#========================================================================== | ||||
r1271 | # Add table `users_group_repo_to_perm` | |||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroupRepoToPerm | |||
r1271 | UsersGroupRepoToPerm().__table__.create() | |||
#========================================================================== | ||||
r1023 | # Add table `users_group_to_perm` | |||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroupToPerm | |||
r1023 | UsersGroupToPerm().__table__.create() | |||
#========================================================================== | ||||
# Upgrade of `users` table | ||||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import User | |||
r1023 | ||||
#add column | ||||
r1762 | ldap_dn = Column("ldap_dn", String(length=255, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None) | |||
r1023 | ldap_dn.create(User().__table__) | |||
r1133 | api_key = Column("api_key", String(length=255, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None) | |||
api_key.create(User().__table__) | ||||
r1023 | ||||
#remove old column | ||||
is_ldap = Column("is_ldap", Boolean(), nullable=False, unique=None, default=False) | ||||
is_ldap.drop(User().__table__) | ||||
r900 | ||||
#========================================================================== | ||||
# Upgrade of `repositories` table | ||||
r1203 | #========================================================================== | |||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import Repository | |||
r900 | ||||
r1509 | #ADD clone_uri column# | |||
clone_uri = Column("clone_uri", String(length=255, convert_unicode=False, | ||||
assert_unicode=None), | ||||
nullable=True, unique=False, default=None) | ||||
clone_uri.create(Repository().__table__) | ||||
r1023 | #ADD downloads column# | |||
enable_downloads = Column("downloads", Boolean(), nullable=True, unique=None, default=True) | ||||
enable_downloads.create(Repository().__table__) | ||||
r1442 | #ADD column created_on | |||
created_on = Column('created_on', DateTime(timezone=False), nullable=True, | ||||
unique=None, default=datetime.datetime.now) | ||||
created_on.create(Repository().__table__) | ||||
r900 | #ADD group_id column# | |||
r908 | group_id = Column("group_id", Integer(), ForeignKey('groups.group_id'), | |||
r900 | nullable=True, unique=False, default=None) | |||
r908 | group_id.create(Repository().__table__) | |||
r900 | ||||
r1133 | ||||
r1442 | #========================================================================== | |||
# Upgrade of `user_followings` table | ||||
#========================================================================== | ||||
r1631 | from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UserFollowing | |||
r1509 | ||||
follows_from = Column('follows_from', DateTime(timezone=False), | ||||
nullable=True, unique=None, | ||||
default=datetime.datetime.now) | ||||
follows_from.create(UserFollowing().__table__) | ||||
r1442 | ||||
r900 | return | |||
def downgrade(migrate_engine): | ||||
meta = MetaData() | ||||
meta.bind = migrate_engine | ||||