Changing non-xml attributes

jgraham · jgraham · commit ad364291cb29 · 2009-04-19T20:58:52.000Z
--HG--
branch : svgmathml
extra : convert_revision : svn%3Aacbfec75-9323-0410-a652-858a13e371e0/branches/svgmathml%401301
diff --git a/src/html5lib/treebuilders/dom.py b/src/html5lib/treebuilders/dom.py
@@ -31,11 +31,20 @@ def __setitem__(self, name, value):
             self.element.setAttribute(infoset_filter.coerceAttribute(name),
                                       infoset_filter.coerceCharacters(value))
         def items(self):
-            return self.element.attributes.items()
+            return [(infoset_filter.fromXmlName(item[0]), item[1]) for item in
+                     self.element.attributes.items()]
         def keys(self):
-            return self.element.attributes.keys()
+            return [infoset_filter.fromXmlName(item) for item in
+                    self.element.attributes.keys()]
         def __getitem__(self, name):
+            name = infoset_filter.toXmlName(name)
             return self.element.getAttribute(name)
+
+        def __contains__(self, name):
+            if isinstance(name, tuple):
+                raise NotImplementedError
+            else:
+                return self.element.hasAttribute(infoset_filter.toXmlName(name))
     
     class NodeBuilder(_base.Node):
         def __init__(self, element):
