From f721c5ef0cdd9952d014a1dc9e3a944afc0f7f45 Mon Sep 17 00:00:00 2001 From: Ruben Pollan Date: Thu, 31 Aug 2017 10:57:41 +0200 Subject: [feat] add the posibility of extracting a section of the config So each service can have it's own config section passed to the constructor. --- src/leap/bitmask/config.py | 15 +++++++++++++++ src/leap/bitmask/core/configurable.py | 3 +++ 2 files changed, 18 insertions(+) (limited to 'src') diff --git a/src/leap/bitmask/config.py b/src/leap/bitmask/config.py index 833980d9..4c1fbca7 100644 --- a/src/leap/bitmask/config.py +++ b/src/leap/bitmask/config.py @@ -67,6 +67,9 @@ class Configuration(object): self.read() assert self.config.get(section, option) == value + def get_section(self, section): + return _ConfigurationSection(self, section) + def read(self): self.config = ConfigParser.SafeConfigParser() if not os.path.isfile(self.config_path): @@ -82,3 +85,15 @@ class Configuration(object): def _create_default_config(self): with open(self.config_path, 'w') as outf: outf.write(self.default_config) + + +class _ConfigurationSection(object): + def __init__(self, config, section): + self.config = config + self.section = section + + def get(self, option, default=None, boolean=False): + return self.config.get(self.section, option, default, boolean) + + def set(self, option, value): + return self.config.set(self.section, option, value) diff --git a/src/leap/bitmask/core/configurable.py b/src/leap/bitmask/core/configurable.py index 262f0300..5a9e5c94 100644 --- a/src/leap/bitmask/core/configurable.py +++ b/src/leap/bitmask/core/configurable.py @@ -42,6 +42,9 @@ class ConfigurableService(service.MultiService): def set_config(self, section, option, value): return self.cfg.set(section, option, value) + def get_config_section(self, section): + return self.cfg.get_section(section) + DEFAULT_CONFIG = """ [services] -- cgit v1.2.3