def main(): parser = argparse.ArgumentParser() parser.set_defaults(func=lambda _: parser.print_usage()) subparsers = parser.add_subparsers() upgrade_p = subparsers.add_parser( 'upgrade', help='Upgrade the user database', formatter_class=argparse.ArgumentDefaultsHelpFormatter) upgrade_p.set_defaults(func=upgrade) downgrade_p = subparsers.add_parser( 'downgrade', help='Downgrade the user database', formatter_class=argparse.ArgumentDefaultsHelpFormatter) downgrade_p.set_defaults(func=downgrade) args = parser.parse_args() config = load_config('main.cfg') db_path = config['data_storage']['user_database'][len('sqlite:///'):] conn = sqlite3.connect(db_path) cur = conn.cursor() args.func(cur) conn.commit() conn.close()
def test_read_list_from_config(monkeypatch, input_data, expected): monkeypatch.setattr('helperFunctions.config.get_config_dir', lambda: '{}/helperFunctions'.format(get_test_data_dir())) test_config = load_config('test.cfg') test_config.add_section('test_section') test_config.set('test_section', 'test_option', input_data) result = read_list_from_config(test_config, 'test_section', 'test_option') assert result == expected
def main(distribution): xenial = distribution == 'xenial' logging.info('Updating package lists') apt_update_sources() # Non python dependencies apt_install_packages('build-essential', 'automake', 'autoconf', 'libtool') # python dependencies apt_install_packages('python3', 'python3-dev', 'python', 'python-dev', 'python-wheel', 'python-setuptools') pip3_install_packages('pytest', 'pytest-cov', 'pytest-pep8') if not xenial: pip3_install_packages('testresources') # make bin dir with suppress(FileExistsError): os.mkdir('../bin') config = load_config('main.cfg') data_folder = config.get('unpack', 'data_folder') os.makedirs(str(Path(data_folder, 'files')), exist_ok=True) os.makedirs(str(Path(data_folder, 'reports')), exist_ok=True) return 0
def test_read_list_from_config__key_not_in_config(monkeypatch): monkeypatch.setattr('helperFunctions.config.get_config_dir', lambda: '{}/helperFunctions'.format(get_test_data_dir())) test_config = load_config('test.cfg') result = read_list_from_config(test_config, 'foo', 'bar') assert result == [] result = read_list_from_config(test_config, 'test', 'bar') assert result == []
def _set_config(cls): cls.config = load_config('main.cfg') cls.config.set('data_storage', 'main_database', TMP_DB_NAME) cls.config.set('data_storage', 'intercom_database_prefix', TMP_DB_NAME) cls.config.set('data_storage', 'statistic_database', TMP_DB_NAME) cls.config.set('data_storage', 'firmware_file_storage_directory', TMP_DIR.name) cls.config.set('ExpertSettings', 'authentication', 'false') cls.config.set('Logging', 'mongoDbLogFile', os.path.join(TMP_DIR.name, 'mongo.log'))
def __init__(self, config=None): self.matches = [] if config: self.config = config else: self.config = load_config(CONFIG_FILE) self.db_path = self.config['data_storage'][ 'firmware_file_storage_directory']
def load_users_from_main_config(config: ConfigParser): fact_config = load_config('main.cfg') config.set('data_storage', 'db_admin_user', fact_config['data_storage']['db_admin_user']) config.set('data_storage', 'db_admin_pw', fact_config['data_storage']['db_admin_pw']) config.set('data_storage', 'db_readonly_user', fact_config['data_storage']['db_readonly_user']) config.set('data_storage', 'db_readonly_pw', fact_config['data_storage']['db_readonly_pw'])
def main(): args = setup_argparse() file_name = os.path.basename(args.config_file) config = load_config(file_name) frontend = WebFrontEnd(config) start_user_management(frontend.app) return 0
def _set_config(self): self.config = load_config('main.cfg') self.config.set('data_storage', 'main_database', TMP_DB_NAME) self.config.set('data_storage', 'intercom_database_prefix', TMP_DB_NAME) self.config.set('data_storage', 'statistic_database', TMP_DB_NAME) self.config.set('data_storage', 'firmware_file_storage_directory', TMP_DIR.name) self.config.set('Logging', 'mongoDbLogFile', os.path.join(TMP_DIR.name, 'mongo.log'))
def main(): args = setup_argparse() file_name = Path(args.config_file).name config = load_config(file_name) frontend = WebFrontEnd(config) start_user_management(frontend.app, frontend.user_datastore, frontend.user_db, PromptSession()) return 0
def main(distribution): logging.info('Updating package lists') apt_update_sources() # install dependencies install_dependencies(DEPENDENCIES['common']) install_dependencies(DEPENDENCIES[distribution]) # make bin dir with suppress(FileExistsError): os.mkdir('../bin') config = load_config('main.cfg') data_folder = config.get('unpack', 'data_folder') os.makedirs(str(Path(data_folder, 'files')), exist_ok=True) os.makedirs(str(Path(data_folder, 'reports')), exist_ok=True) return 0
def init_basic_config(self): config = ConfigParser() config.add_section(self.PLUGIN_NAME) config.set(self.PLUGIN_NAME, 'threads', '1') config.add_section('ExpertSettings') config.set('ExpertSettings', 'block_delay', '2') config.add_section('data_storage') faf_config = load_config('main.cfg') config.set('data_storage', 'db_admin_user', faf_config['data_storage']['db_admin_user']) config.set('data_storage', 'db_admin_pw', faf_config['data_storage']['db_admin_pw']) config.set('data_storage', 'db_readonly_user', faf_config['data_storage']['db_readonly_user']) config.set('data_storage', 'db_readonly_pw', faf_config['data_storage']['db_readonly_pw']) config.set('data_storage', 'mongo_server', 'localhost') config.set('data_storage', 'mongo_port', '54321') config.set('data_storage', 'view_storage', 'tmp_view') return config
def _set_config(cls): cls.config = load_config('main.cfg') cls.config.set('data_storage', 'main_database', TMP_DB_NAME) cls.config.set('data_storage', 'intercom_database_prefix', TMP_DB_NAME) cls.config.set('data_storage', 'statistic_database', TMP_DB_NAME) cls.config.set('ExpertSettings', 'authentication', 'false')
def test_load_config(monkeypatch): monkeypatch.setattr('helperFunctions.config.get_config_dir', lambda: '{}/helperFunctions'.format(get_test_data_dir())) test_config = load_config('test.cfg') assert test_config['test']['test'] == 'test_config', 'config not correct'