aselimov/pubs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added get operator to config

Browse Source
This commit is contained in:
Fabien Benureau 2013-07-01 14:56:16 +01:00
parent 45da61f4d2
commit 320a488667
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
papers/configs.py
View File

@ -72,6 +72,12 @@ class Config(object):
value = str2bool(value) value = str2bool(value)
return value return value
def get(self, name, default = None):
try:
return self.__getattr__(name)
except (configparser.NoOptionError, configparser.NoSectionError):
return default
def items(self): def items(self):
for name, value in self._cfg.items(self._section): for name, value in self._cfg.items(self._section):
if name in BOOLEANS: if name in BOOLEANS:

3
tests/test_config.py
View File

@ -52,6 +52,9 @@ class TestConfig(unittest.TestCase):
with self.assertRaises(configparser.NoOptionError): with self.assertRaises(configparser.NoOptionError):
config().color2 config().color2
self.assertEqual(config().get('color2', default = 'blue'), 'blue')
with self.assertRaises(configparser.NoSectionError): 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)