Spaces:
Running
on
L40S
Running
on
L40S
File size: 3,561 Bytes
4450790 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
import os
import json
import re
from .utils import get_dict_value, set_dict_value, dict_has_key, load_json_file
def get_config_value(key):
return get_dict_value(RGTHREE_CONFIG, key)
def extend_config(default_config, user_config):
""" Returns a new config dict combining user_config into defined keys for default_config."""
cfg = {}
for key, value in default_config.items():
if key not in user_config:
cfg[key] = value
elif isinstance(value, dict):
cfg[key] = extend_config(value, user_config[key])
else:
cfg[key] = user_config[key] if key in user_config else value
return cfg
def set_user_config(data: dict):
""" Sets the user configuration."""
count = 0
for key, value in data.items():
if dict_has_key(DEFAULT_CONFIG, key):
set_dict_value(USER_CONFIG, key, value)
set_dict_value(RGTHREE_CONFIG, key, value)
count += 1
if count > 0:
write_user_config()
def get_rgthree_default_config():
""" Gets the default configuration."""
return load_json_file(DEFAULT_CONFIG_FILE, default={})
def get_rgthree_user_config():
""" Gets the user configuration."""
return load_json_file(USER_CONFIG_FILE, default={})
def write_user_config():
""" Writes the user configuration."""
with open(USER_CONFIG_FILE, 'w+', encoding='UTF-8') as file:
json.dump(USER_CONFIG, file, sort_keys=True, indent=2, separators=(",", ": "))
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
DEFAULT_CONFIG_FILE = os.path.join(THIS_DIR, '..', 'rgthree_config.json.default')
USER_CONFIG_FILE = os.path.join(THIS_DIR, '..', 'rgthree_config.json')
DEFAULT_CONFIG = {}
USER_CONFIG = {}
RGTHREE_CONFIG = {}
def refresh_config():
"""Refreshes the config."""
global DEFAULT_CONFIG, USER_CONFIG, RGTHREE_CONFIG
DEFAULT_CONFIG = get_rgthree_default_config()
USER_CONFIG = get_rgthree_user_config()
# Migrate old config options into "features"
needs_to_write_user_config = False
if 'patch_recursive_execution' in USER_CONFIG:
del USER_CONFIG['patch_recursive_execution']
needs_to_write_user_config = True
if 'features' in USER_CONFIG and 'patch_recursive_execution' in USER_CONFIG['features']:
del USER_CONFIG['features']['patch_recursive_execution']
needs_to_write_user_config = True
if 'show_alerts_for_corrupt_workflows' in USER_CONFIG:
if 'features' not in USER_CONFIG:
USER_CONFIG['features'] = {}
USER_CONFIG['features']['show_alerts_for_corrupt_workflows'] = USER_CONFIG[
'show_alerts_for_corrupt_workflows']
del USER_CONFIG['show_alerts_for_corrupt_workflows']
needs_to_write_user_config = True
if 'monitor_for_corrupt_links' in USER_CONFIG:
if 'features' not in USER_CONFIG:
USER_CONFIG['features'] = {}
USER_CONFIG['features']['monitor_for_corrupt_links'] = USER_CONFIG['monitor_for_corrupt_links']
del USER_CONFIG['monitor_for_corrupt_links']
needs_to_write_user_config = True
if needs_to_write_user_config is True:
print('writing new user config.')
write_user_config()
RGTHREE_CONFIG = extend_config(DEFAULT_CONFIG, USER_CONFIG)
if "unreleased" in USER_CONFIG and "unreleased" not in RGTHREE_CONFIG:
RGTHREE_CONFIG["unreleased"] = USER_CONFIG["unreleased"]
if "debug" in USER_CONFIG and "debug" not in RGTHREE_CONFIG:
RGTHREE_CONFIG["debug"] = USER_CONFIG["debug"]
def get_config():
"""Returns the congfig."""
return RGTHREE_CONFIG
refresh_config()
|