Merge pull request #10 from shakefu/namespace_key_access

Namespace key access

pytool/lang.py

@@ -379,6 +379,10 @@ class Namespace(object):
         [[1, 2], 3]
         >>> b.foo
         [[9, 2], 3]
+        >>> # You can access keys using dict-like syntax, which is useful
+        >>> myns.foo.bar = True
+        >>> myns['foo'].bar
+        True
 
     Namespaces are useful!
 
@@ -394,6 +398,10 @@ class Namespace(object):
 
         Added deepcopy capability to Namespaces.
 
+    .. versionadded:: 3.8.0
+
+        Added dict-like access capability to Namespaces.
+
     """
     def __init__(self, obj=None):
         if obj is not None:
@@ -407,6 +415,9 @@ def __getattribute__(self, name):
             value = value.__get__(self, self.__class__)
         return value
 
+    # Allow for dict-like key access
+    __getitem__ = __getattribute__
+
     def __getattr__(self, name):
         # Allow implicit nested namespaces by attribute access
         new_space = Namespace()

test/test_lang.py

@@ -338,6 +338,24 @@ def test_namespaces_reject_top_level_lists():
     pytool.lang.Namespace(obj)
 
 
+def test_namespaces_allow_key_access():
+    obj = {'foo': 1, 'bar': 2}
+    ns = pytool.lang.Namespace(obj)
+    eq_(ns['foo'], 1)
+
+
+def test_namespaces_allow_key_access_for_reserved_words():
+    obj = {'foo': 1, 'bar': 2, 'in': 3}
+    ns = pytool.lang.Namespace(obj)
+    eq_(ns['in'], 3)
+
+
+def test_namespaces_allow_key_access_for_nested_reserved_words():
+    obj = {'foo': {'in': {'bar': 1}}}
+    ns = pytool.lang.Namespace(obj)
+    eq_(ns.foo['in'].bar, 1)
+
+
 def test_hashed_singleton_no_args():
     t = HashedSingleton()
     ok_(t is HashedSingleton())