:copyright: (c) 2013 RhodeCode GmbH, and others.

:license: GPLv3, see LICENSE.md for more details.

"""

from sqlalchemy import *

from sqlalchemy.ext.declarative import declarative_base

from kallithea.lib.dbmigrate.migrate import *

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

from kallithea.model import meta

def notify(msg, caps=True):

    """

    Notification for migrations messages

    """

    ml = len(msg) + (4 * 2)

    formatted_msg = ('\n%s\n*** %s ***\n%s' % ('*' * ml, msg, '*' * ml))

    if caps:

        formatted_msg = formatted_msg.upper()

    print(formatted_msg)

def _reset_base(migrate_engine):

    Base = declarative_base()

    Base.metadata.clear()

    Base.metadata = MetaData()

    Base.metadata.bind = migrate_engine

    # new session and base

    #meta.Session = scoped_session(sessionmaker(expire_on_commit=True,))

    #meta.Session.configure(bind=migrate_engine)

    meta.Base = Base

    notify('SQLA BASE RESET !')