# -*- coding: utf-8 -*- # Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved # # Licensed under CLOUD LINUX LICENSE AGREEMENT # http://cloudlinux.com/docs/LICENSE.TXT # import json import copy from collections.abc import Mapping UI_CONFIG_FILE = "/usr/share/l.v.e-manager/lvemanager-config.json" DEFAULT_UI_CONFIG = {'inodeLimits': {'showUserInodesUsage': False}, 'uiSettings': {'hideRubyApp': False, 'hideLVEUserStat': False, 'hidePythonApp': False, 'hideNodeJsApp': False, 'hidePHPextensions': False, 'hideDomainsTab': False, 'hidePhpApp': False, 'hideXrayApp': True, 'hideAccelerateWPApp': True, } } class UIConfig: def __init__(self, ui_config_path=UI_CONFIG_FILE): self.ui_config_path = ui_config_path self.file_content = {} self.config = {} try: with open(ui_config_path, encoding='utf-8') as json_file: self.file_content = json.load(json_file) self.config = self.file_content['ui_config'] except IOError: self.config = DEFAULT_UI_CONFIG except (ValueError, KeyError): self.file_content['ui_config'] = DEFAULT_UI_CONFIG self.config = DEFAULT_UI_CONFIG self._update_file(ui_config_path) def _update_file(self, file_path): with open(file_path, 'w', encoding='utf-8') as update_file: json.dump(self.file_content, update_file) def get_config(self): initial_config = copy.deepcopy(DEFAULT_UI_CONFIG) self._recursive_assign(initial_config, self.config) return initial_config def set_config(self, new_config): initial_config = copy.deepcopy(self.get_config()) self._recursive_assign(initial_config, new_config) with open(self.ui_config_path, 'w', encoding='utf-8') as json_file: json.dump({'ui_config': initial_config}, json_file) def get_param(self, param, parent_param=None): config = self.get_config() if parent_param is None: return config.get(param) elif config.get(parent_param) is not None: return config.get(parent_param).get(param) return None def _recursive_assign(self, initial_config, new_config): """ Function that recursively assignees values received from the UI :param initial_config: Config to which values are assigned :param new_config: New config file received from UI :return: """ for key, value in new_config.items(): if isinstance(value, Mapping): initial_config[key] = self._recursive_assign(initial_config.get(key, {}), value) elif value is not None and initial_config.get(key) is not None: initial_config[key] = value return initial_config