Merge pull request #40 from reverbc/support-membership-operator

support 'in' and 'not in' operator

AUTHORS

@@ -4,3 +4,4 @@ Christian Stefanescu <chris@0chris.com>
 
 Florian Idelberger <flo@terrorpop.de>
 Apalala <apalala@gmail.com>
+Reverb Chu <reverbc@me.com>

tests/tests.py

@@ -14,13 +14,21 @@ def test_basic(self):
         self.assert_(o.a is not None)
         self.assert_(o.a.b is not None)
         self.assert_(o.a.c is not None)
+        self.assert_('a' in o)
+        self.assert_('b' in o.a)
+        self.assert_('c' in o.a)
+        self.assert_('d' not in o.a)
 
     def test_basic_with_decl(self):
         o = untangle.parse("<?xml version='1.0'?><a><b/><c/></a>")
         self.assert_(o is not None)
         self.assert_(o.a is not None)
         self.assert_(o.a.b is not None)
         self.assert_(o.a.c is not None)
+        self.assert_('a' in o)
+        self.assert_('b' in o.a)
+        self.assert_('c' in o.a)
+        self.assert_('d' not in o.a)
 
     def test_with_attributes(self):
         o = untangle.parse('''

untangle.py

@@ -120,6 +120,9 @@ def __dir__(self):
     def __len__(self):
         return len(self.children)
 
+    def __contains__(self, key):
+        return key in dir(self)
+
 
 class Handler(handler.ContentHandler):
     """