configfile: Add helpers for parsing list of parameters from the config

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
This commit is contained in:
Kevin O'Connor 2021-08-19 13:03:35 -04:00
parent c64ea474d7
commit f9ca5b9098
1 changed files with 28 additions and 0 deletions

View File

@ -74,6 +74,34 @@ class ConfigWrapper:
raise error("Choice '%s' for option '%s' in section '%s'" raise error("Choice '%s' for option '%s' in section '%s'"
" is not a valid choice" % (c, option, self.section)) " is not a valid choice" % (c, option, self.section))
return choices[c] return choices[c]
def getlists(self, option, default=sentinel, seps=(',',), count=None,
parser=str, note_valid=True):
def lparser(value, pos):
if pos:
# Nested list
parts = [p.strip() for p in value.split(seps[pos])]
return tuple([lparser(p, pos - 1) for p in parts if p])
res = [parser(p.strip()) for p in value.split(seps[pos])]
if count is not None and len(res) != count:
raise error("Option '%s' in section '%s' must have %d elements"
% (option, self.section, count))
return tuple(res)
def fcparser(section, option):
return lparser(self.fileconfig.get(section, option), len(seps) - 1)
return self._get_wrapper(fcparser, option, default,
note_valid=note_valid)
def getlist(self, option, default=sentinel, sep=',', count=None,
note_valid=True):
return self.getlists(option, default, seps=(sep,), count=count,
parser=str, note_valid=note_valid)
def getintlist(self, option, default=sentinel, sep=',', count=None,
note_valid=True):
return self.getlists(option, default, seps=(sep,), count=count,
parser=int, note_valid=note_valid)
def getfloatlist(self, option, default=sentinel, sep=',', count=None,
note_valid=True):
return self.getlists(option, default, seps=(sep,), count=count,
parser=float, note_valid=note_valid)
def getsection(self, section): def getsection(self, section):
return ConfigWrapper(self.printer, self.fileconfig, return ConfigWrapper(self.printer, self.fileconfig,
self.access_tracking, section) self.access_tracking, section)