diff options
| author | Alexandre Flament <alex@al-f.net> | 2020-11-27 19:32:45 +0100 |
|---|---|---|
| committer | Alexandre Flament <alex@al-f.net> | 2020-11-27 19:40:04 +0100 |
| commit | b4b81a5e1a74f03926e3c2e6f8c5fade99f7eabb (patch) | |
| tree | f72272f8c132759a2e792765632352a38007d1ba /utils/update_user_settings.py | |
| parent | 1cfe7f2a7543b2994a1afd0d81da1962d04423b0 (diff) | |
[enh] settings.yml: add use_default_settings option (2nd version)
Diffstat (limited to 'utils/update_user_settings.py')
| -rw-r--r-- | utils/update_user_settings.py | 98 |
1 files changed, 0 insertions, 98 deletions
diff --git a/utils/update_user_settings.py b/utils/update_user_settings.py deleted file mode 100644 index fb6fd0b3f..000000000 --- a/utils/update_user_settings.py +++ /dev/null @@ -1,98 +0,0 @@ -#!/usr/bin/env python - -# set path -from sys import path -from os.path import realpath, dirname, join -path.append(realpath(dirname(realpath(__file__)) + '/../')) - -import argparse -import sys -import string -import ruamel.yaml -import secrets -import collections -from ruamel.yaml.scalarstring import SingleQuotedScalarString, DoubleQuotedScalarString -from searx.settings import load_settings, check_settings_yml, get_default_settings_path -from searx.exceptions import SearxSettingsException - - -RANDOM_STRING_LETTERS = string.ascii_lowercase + string.digits + string.ascii_uppercase - - -def get_random_string(): - r = [secrets.choice(RANDOM_STRING_LETTERS) for _ in range(64)] - return ''.join(r) - - -def main(prog_arg): - yaml = ruamel.yaml.YAML() - yaml.preserve_quotes = True - yaml.indent(mapping=4, sequence=1, offset=2) - user_settings_path = prog_args.get('user-settings-yaml') - - try: - default_settings, _ = load_settings(False) - if check_settings_yml(user_settings_path): - with open(user_settings_path, 'r', encoding='utf-8') as f: - user_settings = yaml.load(f.read()) - new_user_settings = False - else: - user_settings = yaml.load('use_default_settings: True') - new_user_settings = True - except SearxSettingsException as e: - sys.stderr.write(str(e)) - return - - if not new_user_settings and not user_settings.get('use_default_settings'): - sys.stderr.write('settings.yml already exists and use_default_settings is not True') - return - - user_settings['use_default_settings'] = True - use_default_settings_comment = "settings based on " + get_default_settings_path() - user_settings.yaml_add_eol_comment(use_default_settings_comment, 'use_default_settings') - - if user_settings.get('server', {}).get('secret_key') in [None, 'ultrasecretkey']: - user_settings.setdefault('server', {})['secret_key'] = DoubleQuotedScalarString(get_random_string()) - - user_engines = user_settings.get('engines') - if user_engines: - has_user_engines = True - user_engines_dict = dict((definition['name'], definition) for definition in user_engines) - else: - has_user_engines = False - user_engines_dict = {} - user_engines = [] - - # remove old engines - if prog_arg.get('add-engines') or has_user_engines: - default_engines_dict = dict((definition['name'], definition) for definition in default_settings['engines']) - for i, engine in enumerate(user_engines): - if engine['name'] not in default_engines_dict: - del user_engines[i] - - # add new engines - if prog_arg.get('add-engines'): - for engine in default_settings.get('engines', {}): - if engine['name'] not in user_engines_dict: - user_engines.append({'name': engine['name']}) - user_settings['engines'] = user_engines - - # output - if prog_arg.get('dry-run'): - yaml.dump(user_settings, sys.stdout) - else: - with open(user_settings_path, 'w', encoding='utf-8') as f: - yaml.dump(user_settings, f) - - -def parse_args(): - parser = argparse.ArgumentParser(description='Update user settings.yml') - parser.add_argument('--add-engines', dest='add-engines', default=False, action='store_true', help='Add new engines') - parser.add_argument('--dry-run', dest='dry-run', default=False, action='store_true', help='Dry run') - parser.add_argument('user-settings-yaml', type=str) - return vars(parser.parse_args()) - - -if __name__ == '__main__': - prog_args = parse_args() - main(prog_args) |