settings.py
889 lines
| 31.3 KiB
| text/x-python
|
PythonLexer
r5088 | # Copyright (C) 2010-2023 RhodeCode GmbH | |||
r1 | # | |||
# 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/ | ||||
r1682 | import os | |||
r4834 | import re | |||
r1 | import logging | |||
r4834 | import time | |||
r4945 | import functools | |||
r1 | from collections import namedtuple | |||
r4945 | ||||
r5070 | from pyramid.threadlocal import get_current_request | |||
r1 | ||||
r2932 | from rhodecode.lib import rc_cache | |||
r5070 | from rhodecode.lib.hash_utils import sha1_safe | |||
r5098 | from rhodecode.lib.html_filters import sanitize_html | |||
r1 | from rhodecode.lib.utils2 import ( | |||
Optional, AttributeDict, safe_str, remove_prefix, str2bool) | ||||
r754 | from rhodecode.lib.vcs.backends import base | |||
r4833 | from rhodecode.lib.statsd_client import StatsdClient | |||
r1 | from rhodecode.model import BaseModel | |||
from rhodecode.model.db import ( | ||||
r4834 | RepoRhodeCodeUi, RepoRhodeCodeSetting, RhodeCodeUi, RhodeCodeSetting) | |||
r1 | from rhodecode.model.meta import Session | |||
log = logging.getLogger(__name__) | ||||
UiSetting = namedtuple( | ||||
'UiSetting', ['section', 'key', 'value', 'active']) | ||||
SOCIAL_PLUGINS_LIST = ['github', 'bitbucket', 'twitter', 'google'] | ||||
class SettingNotFound(Exception): | ||||
r2180 | def __init__(self, setting_id): | |||
r5095 | msg = f'Setting `{setting_id}` is not found' | |||
super().__init__(msg) | ||||
r1 | ||||
class SettingsModel(BaseModel): | ||||
BUILTIN_HOOKS = ( | ||||
RhodeCodeUi.HOOK_REPO_SIZE, RhodeCodeUi.HOOK_PUSH, | ||||
r1461 | RhodeCodeUi.HOOK_PRE_PUSH, RhodeCodeUi.HOOK_PRETX_PUSH, | |||
r1755 | RhodeCodeUi.HOOK_PULL, RhodeCodeUi.HOOK_PRE_PULL, | |||
RhodeCodeUi.HOOK_PUSH_KEY,) | ||||
r1 | HOOKS_SECTION = 'hooks' | |||
def __init__(self, sa=None, repo=None): | ||||
self.repo = repo | ||||
self.UiDbModel = RepoRhodeCodeUi if repo else RhodeCodeUi | ||||
self.SettingsDbModel = ( | ||||
RepoRhodeCodeSetting if repo else RhodeCodeSetting) | ||||
r5095 | super().__init__(sa) | |||
r1 | ||||
r5144 | def get_keyname(self, key_name, prefix='rhodecode_'): | |||
return f'{prefix}{key_name}' | ||||
r1 | def get_ui_by_key(self, key): | |||
q = self.UiDbModel.query() | ||||
q = q.filter(self.UiDbModel.ui_key == key) | ||||
q = self._filter_by_repo(RepoRhodeCodeUi, q) | ||||
return q.scalar() | ||||
def get_ui_by_section(self, section): | ||||
q = self.UiDbModel.query() | ||||
q = q.filter(self.UiDbModel.ui_section == section) | ||||
q = self._filter_by_repo(RepoRhodeCodeUi, q) | ||||
return q.all() | ||||
def get_ui_by_section_and_key(self, section, key): | ||||
q = self.UiDbModel.query() | ||||
q = q.filter(self.UiDbModel.ui_section == section) | ||||
q = q.filter(self.UiDbModel.ui_key == key) | ||||
q = self._filter_by_repo(RepoRhodeCodeUi, q) | ||||
return q.scalar() | ||||
def get_ui(self, section=None, key=None): | ||||
q = self.UiDbModel.query() | ||||
q = self._filter_by_repo(RepoRhodeCodeUi, q) | ||||
if section: | ||||
q = q.filter(self.UiDbModel.ui_section == section) | ||||
if key: | ||||
q = q.filter(self.UiDbModel.ui_key == key) | ||||
# TODO: mikhail: add caching | ||||
result = [ | ||||
UiSetting( | ||||
section=safe_str(r.ui_section), key=safe_str(r.ui_key), | ||||
value=safe_str(r.ui_value), active=r.ui_active | ||||
) | ||||
for r in q.all() | ||||
] | ||||
return result | ||||
def get_builtin_hooks(self): | ||||
q = self.UiDbModel.query() | ||||
q = q.filter(self.UiDbModel.ui_key.in_(self.BUILTIN_HOOKS)) | ||||
return self._get_hooks(q) | ||||
def get_custom_hooks(self): | ||||
q = self.UiDbModel.query() | ||||
q = q.filter(~self.UiDbModel.ui_key.in_(self.BUILTIN_HOOKS)) | ||||
return self._get_hooks(q) | ||||
def create_ui_section_value(self, section, val, key=None, active=True): | ||||
new_ui = self.UiDbModel() | ||||
new_ui.ui_section = section | ||||
new_ui.ui_value = val | ||||
new_ui.ui_active = active | ||||
r3625 | repository_id = '' | |||
r1 | if self.repo: | |||
repo = self._get_repo(self.repo) | ||||
repository_id = repo.repo_id | ||||
new_ui.repository_id = repository_id | ||||
if not key: | ||||
# keys are unique so they need appended info | ||||
if self.repo: | ||||
r5070 | key = sha1_safe(f'{section}{val}{repository_id}') | |||
r1 | else: | |||
r5070 | key = sha1_safe(f'{section}{val}') | |||
r1 | ||||
new_ui.ui_key = key | ||||
Session().add(new_ui) | ||||
return new_ui | ||||
def create_or_update_hook(self, key, value): | ||||
ui = ( | ||||
self.get_ui_by_section_and_key(self.HOOKS_SECTION, key) or | ||||
self.UiDbModel()) | ||||
ui.ui_section = self.HOOKS_SECTION | ||||
ui.ui_active = True | ||||
ui.ui_key = key | ||||
ui.ui_value = value | ||||
if self.repo: | ||||
repo = self._get_repo(self.repo) | ||||
repository_id = repo.repo_id | ||||
ui.repository_id = repository_id | ||||
Session().add(ui) | ||||
return ui | ||||
def delete_ui(self, id_): | ||||
ui = self.UiDbModel.get(id_) | ||||
if not ui: | ||||
r2180 | raise SettingNotFound(id_) | |||
r1 | Session().delete(ui) | |||
def get_setting_by_name(self, name): | ||||
q = self._get_settings_query() | ||||
q = q.filter(self.SettingsDbModel.app_settings_name == name) | ||||
return q.scalar() | ||||
def create_or_update_setting( | ||||
r5144 | self, name, val: Optional | str = Optional(''), type_: Optional | str = Optional('unicode')): | |||
r1 | """ | |||
r5144 | Creates or updates RhodeCode setting. If updates are triggered, it will | |||
r4835 | only update parameters that are explicitly set Optional instance will | |||
r1 | be skipped | |||
:param name: | ||||
:param val: | ||||
:param type_: | ||||
:return: | ||||
""" | ||||
res = self.get_setting_by_name(name) | ||||
repo = self._get_repo(self.repo) if self.repo else None | ||||
if not res: | ||||
val = Optional.extract(val) | ||||
type_ = Optional.extract(type_) | ||||
args = ( | ||||
(repo.repo_id, name, val, type_) | ||||
if repo else (name, val, type_)) | ||||
res = self.SettingsDbModel(*args) | ||||
else: | ||||
if self.repo: | ||||
res.repository_id = repo.repo_id | ||||
res.app_settings_name = name | ||||
if not isinstance(type_, Optional): | ||||
# update if set | ||||
res.app_settings_type = type_ | ||||
if not isinstance(val, Optional): | ||||
# update if set | ||||
res.app_settings_value = val | ||||
r506 | Session().add(res) | |||
r1 | return res | |||
r4833 | def get_cache_region(self): | |||
repo = self._get_repo(self.repo) if self.repo else None | ||||
r5106 | cache_key = f"repo.v1.{repo.repo_id}" if repo else "repo.v1.ALL" | |||
r5070 | cache_namespace_uid = f'cache_settings.{cache_key}' | |||
r4833 | region = rc_cache.get_or_create_region('cache_general', cache_namespace_uid) | |||
r5070 | return region, cache_namespace_uid | |||
r276 | ||||
r5070 | def invalidate_settings_cache(self, hard=False): | |||
region, namespace_key = self.get_cache_region() | ||||
log.debug('Invalidation cache [%s] region %s for cache_key: %s', | ||||
'invalidate_settings_cache', region, namespace_key) | ||||
r4200 | ||||
r5070 | # we use hard cleanup if invalidation is sent | |||
rc_cache.clear_cache_namespace(region, namespace_key, method=rc_cache.CLEAR_DELETE) | ||||
r4200 | ||||
r5070 | def get_cache_call_method(self, cache=True): | |||
r4833 | region, cache_key = self.get_cache_region() | |||
r1669 | ||||
r2892 | @region.conditional_cache_on_arguments(condition=cache) | |||
r2883 | def _get_all_settings(name, key): | |||
r259 | q = self._get_settings_query() | |||
if not q: | ||||
raise Exception('Could not get application settings !') | ||||
settings = { | ||||
r5144 | self.get_keyname(res.app_settings_name): res.app_settings_value | |||
r4833 | for res in q | |||
r259 | } | |||
return settings | ||||
r5070 | return _get_all_settings | |||
def get_all_settings(self, cache=False, from_request=True): | ||||
# defines if we use GLOBAL, or PER_REPO | ||||
repo = self._get_repo(self.repo) if self.repo else None | ||||
r5144 | # initially try the request context; this is the fastest | |||
r5070 | # we only fetch global config, NOT for repo-specific | |||
if from_request and not repo: | ||||
request = get_current_request() | ||||
if request and hasattr(request, 'call_context') and hasattr(request.call_context, 'rc_config'): | ||||
rc_config = request.call_context.rc_config | ||||
if rc_config: | ||||
return rc_config | ||||
_region, cache_key = self.get_cache_region() | ||||
_get_all_settings = self.get_cache_call_method(cache=cache) | ||||
r259 | ||||
r4834 | start = time.time() | |||
result = _get_all_settings('rhodecode_settings', cache_key) | ||||
compute_time = time.time() - start | ||||
r4936 | log.debug('cached method:%s took %.4fs', _get_all_settings.__name__, compute_time) | |||
r4834 | ||||
r4833 | statsd = StatsdClient.statsd | |||
r4834 | if statsd: | |||
elapsed_time_ms = round(1000.0 * compute_time) # use ms only | ||||
statsd.timing("rhodecode_settings_timing.histogram", elapsed_time_ms, | ||||
use_decimals=False) | ||||
r2937 | ||||
r4835 | log.debug('Fetching app settings for key: %s took: %.4fs: cache: %s', cache_key, compute_time, cache) | |||
r1 | ||||
r2883 | return result | |||
r1 | ||||
def get_auth_settings(self): | ||||
q = self._get_settings_query() | ||||
q = q.filter( | ||||
self.SettingsDbModel.app_settings_name.startswith('auth_')) | ||||
rows = q.all() | ||||
auth_settings = { | ||||
row.app_settings_name: row.app_settings_value for row in rows} | ||||
return auth_settings | ||||
def get_auth_plugins(self): | ||||
auth_plugins = self.get_setting_by_name("auth_plugins") | ||||
return auth_plugins.app_settings_value | ||||
def get_default_repo_settings(self, strip_prefix=False): | ||||
q = self._get_settings_query() | ||||
q = q.filter( | ||||
self.SettingsDbModel.app_settings_name.startswith('default_')) | ||||
rows = q.all() | ||||
result = {} | ||||
for row in rows: | ||||
key = row.app_settings_name | ||||
if strip_prefix: | ||||
key = remove_prefix(key, prefix='default_') | ||||
result.update({key: row.app_settings_value}) | ||||
return result | ||||
def get_repo(self): | ||||
repo = self._get_repo(self.repo) | ||||
if not repo: | ||||
raise Exception( | ||||
r5144 | f'Repository `{self.repo}` cannot be found inside the database') | |||
r1 | return repo | |||
def _filter_by_repo(self, model, query): | ||||
if self.repo: | ||||
repo = self.get_repo() | ||||
query = query.filter(model.repository_id == repo.repo_id) | ||||
return query | ||||
def _get_hooks(self, query): | ||||
query = query.filter(self.UiDbModel.ui_section == self.HOOKS_SECTION) | ||||
query = self._filter_by_repo(RepoRhodeCodeUi, query) | ||||
return query.all() | ||||
def _get_settings_query(self): | ||||
q = self.SettingsDbModel.query() | ||||
return self._filter_by_repo(RepoRhodeCodeSetting, q) | ||||
def list_enabled_social_plugins(self, settings): | ||||
enabled = [] | ||||
for plug in SOCIAL_PLUGINS_LIST: | ||||
r5070 | if str2bool(settings.get(f'rhodecode_auth_{plug}_enabled')): | |||
r1 | enabled.append(plug) | |||
return enabled | ||||
def assert_repo_settings(func): | ||||
r4945 | @functools.wraps(func) | |||
r1 | def _wrapper(self, *args, **kwargs): | |||
if not self.repo_settings: | ||||
raise Exception('Repository is not specified') | ||||
return func(self, *args, **kwargs) | ||||
return _wrapper | ||||
class IssueTrackerSettingsModel(object): | ||||
INHERIT_SETTINGS = 'inherit_issue_tracker_settings' | ||||
SETTINGS_PREFIX = 'issuetracker_' | ||||
def __init__(self, sa=None, repo=None): | ||||
self.global_settings = SettingsModel(sa=sa) | ||||
self.repo_settings = SettingsModel(sa=sa, repo=repo) if repo else None | ||||
@property | ||||
def inherit_global_settings(self): | ||||
if not self.repo_settings: | ||||
return True | ||||
setting = self.repo_settings.get_setting_by_name(self.INHERIT_SETTINGS) | ||||
return setting.app_settings_value if setting else True | ||||
@inherit_global_settings.setter | ||||
def inherit_global_settings(self, value): | ||||
if self.repo_settings: | ||||
settings = self.repo_settings.create_or_update_setting( | ||||
self.INHERIT_SETTINGS, value, type_='bool') | ||||
Session().add(settings) | ||||
r5144 | def _get_keyname(self, key, uid, prefix='rhodecode_'): | |||
return f'{prefix}{self.SETTINGS_PREFIX}{key}_{uid}' | ||||
r1 | ||||
def _make_dict_for_settings(self, qs): | ||||
r5144 | prefix_match = self._get_keyname('pat', '',) | |||
r1 | ||||
issuetracker_entries = {} | ||||
# create keys | ||||
for k, v in qs.items(): | ||||
if k.startswith(prefix_match): | ||||
uid = k[len(prefix_match):] | ||||
issuetracker_entries[uid] = None | ||||
r3439 | def url_cleaner(input_str): | |||
input_str = input_str.replace('"', '').replace("'", '') | ||||
r5098 | input_str = sanitize_html(input_str, strip=True) | |||
r3439 | return input_str | |||
r1 | # populate | |||
for uid in issuetracker_entries: | ||||
r5144 | url_data = qs.get(self._get_keyname('url', uid)) | |||
r3439 | ||||
r5144 | pat = qs.get(self._get_keyname('pat', uid)) | |||
r4204 | try: | |||
pat_compiled = re.compile(r'%s' % pat) | ||||
except re.error: | ||||
pat_compiled = None | ||||
r1 | issuetracker_entries[uid] = AttributeDict({ | |||
r4204 | 'pat': pat, | |||
'pat_compiled': pat_compiled, | ||||
r3439 | 'url': url_cleaner( | |||
r5144 | qs.get(self._get_keyname('url', uid)) or ''), | |||
r5098 | 'pref': sanitize_html( | |||
r5144 | qs.get(self._get_keyname('pref', uid)) or ''), | |||
r2444 | 'desc': qs.get( | |||
r5144 | self._get_keyname('desc', uid)), | |||
r1 | }) | |||
r3439 | ||||
r1 | return issuetracker_entries | |||
def get_global_settings(self, cache=False): | ||||
""" | ||||
Returns list of global issue tracker settings | ||||
""" | ||||
defaults = self.global_settings.get_all_settings(cache=cache) | ||||
settings = self._make_dict_for_settings(defaults) | ||||
return settings | ||||
def get_repo_settings(self, cache=False): | ||||
""" | ||||
Returns list of issue tracker settings per repository | ||||
""" | ||||
if not self.repo_settings: | ||||
raise Exception('Repository is not specified') | ||||
all_settings = self.repo_settings.get_all_settings(cache=cache) | ||||
settings = self._make_dict_for_settings(all_settings) | ||||
return settings | ||||
def get_settings(self, cache=False): | ||||
if self.inherit_global_settings: | ||||
return self.get_global_settings(cache=cache) | ||||
else: | ||||
return self.get_repo_settings(cache=cache) | ||||
def delete_entries(self, uid): | ||||
if self.repo_settings: | ||||
all_patterns = self.get_repo_settings() | ||||
settings_model = self.repo_settings | ||||
else: | ||||
all_patterns = self.get_global_settings() | ||||
settings_model = self.global_settings | ||||
r2247 | entries = all_patterns.get(uid, []) | |||
r1 | ||||
for del_key in entries: | ||||
r5144 | setting_name = self._get_keyname(del_key, uid, prefix='') | |||
r1 | entry = settings_model.get_setting_by_name(setting_name) | |||
if entry: | ||||
Session().delete(entry) | ||||
Session().commit() | ||||
def create_or_update_setting( | ||||
self, name, val=Optional(''), type_=Optional('unicode')): | ||||
if self.repo_settings: | ||||
setting = self.repo_settings.create_or_update_setting( | ||||
name, val, type_) | ||||
else: | ||||
setting = self.global_settings.create_or_update_setting( | ||||
name, val, type_) | ||||
return setting | ||||
class VcsSettingsModel(object): | ||||
INHERIT_SETTINGS = 'inherit_vcs_settings' | ||||
Martin Bornhold
|
r358 | GENERAL_SETTINGS = ( | ||
r754 | 'use_outdated_comments', | |||
'pr_merge_enabled', | ||||
r5657 | 'auto_merge_enabled', | |||
Mathieu Cantin
|
r2055 | 'hg_use_rebase_for_merging', | ||
r2056 | 'hg_close_branch_before_merging', | |||
'git_use_rebase_for_merging', | ||||
Bartłomiej Wołyńczyk
|
r2685 | 'git_close_branch_before_merging', | ||
'diff_cache', | ||||
) | ||||
r754 | ||||
r1 | HOOKS_SETTINGS = ( | |||
('hooks', 'changegroup.repo_size'), | ||||
('hooks', 'changegroup.push_logger'), | ||||
r3625 | ('hooks', 'outgoing.pull_logger'), | |||
) | ||||
r1 | HG_SETTINGS = ( | |||
r754 | ('extensions', 'largefiles'), | |||
r1738 | ('phases', 'publish'), | |||
r3625 | ('extensions', 'evolve'), | |||
('extensions', 'topic'), | ||||
('experimental', 'evolution'), | ||||
r3626 | ('experimental', 'evolution.exchange'), | |||
r3625 | ) | |||
r1570 | GIT_SETTINGS = ( | |||
r3625 | ('vcs_git_lfs', 'enabled'), | |||
) | ||||
r754 | GLOBAL_HG_SETTINGS = ( | |||
('extensions', 'largefiles'), | ||||
('phases', 'publish'), | ||||
r3625 | ('extensions', 'evolve'), | |||
('extensions', 'topic'), | ||||
('experimental', 'evolution'), | ||||
r3626 | ('experimental', 'evolution.exchange'), | |||
r3625 | ) | |||
r1570 | GLOBAL_GIT_SETTINGS = ( | |||
('vcs_git_lfs', 'enabled'), | ||||
r3625 | ) | |||
Bartłomiej Wołyńczyk
|
r2685 | |||
r1 | SVN_BRANCH_SECTION = 'vcs_svn_branch' | |||
SVN_TAG_SECTION = 'vcs_svn_tag' | ||||
PATH_SETTING = ('paths', '/') | ||||
def __init__(self, sa=None, repo=None): | ||||
self.global_settings = SettingsModel(sa=sa) | ||||
self.repo_settings = SettingsModel(sa=sa, repo=repo) if repo else None | ||||
r1570 | self._ui_settings = ( | |||
self.HG_SETTINGS + self.GIT_SETTINGS + self.HOOKS_SETTINGS) | ||||
r1 | self._svn_sections = (self.SVN_BRANCH_SECTION, self.SVN_TAG_SECTION) | |||
@property | ||||
@assert_repo_settings | ||||
def inherit_global_settings(self): | ||||
setting = self.repo_settings.get_setting_by_name(self.INHERIT_SETTINGS) | ||||
return setting.app_settings_value if setting else True | ||||
@inherit_global_settings.setter | ||||
@assert_repo_settings | ||||
def inherit_global_settings(self, value): | ||||
self.repo_settings.create_or_update_setting( | ||||
self.INHERIT_SETTINGS, value, type_='bool') | ||||
r5144 | def get_keyname(self, key_name, prefix='rhodecode_'): | |||
return f'{prefix}{key_name}' | ||||
r1 | def get_global_svn_branch_patterns(self): | |||
return self.global_settings.get_ui_by_section(self.SVN_BRANCH_SECTION) | ||||
@assert_repo_settings | ||||
def get_repo_svn_branch_patterns(self): | ||||
return self.repo_settings.get_ui_by_section(self.SVN_BRANCH_SECTION) | ||||
def get_global_svn_tag_patterns(self): | ||||
return self.global_settings.get_ui_by_section(self.SVN_TAG_SECTION) | ||||
@assert_repo_settings | ||||
def get_repo_svn_tag_patterns(self): | ||||
return self.repo_settings.get_ui_by_section(self.SVN_TAG_SECTION) | ||||
def get_global_settings(self): | ||||
return self._collect_all_settings(global_=True) | ||||
@assert_repo_settings | ||||
def get_repo_settings(self): | ||||
return self._collect_all_settings(global_=False) | ||||
@assert_repo_settings | ||||
r3894 | def get_repo_settings_inherited(self): | |||
global_settings = self.get_global_settings() | ||||
global_settings.update(self.get_repo_settings()) | ||||
return global_settings | ||||
@assert_repo_settings | ||||
r1 | def create_or_update_repo_settings( | |||
self, data, inherit_global_settings=False): | ||||
from rhodecode.model.scm import ScmModel | ||||
self.inherit_global_settings = inherit_global_settings | ||||
repo = self.repo_settings.get_repo() | ||||
if not inherit_global_settings: | ||||
if repo.repo_type == 'svn': | ||||
self.create_repo_svn_settings(data) | ||||
else: | ||||
self.create_or_update_repo_hook_settings(data) | ||||
self.create_or_update_repo_pr_settings(data) | ||||
if repo.repo_type == 'hg': | ||||
self.create_or_update_repo_hg_settings(data) | ||||
r1570 | if repo.repo_type == 'git': | |||
self.create_or_update_repo_git_settings(data) | ||||
r1 | ScmModel().mark_for_invalidation(repo.repo_name, delete=True) | |||
@assert_repo_settings | ||||
def create_or_update_repo_hook_settings(self, data): | ||||
for section, key in self.HOOKS_SETTINGS: | ||||
data_key = self._get_form_ui_key(section, key) | ||||
if data_key not in data: | ||||
raise ValueError( | ||||
r5095 | f'The given data does not contain {data_key} key') | |||
r1 | ||||
active = data.get(data_key) | ||||
repo_setting = self.repo_settings.get_ui_by_section_and_key( | ||||
section, key) | ||||
if not repo_setting: | ||||
global_setting = self.global_settings.\ | ||||
get_ui_by_section_and_key(section, key) | ||||
self.repo_settings.create_ui_section_value( | ||||
section, global_setting.ui_value, key=key, active=active) | ||||
else: | ||||
repo_setting.ui_active = active | ||||
Session().add(repo_setting) | ||||
def update_global_hook_settings(self, data): | ||||
for section, key in self.HOOKS_SETTINGS: | ||||
data_key = self._get_form_ui_key(section, key) | ||||
if data_key not in data: | ||||
raise ValueError( | ||||
r5095 | f'The given data does not contain {data_key} key') | |||
r1 | active = data.get(data_key) | |||
repo_setting = self.global_settings.get_ui_by_section_and_key( | ||||
section, key) | ||||
repo_setting.ui_active = active | ||||
Session().add(repo_setting) | ||||
@assert_repo_settings | ||||
def create_or_update_repo_pr_settings(self, data): | ||||
return self._create_or_update_general_settings( | ||||
self.repo_settings, data) | ||||
def create_or_update_global_pr_settings(self, data): | ||||
return self._create_or_update_general_settings( | ||||
self.global_settings, data) | ||||
@assert_repo_settings | ||||
def create_repo_svn_settings(self, data): | ||||
return self._create_svn_settings(self.repo_settings, data) | ||||
r3625 | def _set_evolution(self, settings, is_enabled): | |||
if is_enabled: | ||||
# if evolve is active set evolution=all | ||||
self._create_or_update_ui( | ||||
settings, *('experimental', 'evolution'), value='all', | ||||
active=True) | ||||
self._create_or_update_ui( | ||||
settings, *('experimental', 'evolution.exchange'), value='yes', | ||||
active=True) | ||||
# if evolve is active set topics server support | ||||
self._create_or_update_ui( | ||||
settings, *('extensions', 'topic'), value='', | ||||
active=True) | ||||
else: | ||||
self._create_or_update_ui( | ||||
settings, *('experimental', 'evolution'), value='', | ||||
active=False) | ||||
self._create_or_update_ui( | ||||
settings, *('experimental', 'evolution.exchange'), value='no', | ||||
active=False) | ||||
self._create_or_update_ui( | ||||
settings, *('extensions', 'topic'), value='', | ||||
active=False) | ||||
r1 | @assert_repo_settings | |||
def create_or_update_repo_hg_settings(self, data): | ||||
r1738 | largefiles, phases, evolve = \ | |||
r3625 | self.HG_SETTINGS[:3] | |||
r1738 | largefiles_key, phases_key, evolve_key = \ | |||
r3625 | self._get_settings_keys(self.HG_SETTINGS[:3], data) | |||
r1570 | ||||
r1 | self._create_or_update_ui( | |||
self.repo_settings, *largefiles, value='', | ||||
active=data[largefiles_key]) | ||||
self._create_or_update_ui( | ||||
r1738 | self.repo_settings, *evolve, value='', | |||
active=data[evolve_key]) | ||||
r3625 | self._set_evolution(self.repo_settings, is_enabled=data[evolve_key]) | |||
r1738 | self._create_or_update_ui( | |||
r1 | self.repo_settings, *phases, value=safe_str(data[phases_key])) | |||
def create_or_update_global_hg_settings(self, data): | ||||
r5535 | opts_len = 3 | |||
largefiles, phases, evolve \ | ||||
r5250 | = self.GLOBAL_HG_SETTINGS[:opts_len] | |||
r5535 | largefiles_key, phases_key, evolve_key \ | |||
r5250 | = self._get_settings_keys(self.GLOBAL_HG_SETTINGS[:opts_len], data) | |||
r1738 | ||||
r1 | self._create_or_update_ui( | |||
self.global_settings, *largefiles, value='', | ||||
active=data[largefiles_key]) | ||||
self._create_or_update_ui( | ||||
self.global_settings, *phases, value=safe_str(data[phases_key])) | ||||
self._create_or_update_ui( | ||||
r1738 | self.global_settings, *evolve, value='', | |||
active=data[evolve_key]) | ||||
r3625 | self._set_evolution(self.global_settings, is_enabled=data[evolve_key]) | |||
r1 | ||||
r1570 | def create_or_update_repo_git_settings(self, data): | |||
r3625 | # NOTE(marcink): # comma makes unpack work properly | |||
r1570 | lfs_enabled, \ | |||
= self.GIT_SETTINGS | ||||
lfs_enabled_key, \ | ||||
= self._get_settings_keys(self.GIT_SETTINGS, data) | ||||
self._create_or_update_ui( | ||||
self.repo_settings, *lfs_enabled, value=data[lfs_enabled_key], | ||||
active=data[lfs_enabled_key]) | ||||
def create_or_update_global_git_settings(self, data): | ||||
r5535 | lfs_enabled = self.GLOBAL_GIT_SETTINGS[0] | |||
lfs_enabled_key = self._get_settings_keys(self.GLOBAL_GIT_SETTINGS, data)[0] | ||||
r1570 | ||||
self._create_or_update_ui( | ||||
self.global_settings, *lfs_enabled, value=data[lfs_enabled_key], | ||||
active=data[lfs_enabled_key]) | ||||
r748 | def create_or_update_global_svn_settings(self, data): | |||
r754 | # branch/tags patterns | |||
self._create_svn_settings(self.global_settings, data) | ||||
r1 | @assert_repo_settings | |||
def delete_repo_svn_pattern(self, id_): | ||||
r2180 | ui = self.repo_settings.UiDbModel.get(id_) | |||
if ui and ui.repository.repo_name == self.repo_settings.repo: | ||||
# only delete if it's the same repo as initialized settings | ||||
self.repo_settings.delete_ui(id_) | ||||
else: | ||||
# raise error as if we wouldn't find this option | ||||
self.repo_settings.delete_ui(-1) | ||||
r1 | ||||
def delete_global_svn_pattern(self, id_): | ||||
self.global_settings.delete_ui(id_) | ||||
@assert_repo_settings | ||||
def get_repo_ui_settings(self, section=None, key=None): | ||||
global_uis = self.global_settings.get_ui(section, key) | ||||
repo_uis = self.repo_settings.get_ui(section, key) | ||||
r3625 | ||||
r1 | filtered_repo_uis = self._filter_ui_settings(repo_uis) | |||
filtered_repo_uis_keys = [ | ||||
(s.section, s.key) for s in filtered_repo_uis] | ||||
def _is_global_ui_filtered(ui): | ||||
return ( | ||||
(ui.section, ui.key) in filtered_repo_uis_keys | ||||
or ui.section in self._svn_sections) | ||||
filtered_global_uis = [ | ||||
ui for ui in global_uis if not _is_global_ui_filtered(ui)] | ||||
return filtered_global_uis + filtered_repo_uis | ||||
def get_global_ui_settings(self, section=None, key=None): | ||||
return self.global_settings.get_ui(section, key) | ||||
r754 | def get_ui_settings_as_config_obj(self, section=None, key=None): | |||
config = base.Config() | ||||
ui_settings = self.get_ui_settings(section=section, key=key) | ||||
for entry in ui_settings: | ||||
config.set(entry.section, entry.key, entry.value) | ||||
return config | ||||
r1 | def get_ui_settings(self, section=None, key=None): | |||
if not self.repo_settings or self.inherit_global_settings: | ||||
return self.get_global_ui_settings(section, key) | ||||
else: | ||||
return self.get_repo_ui_settings(section, key) | ||||
def get_svn_patterns(self, section=None): | ||||
if not self.repo_settings: | ||||
return self.get_global_ui_settings(section) | ||||
else: | ||||
return self.get_repo_ui_settings(section) | ||||
@assert_repo_settings | ||||
def get_repo_general_settings(self): | ||||
global_settings = self.global_settings.get_all_settings() | ||||
repo_settings = self.repo_settings.get_all_settings() | ||||
filtered_repo_settings = self._filter_general_settings(repo_settings) | ||||
global_settings.update(filtered_repo_settings) | ||||
return global_settings | ||||
def get_global_general_settings(self): | ||||
return self.global_settings.get_all_settings() | ||||
def get_general_settings(self): | ||||
if not self.repo_settings or self.inherit_global_settings: | ||||
return self.get_global_general_settings() | ||||
else: | ||||
return self.get_repo_general_settings() | ||||
def _filter_ui_settings(self, settings): | ||||
filtered_settings = [ | ||||
s for s in settings if self._should_keep_setting(s)] | ||||
return filtered_settings | ||||
def _should_keep_setting(self, setting): | ||||
keep = ( | ||||
(setting.section, setting.key) in self._ui_settings or | ||||
setting.section in self._svn_sections) | ||||
return keep | ||||
def _filter_general_settings(self, settings): | ||||
r5144 | keys = [self.get_keyname(key) for key in self.GENERAL_SETTINGS] | |||
r1 | return { | |||
k: settings[k] | ||||
for k in settings if k in keys} | ||||
def _collect_all_settings(self, global_=False): | ||||
settings = self.global_settings if global_ else self.repo_settings | ||||
result = {} | ||||
for section, key in self._ui_settings: | ||||
ui = settings.get_ui_by_section_and_key(section, key) | ||||
result_key = self._get_form_ui_key(section, key) | ||||
r1570 | ||||
r1 | if ui: | |||
if section in ('hooks', 'extensions'): | ||||
result[result_key] = ui.ui_active | ||||
r1570 | elif result_key in ['vcs_git_lfs_enabled']: | |||
result[result_key] = ui.ui_active | ||||
r1 | else: | |||
result[result_key] = ui.ui_value | ||||
for name in self.GENERAL_SETTINGS: | ||||
setting = settings.get_setting_by_name(name) | ||||
if setting: | ||||
r5144 | result_key = self.get_keyname(name) | |||
r1 | result[result_key] = setting.app_settings_value | |||
return result | ||||
def _get_form_ui_key(self, section, key): | ||||
return '{section}_{key}'.format( | ||||
section=section, key=key.replace('.', '_')) | ||||
def _create_or_update_ui( | ||||
self, settings, section, key, value=None, active=None): | ||||
ui = settings.get_ui_by_section_and_key(section, key) | ||||
if not ui: | ||||
active = True if active is None else active | ||||
settings.create_ui_section_value( | ||||
section, value, key=key, active=active) | ||||
else: | ||||
if active is not None: | ||||
ui.ui_active = active | ||||
if value is not None: | ||||
ui.ui_value = value | ||||
Session().add(ui) | ||||
def _create_svn_settings(self, settings, data): | ||||
svn_settings = { | ||||
'new_svn_branch': self.SVN_BRANCH_SECTION, | ||||
'new_svn_tag': self.SVN_TAG_SECTION | ||||
} | ||||
for key in svn_settings: | ||||
if data.get(key): | ||||
settings.create_ui_section_value(svn_settings[key], data[key]) | ||||
def _create_or_update_general_settings(self, settings, data): | ||||
for name in self.GENERAL_SETTINGS: | ||||
r5144 | data_key = self.get_keyname(name) | |||
r1 | if data_key not in data: | |||
raise ValueError( | ||||
r5095 | f'The given data does not contain {data_key} key') | |||
r1 | setting = settings.create_or_update_setting( | |||
name, data[data_key], 'bool') | ||||
Session().add(setting) | ||||
r754 | def _get_settings_keys(self, settings, data): | |||
r1 | data_keys = [self._get_form_ui_key(*s) for s in settings] | |||
for data_key in data_keys: | ||||
if data_key not in data: | ||||
raise ValueError( | ||||
r5095 | f'The given data does not contain {data_key} key') | |||
r1 | return data_keys | |||
r1682 | ||||
def create_largeobjects_dirs_if_needed(self, repo_store_path): | ||||
""" | ||||
This is subscribed to the `pyramid.events.ApplicationCreated` event. It | ||||
does a repository scan if enabled in the settings. | ||||
""" | ||||
from rhodecode.lib.vcs.backends.hg import largefiles_store | ||||
from rhodecode.lib.vcs.backends.git import lfs_store | ||||
paths = [ | ||||
largefiles_store(repo_store_path), | ||||
lfs_store(repo_store_path)] | ||||
for path in paths: | ||||
if os.path.isdir(path): | ||||
continue | ||||
if os.path.isfile(path): | ||||
continue | ||||
# not a file nor dir, we try to create it | ||||
try: | ||||
os.makedirs(path) | ||||
except Exception: | ||||
log.warning('Failed to create largefiles dir:%s', path) | ||||