Use six to import urlparse

Drew Hubl · CaptainCodeman · commit a1abf71b147b · 2014-11-11T08:49:12.000-07:00
diff --git a/html5lib/sanitizer.py b/html5lib/sanitizer.py
@@ -2,10 +2,7 @@
 
 import re
 from xml.sax.saxutils import escape, unescape
-try:
-    from urllib.parse import urlparse
-except ImportError:
-    from urlparse import urlparse
+from six.moves import urllib_parse as urlparse
 
 from .tokenizer import HTMLTokenizer
 from .constants import tokenTypes
@@ -196,7 +193,7 @@ def allowed_token(self, token, token_type):
                                        unescape(attrs[attr])).lower()
                 # remove replacement characters from unescaped characters
                 val_unescaped = val_unescaped.replace("\ufffd", "")
-                uri = urlparse(val_unescaped)
+                uri = urlparse.urlparse(val_unescaped)
                 if uri:
                     if uri.scheme not in self.allowed_protocols:
                         del attrs[attr]
