Add html5lib.serialize.serialize and html5lib.serialize

gsnedders · gsnedders · commit 6cd3237cdb29 · 2009-03-21T15:24:06.000Z
--HG--
extra : convert_revision : svn%3Aacbfec75-9323-0410-a652-858a13e371e0/trunk%401273
diff --git a/src/html5lib/__init__.py b/src/html5lib/__init__.py
@@ -13,5 +13,6 @@
 """
 from html5parser import HTMLParser, parse
 from treebuilders import getTreeBuilder
+from serializer import serialize
 
 from liberalxmlparser import XMLParser, XHTMLParser
diff --git a/src/html5lib/serializer/__init__.py b/src/html5lib/serializer/__init__.py
@@ -1,3 +1,17 @@
 
+from html5lib import treewalkers
+
 from htmlserializer import HTMLSerializer
 from xhtmlserializer import XHTMLSerializer
+
+def serialize(input, tree="simpletree", format="html", encoding=None,
+              **serializer_opts):
+    # XXX: Should we cache this?
+    walker = treewalkers.getTreeWalker(tree) 
+    if format == "html":
+        s = HTMLSerializer(**serializer_opts)
+    elif format == "xhtml":
+        s = XHTMLSerializer(**serializer_opts)
+    else:
+        raise ValueError, "type must be either html or xhtml"
+    return s.render(walker(input), encoding)
