diff options
Diffstat (limited to 'scripts/rpki/config.py')
-rw-r--r-- | scripts/rpki/config.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/scripts/rpki/config.py b/scripts/rpki/config.py index 962f28fd..7a60f95b 100644 --- a/scripts/rpki/config.py +++ b/scripts/rpki/config.py @@ -9,20 +9,23 @@ import ConfigParser class parser(ConfigParser.RawConfigParser): - def __init__(self, file=None): + def __init__(self, file = None, section = None): """Initialize this parser.""" ConfigParser.RawConfigParser.__init__(self) if file: self.read(file) + self.default_section = section - def multiget(self, section, option): + def multiget(self, option, section = None): """Parse OpenSSL-style foo.0, foo.1, ... subscripted options. Returns a list of values matching the specified option name. """ matches = [] + if section is None: + section = self.default_section if self.has_option(section, option): - matches.append((-1, self.get(section, option))) + matches.append((-1, self.get(option, section = section))) for key, value in self.items(section): s = key.rsplit(".", 1) if len(s) == 2 and s[0] == option and s[1].isdigit(): @@ -30,8 +33,10 @@ class parser(ConfigParser.RawConfigParser): matches.sort() return [match[1] for match in matches] - def get(self, section, option, default=None): + def get(self, option, default = None, section = None): """Get an option, perhaps with a default value.""" + if section is None: + section = self.default_section if default is None or self.has_option(section, option): return ConfigParser.RawConfigParser.get(self, section, option) else: |