fix sub-namespace pollution problem

This commit is contained in:
Mark McClain
2011-02-16 17:05:48 -05:00
parent 9ae9e1f13d
commit 1a6eefe0b2
3 changed files with 21 additions and 14 deletions

View File

@@ -47,7 +47,7 @@ class ServeCommand(_ServeCommand, Command):
# for file-watching to work, we need a filename, not a module
if self.requires_config_file and self.args:
self.config = self.load_configuration(self.args[0])
self.args[0] = self.config._filename
self.args[0] = self.config.__file__
if self.options.reload is None:
self.options.reload = getattr(self.config.app, 'reload', False)

View File

@@ -11,7 +11,8 @@ class ConfigDict(dict):
class Config(object):
def __init__(self, conf_dict={}, filename=''):
self._filename = filename
self.__values__ = {}
self.__file__ = filename
self.update(conf_dict)
def update(self, conf_dict):
@@ -24,7 +25,7 @@ class Config(object):
if not IDENTIFIER.match(k):
raise ValueError('\'%s\' is not a valid indentifier' % k)
cur_val = self.__dict__.get(k)
cur_val = self.__values__.get(k)
if isinstance(cur_val, Config):
cur_val.update(conf_dict[k])
@@ -34,30 +35,36 @@ class Config(object):
def update_with_module(self, module):
self.update(conf_from_module(module))
def __getattr__(self, name):
try:
return self.__values__[name]
except KeyError:
raise AttributeError, "'pecan.conf' object has no attribute '%s'" % name
def __getitem__(self, key):
return self.__dict__[key]
return self.__values__[key]
def __setitem__(self, key, value):
if isinstance(value, dict) and not isinstance(value, ConfigDict):
if value.get('__force_dict__'):
del value['__force_dict__']
self.__dict__[key] = ConfigDict(value)
self.__values__[key] = ConfigDict(value)
else:
self.__dict__[key] = Config(value, filename=self._filename)
self.__values__[key] = Config(value, filename=self.__file__)
elif isinstance(value, basestring) and '%(confdir)s' in value:
confdir = os.path.dirname(self._filename) or os.getcwd()
self.__dict__[key] = value.replace('%(confdir)s', confdir)
confdir = os.path.dirname(self.__file__) or os.getcwd()
self.__values__[key] = value.replace('%(confdir)s', confdir)
else:
self.__dict__[key] = value
self.__values__[key] = value
def __iter__(self):
return self.__dict__.iteritems()
return self.__values__.iteritems()
def __dir__(self):
return self.__dict__.keys()
return self.__values__.keys()
def __repr__(self):
return 'Config(%s)' % str(self.__dict__)
return 'Config(%s)' % str(self.__values__)
def conf_from_module(module):
if isinstance(module, str):

View File

@@ -79,7 +79,7 @@ class TestConf(TestCase):
def test_config_repr(self):
conf = configuration.Config({'a':1})
self.assertEqual(repr(conf),"Config({'a': 1, '_filename': ''})")
self.assertEqual(repr(conf),"Config({'a': 1})")
def test_config_from_dict(self):
conf = configuration.conf_from_dict({})
@@ -94,7 +94,7 @@ class TestConf(TestCase):
def test_config_illegal_ids(self):
conf = configuration.Config({})
conf.update_with_module('bad.module_and_underscore')
self.assertEqual(['_filename'], dir(conf))
self.assertEqual([], dir(conf))
def test_config_bad_module(self):
conf = configuration.Config({})