add top level parseFragment function

jgraham · jgraham · commit 4cbd2bfb2399 · 2009-09-09T22:50:07.000+02:00
diff --git a/src/html5lib/__init__.py b/src/html5lib/__init__.py
@@ -11,6 +11,6 @@
 p = html5lib.HTMLParser()
 tree = p.parse(f) 
 """
-from html5parser import HTMLParser, parse
+from html5parser import HTMLParser, parse, parseFragment
 from treebuilders import getTreeBuilder
 from serializer import serialize
diff --git a/src/html5lib/html5parser.py b/src/html5lib/html5parser.py
@@ -26,6 +26,12 @@ def parse(doc, treebuilder="simpletree", encoding=None,
     p = HTMLParser(tb, namespaceHTMLElements=namespaceHTMLElements)
     return p.parse(doc, encoding=encoding)
 
+def parseFragment(doc, container="div", treebuilder="simpletree", encoding=None, 
+                  namespaceHTMLElements=True):
+    tb = treebuilders.getTreeBuilder(treebuilder)
+    p = HTMLParser(tb, namespaceHTMLElements=namespaceHTMLElements)
+    return p.parseFragment(doc, container=container, encoding=encoding)
+
 class HTMLParser(object):
     """HTML parser. Generates a tree structure from a stream of (possibly
         malformed) HTML"""
