authn: Data migration for renaming 'container' to 'headers'.

Rename the 'auth_plugins' setting and all 'auth_container_%s' settings.

rhodecode/lib/dbmigrate/versions/052_version_4_1_0.py

@@ -0,0 +1,80 b''
+# -*- coding: utf-8 -*-
+
+import logging
+
+from sqlalchemy.orm.attributes import flag_modified
+
+from rhodecode.lib.dbmigrate.versions import _reset_base
+from rhodecode.model import init_model_encryption, meta
+
+log = logging.getLogger(__name__)
+
+
+def upgrade(migrate_engine):
+    """
+    Upgrade operations go here.
+    Don't create your own engine; bind migrate_engine to your metadata
+    """
+    _reset_base(migrate_engine)
+    from rhodecode.lib.dbmigrate.schema import db_3_7_0_0
+    init_model_encryption(db_3_7_0_0)
+    fixups(db_3_7_0_0, meta.Session)
+
+
+def downgrade(migrate_engine):
+    pass
+
+
+AUTH_PLUGINS_SETTING = "auth_plugins"
+
+PLUGIN_RENAME_MAP = {
+    'egg:rhodecode-enterprise-ce#container': 'egg:rhodecode-enterprise-ce#headers',
+}
+
+SETTINGS_RENAME_MAP = {
+    'auth_container_cache_ttl': 'auth_headers_cache_ttl',
+    'auth_container_clean_username': 'auth_headers_clean_username',
+    'auth_container_enabled': 'auth_headers_enabled',
+    'auth_container_fallback_header': 'auth_headers_fallback_header',
+    'auth_container_header': 'auth_headers_header',
+}
+
+
+def rename_plugins(models, Session):
+    query = models.RhodeCodeSetting.query().filter(
+        models.RhodeCodeSetting.app_settings_name == AUTH_PLUGINS_SETTING)
+    plugin_setting = query.scalar()
+    plugins = plugin_setting.app_settings_value
+
+    new_plugins = []
+
+    for plugin_id in plugins:
+        new_plugin_id = PLUGIN_RENAME_MAP.get(plugin_id, None)
+        if new_plugin_id:
+            new_plugins.append(new_plugin_id)
+        else:
+            new_plugins.append(plugin_id)
+
+    plugin_setting.app_settings_value = ','.join(new_plugins)
+
+    log.info("Rename of auth plugin IDs")
+    log.info("Original setting value: %s", plugins)
+    log.info("New setting value: %s", new_plugins)
+
+
+def rename_plugin_settings(models, Session):
+    for old_name, new_name in SETTINGS_RENAME_MAP.items():
+        query = models.RhodeCodeSetting.query().filter(
+            models.RhodeCodeSetting.app_settings_name == old_name)
+        setting = query.scalar()
+        if setting:
+            setting.app_settings_name = new_name
+            log.info(
+                'Rename of plugin setting "%s" to "%s"', old_name, new_name)
+
+
+def fixups(models, Session):
+    rename_plugins(models, Session)
+    rename_plugin_settings(models, Session)
+
+    Session().commit()

rhodecode/__init__.py

@@ -1,58 +1,58 b''
 # -*- coding: utf-8 -*-
 
 # Copyright (C) 2010-2016  RhodeCode GmbH
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License, version 3
 # (only), as published by the Free Software Foundation.
 #
 # 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 Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 # This program is dual-licensed. If you wish to learn more about the
 # RhodeCode Enterprise Edition, including its added features, Support services,
 # and proprietary license terms, please see https://rhodecode.com/licenses/
 
 """
 
 RhodeCode, a web based repository management software
 versioning implementation: http://www.python.org/dev/peps/pep-0386/
 """
 
 import os
 import sys
 import platform
 
 VERSION = tuple(open(os.path.join(
     os.path.dirname(__file__), 'VERSION')).read().split('.'))
 
 BACKENDS = {
     'hg': 'Mercurial repository',
     'git': 'Git repository',
     'svn': 'Subversion repository',
 }
 
 CELERY_ENABLED = False
 CELERY_EAGER = False
 
 # link to config for pylons
 CONFIG = {}
 
 # Linked module for extensions
 EXTENSIONS = {}
 
 __version__ = ('.'.join((str(each) for each in VERSION[:3])))
-__dbversion__ = 51  # defines current db version for migrations
+__dbversion__ = 52  # defines current db version for migrations
 __platform__ = platform.system()
 __license__ = 'AGPLv3, and Commercial License'
 __author__ = 'RhodeCode GmbH'
 __url__ = 'http://rhodecode.com'
 
 is_windows = __platform__ in ['Windows']
 is_unix = not is_windows
 is_test = False