Make etree work with namespaced root

jgraham · jgraham · commit 6fdd5d7dc7e7 · 2010-03-13T00:15:03.000+01:00
diff --git a/src/html5lib/inputstream.py b/src/html5lib/inputstream.py
@@ -158,7 +158,6 @@ def __init__(self, source, encoding=None, parseMeta=True, chardet=True):
         if (self.charEncoding[0] is None):
             self.charEncoding = self.detectEncoding(parseMeta, chardet)
 
-
         self.reset()
 
     def reset(self):
diff --git a/src/html5lib/treebuilders/etree.py b/src/html5lib/treebuilders/etree.py
@@ -321,7 +321,11 @@ def getDocument(self):
             if fullTree:
                 return self.document._element
             else:
-                return self.document._element.find("html")
+                if self.defaultNamespace is not None:
+                    return self.document._element.find(
+                        "{%s}html"%self.defaultNamespace)
+                else:
+                    return self.document._element.find("html")
         
         def getFragment(self):
             return _base.TreeBuilder.getFragment(self)._element
