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

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``

    #==========================================================================

    Group().__table__.create()

    #==========================================================================

    # Add table `group_to_perm`

    #==========================================================================

    from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UserRepoGroupToPerm

    UserRepoGroupToPerm().__table__.create()

    #==========================================================================

    # Add table `users_groups`

    #==========================================================================

    from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroup

    UsersGroup().__table__.create()

    #==========================================================================

    # Add table `users_groups_members`

    #==========================================================================

    from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroupMember

    UsersGroupMember().__table__.create()

    #==========================================================================

    # Add table `users_group_repo_to_perm`

    #==========================================================================

    from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroupRepoToPerm

    UsersGroupRepoToPerm().__table__.create()

    #==========================================================================

    # Add table `users_group_to_perm`

    #==========================================================================

    from rhodecode.lib.dbmigrate.schema.db_1_2_0 import UsersGroupToPerm

    UsersGroupToPerm().__table__.create()

    #==========================================================================

    # Upgrade of `users` table

    #==========================================================================

    from rhodecode.lib.dbmigrate.schema.db_1_2_0 import User

    #add column

    ldap_dn = Column("ldap_dn", String(length=255, convert_unicode=False, assert_unicode=None), nullable=True, unique=None, default=None)

    ldap_dn.create(User().__table__)

    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__)

    #remove old column

    is_ldap = Column("is_ldap", Boolean(), nullable=False, unique=None, default=False)

    is_ldap.drop(User().__table__)