__init__.py
60 lines
| 2.0 KiB
| text/x-python
|
PythonLexer
r833 | # -*- coding: utf-8 -*- | |||
r1206 | # This program is free software: you can redistribute it and/or modify | |||
# it under the terms of the GNU General Public License as published by | ||||
# the Free Software Foundation, either version 3 of the License, or | ||||
# (at your option) any later version. | ||||
r1203 | # | |||
r833 | # This program is distributed in the hope that it will be useful, | |||
# but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
# GNU General Public License for more details. | ||||
r1203 | # | |||
r833 | # You should have received a copy of the GNU General Public License | |||
r1206 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |||
|
r4116 | """ | ||
rhodecode.lib.dbmigrate.versions.__init__ | ||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ | ||||
Package containing new versions of database models | ||||
:created_on: Dec 11, 2010 | ||||
:author: marcink | ||||
:copyright: (c) 2013 RhodeCode GmbH. | ||||
:license: GPLv3, see LICENSE for more details. | ||||
""" | ||||
r3148 | from sqlalchemy import * | |||
from sqlalchemy.exc import DatabaseError | ||||
from sqlalchemy.orm import relation, backref, class_mapper, joinedload | ||||
from sqlalchemy.ext.declarative import declarative_base | ||||
|
r4116 | from sqlalchemy.orm import scoped_session, sessionmaker | ||
r3148 | from rhodecode.lib.dbmigrate.migrate import * | |||
from rhodecode.lib.dbmigrate.migrate.changeset import * | ||||
from rhodecode.model import meta | ||||
|
r4116 | 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) | ||||
r3148 | def _reset_base(migrate_engine): | |||
## RESET COMPLETLY THE metadata for sqlalchemy to use previous declared Base | ||||
Base = declarative_base() | ||||
Base.metadata.clear() | ||||
Base.metadata = MetaData() | ||||
Base.metadata.bind = migrate_engine | ||||
|
r4116 | |||
# new session and base | ||||
#meta.Session = scoped_session(sessionmaker(expire_on_commit=True,)) | ||||
#meta.Session.configure(bind=migrate_engine) | ||||
r3148 | meta.Base = Base | |||
|
r4116 | |||
notify('SQLA BASE RESET !') | ||||