From be8feeb3ee124083895b2bb57c4b54711c2ca9b4 Mon Sep 17 00:00:00 2001 From: Itay Mizeretz Date: Sun, 12 Nov 2017 16:11:12 +0200 Subject: [PATCH] Add get config value function --- monkey_island/cc/services/config.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/monkey_island/cc/services/config.py b/monkey_island/cc/services/config.py index 200e24029..2145011c3 100644 --- a/monkey_island/cc/services/config.py +++ b/monkey_island/cc/services/config.py @@ -806,6 +806,14 @@ class ConfigService: config.pop(field, None) return config + @staticmethod + def get_config_value(config_key_as_arr): + config_key = reduce(lambda x, y: x+'.'+y, config_key_as_arr) + config = mongo.db.config.find_one({'name': 'newconfig'}, {config_key: 1}) + for config_key_part in config_key_as_arr: + config = config[config_key_part] + return config + @staticmethod def get_flat_config(): config_json = ConfigService.get_config()