Get some sort of serializer working for elementtree.

jgraham · jgraham · commit c3347a4f8c81 · 2006-12-30T23:32:42.000Z
--HG--
extra : convert_revision : svn%3Aacbfec75-9323-0410-a652-858a13e371e0/trunk%40391
diff --git a/src/treebuilders/etree.py b/src/treebuilders/etree.py
@@ -72,7 +72,6 @@ def cloneNode(self):
         return element
 
 class Comment(Element):
-
     def __init__(self, data):
         self._element = ElementTree.Comment(data)
         self.name = None
@@ -87,20 +86,46 @@ def _setData(self, value):
     data = property(_getData, _setData)
 
 class DocumentType(Element):
-    pass
+    def __init__(self, name):
+        self._element = ElementTree.Element(DocumentType)
+        self.text = "html"
 
 class Document(Element):
     def __init__(self):
         Element.__init__(self, "")
 
+def testSerializer(element):
+    print type(ElementTree.Comment)
+    rv = []
+    rv.append("#document")
+    def serializeElement(element, indent=0):
+        if element.tag is DocumentType:
+            rv.append("|%s<!DOCTYPE %s>"%(' '*indent, element.text))
+        elif element.tag is ElementTree.Comment:
+            rv.append("|%s<!-- %s -->"%(' '*indent, element.text))
+        else:
+            rv.append("|%s<%s>"%(' '*indent, element.tag))
+            if hasattr(element, "attrib"):
+                for name, value in element.attrib.iteritems():
+                    rv.append('|%s%s="%s"' % (' '*(indent+2), name, value))
+            if element.text:
+                rv.append("|%s\"%s\"" %(' '*(indent+2), element.text))
+        indent += 2
+        for child in element.getchildren():
+            serializeElement(child, indent)
+        if element.tail:
+            rv.append("|%s\"%s\"" %(' '*(indent-2), element.tail))
+    serializeElement(element)
+    return "\n".join(rv)
+
 class TreeBuilder(base.TreeBuilder):
     documentClass = Document
     doctypeClass = DocumentType
     elementClass = Element
     commentClass = Comment
 
     def testSerializer(self, element):
-        ElementTree.tostring(element)
+        return testSerializer(element)
 
     def getDocument(self):
-        return self.document.getchildren()[0]._element
+        return self.document._element.getchildren()[0]
