__init__.py
101 lines
| 3.4 KiB
| text/x-python
|
PythonLexer
r833 | # -*- coding: utf-8 -*- | |||
""" | ||||
rhodecode.lib.dbmigrate.__init__ | ||||
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ | ||||
Database migration modules | ||||
:created_on: Dec 11, 2010 | ||||
:author: marcink | ||||
:copyright: (C) 2009-2010 Marcin Kuzminski <marcin@python-works.com> | ||||
:license: GPLv3, see COPYING for more details. | ||||
""" | ||||
# 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; version 2 | ||||
# of the License or (at your opinion) any later version of the license. | ||||
# | ||||
# 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. | ||||
# | ||||
# You should have received a copy of the GNU General Public License | ||||
# along with this program; if not, write to the Free Software | ||||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, | ||||
# MA 02110-1301, USA. | ||||
r835 | import logging | |||
from sqlalchemy import engine_from_config | ||||
r836 | from rhodecode import __dbversion__ | |||
from rhodecode.lib.dbmigrate.migrate.versioning import api | ||||
r835 | from rhodecode.lib.dbmigrate.migrate.exceptions import \ | |||
DatabaseNotControlledError | ||||
r833 | from rhodecode.lib.utils import BasePasterCommand, Command, add_cache | |||
r837 | from rhodecode.lib.db_manage import DbManage | |||
r833 | ||||
r835 | log = logging.getLogger(__name__) | |||
r833 | ||||
class UpgradeDb(BasePasterCommand): | ||||
"""Command used for paster to upgrade our database to newer version | ||||
""" | ||||
max_args = 1 | ||||
min_args = 1 | ||||
usage = "CONFIG_FILE" | ||||
summary = "Upgrades current db to newer version given configuration file" | ||||
group_name = "RhodeCode" | ||||
parser = Command.standard_parser(verbose=True) | ||||
def command(self): | ||||
from pylons import config | ||||
r836 | ||||
r833 | add_cache(config) | |||
r835 | ||||
r837 | ||||
r835 | ||||
r836 | repository_path = 'rhodecode/lib/dbmigrate' | |||
db_uri = config['sqlalchemy.db1.url'] | ||||
r833 | ||||
r835 | try: | |||
r836 | curr_version = api.db_version(db_uri, repository_path) | |||
r835 | msg = ('Found current database under version' | |||
' control with version %s' % curr_version) | ||||
except (RuntimeError, DatabaseNotControlledError), e: | ||||
r836 | curr_version = 1 | |||
r835 | msg = ('Current database is not under version control setting' | |||
' as version %s' % curr_version) | ||||
r836 | api.version_control(db_uri, repository_path, curr_version) | |||
r837 | self.notify_msg(msg) | |||
r835 | ||||
r836 | #now we have our dbversion we can do upgrade | |||
r837 | self.notify_msg('attempting to do database upgrade to version %s' \ | |||
% __dbversion__) | ||||
api.upgrade(db_uri, repository_path, __dbversion__) | ||||
self.notify_msg('Schema upgrade completed') | ||||
r836 | ||||
r837 | #we need to make now some extra operations into database | |||
self.notify_msg('Prociding with database updates') | ||||
dbmanage = DbManage(log_sql=True, dbconf=db_uri, | ||||
root=config['here'], tests=False) | ||||
self.notify_msg('Patching repo paths for newer version of rhodecode') | ||||
dbmanage.fix_repo_paths() | ||||
self.notify_msg('Changing ui settings') | ||||
dbmanage.create_ui_settings() | ||||
r833 | ||||
def update_parser(self): | ||||
self.parser.add_option('--sql', | ||||
action='store_true', | ||||
dest='just_sql', | ||||
help="Prints upgrade sql for further investigation", | ||||
default=False) | ||||