[svn r4032] Small cleanups to lxml.html.diff.

Export lxml.html.parse

--HG--
branch : trunk

1 files changed, 6 insertions, 2 deletions

diff --git a/src/lxml/html/diff.py b/src/lxml/html/diff.py
index 92a1e1b6..bb495e49 100644
--- a/src/lxml/html/diff.py
+++ b/src/lxml/html/diff.py
@@ -1,6 +1,7 @@
 import difflib
 from lxml import etree
 from lxml.html import fragment_fromstring
+from lxml.etree import _Element
 import cgi
 import re
 
@@ -507,7 +508,7 @@ class href_token(token):
     hide_when_equal = True
 
     def html(self):
-        return 'Link: %s' % self
+        return ' Link: %s' % self
 
 def tokenize(html, include_hrefs=True):
     """
@@ -524,7 +525,10 @@ def tokenize(html, include_hrefs=True):
 
     If include_hrefs is true, then the href attribute of <a> tags is
     included as a special kind of diffable token."""
-    body_el = parse_html(html, cleanup=True)
+    if isinstance(html, _Element):
+        body_el = html
+    else:
+        body_el = parse_html(html, cleanup=True)
     # Then we split the document into text chunks for each tag, word, and end tag:
     chunks = flatten_el(body_el, skip_tag=True, include_hrefs=include_hrefs)
     # Finally re-joining them into token objects: