Add html serializer to simpletree

jgraham · jgraham · commit 4bc404fd5c03 · 2007-04-11T22:25:43.000Z
--HG--
extra : convert_revision : svn%3Aacbfec75-9323-0410-a652-858a13e371e0/trunk%40547
diff --git a/parse.py b/parse.py
@@ -88,7 +88,6 @@ def parse():
         sys.stdout.write("\n\nRun took: %fs (plus %fs to print the output)"%(t1-t0, t2-t1))
     else:
         document = p.parse(f)
-        print document
         printOutput(p, document, opts)
 
 def printOutput(parser, document, opts):
diff --git a/src/treebuilders/simpletree.py b/src/treebuilders/simpletree.py
@@ -4,6 +4,7 @@
 
 # Really crappy basic implementation of a DOM-core like thing
 class Node(_base.Node):
+    type = -1
     def __init__(self, name):
         self.name = name
         self.parent = None
@@ -23,9 +24,6 @@ def __unicode__(self):
     def toxml(self):
         raise NotImplementedError
 
-    def __repr__(self):
-        return "<%s %s>" % (self.__class__, self.name)
-
     def printTree(self, indent=0):
         tree = '\n|%s%s' % (' '* indent, unicode(self))
         for child in self.childNodes:
@@ -75,6 +73,7 @@ def hasContent(self):
         return bool(self.childNodes)
 
 class Document(Node):
+    type = 1
     def __init__(self):
         Node.__init__(self, None)
 
@@ -100,6 +99,7 @@ def printTree(self):
         return tree
 
 class DocumentType(Node):
+    type = 3
     def __init__(self, name):
         Node.__init__(self, name)
 
@@ -112,6 +112,7 @@ def hilite(self):
         return '<code class="markup doctype">&lt;!DOCTYPE %s></code>' % self.name
 
 class TextNode(Node):
+    type = 4
     def __init__(self, value):
         Node.__init__(self, None)
         self.value = value
@@ -125,6 +126,7 @@ def toxml(self):
     hilite = toxml
 
 class Element(Node):
+    type = 3
     def __init__(self, name):
         Node.__init__(self, name)
         self.attributes = {}
@@ -170,6 +172,7 @@ def printTree(self, indent):
         return tree
 
 class CommentNode(Node):
+    type = 5
     def __init__(self, data):
         Node.__init__(self, None)
         self.data = data
@@ -184,9 +187,36 @@ def hilite(self):
         return '<code class="markup comment">&lt;!--%s--></code>' % escape(self.data)
 
 class DocumentFragment(Document):
+    type = 2
     def __unicode__(self):
         return "#document-fragment"
 
+class HTMLSerializer(object):
+    def serialize(self, node):
+        rv = self.serializeNode(node)
+        for child in node.childNodes:
+            rv += self.serialize(child)
+        if isinstance(node, Element):
+            rv += "</%s>\n"%node.name
+        return rv
+    
+    def serializeNode(self, node):
+        if node.type == TextNode.type:
+            rv = node.value
+        elif node.type == Element.type:
+            rv = "<%s"%node.name
+            if node.attributes:
+                rv = rv+"".join([" %s='%s'"%(key, escape(value)) for key,value in
+                                 node.attributes.iteritems()])
+            rv += ">"
+        elif node.type == CommentNode.type:
+            rv = "<!-- %s -->" % escape(self.data)        
+        elif node.type == DocumentType.type:
+            rv = "<!DOCTYPE %s>" % node.name
+        else:
+            rv = ""
+        return rv
+
 class TreeBuilder(_base.TreeBuilder):
     documentClass = Document
     doctypeClass = DocumentType
