added get operator to config
This commit is contained in:
parent
45da61f4d2
commit
320a488667
@ -72,6 +72,12 @@ class Config(object):
|
||||
value = str2bool(value)
|
||||
return value
|
||||
|
||||
def get(self, name, default = None):
|
||||
try:
|
||||
return self.__getattr__(name)
|
||||
except (configparser.NoOptionError, configparser.NoSectionError):
|
||||
return default
|
||||
|
||||
def items(self):
|
||||
for name, value in self._cfg.items(self._section):
|
||||
if name in BOOLEANS:
|
||||
|
@ -52,6 +52,9 @@ class TestConfig(unittest.TestCase):
|
||||
|
||||
with self.assertRaises(configparser.NoOptionError):
|
||||
config().color2
|
||||
self.assertEqual(config().get('color2', default = 'blue'), 'blue')
|
||||
|
||||
with self.assertRaises(configparser.NoSectionError):
|
||||
config(section = 'bla3').color
|
||||
config(section = 'bla3').color
|
||||
self.assertEqual(config(section = 'bla3').get('color', default = 'green'), 'green')
|
||||
self.assertEqual(config(section = 'bla3').get('color', default = config().color), True)
|
||||
|
Loading…
x
Reference in New Issue
Block a user