added get operator to config

main
Fabien Benureau 12 years ago
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…
Cancel
Save