Add python2 compatibility to PR #66

[rainbowstream.git] / rainbowstream / py3patch.py

diff --git a/rainbowstream/py3patch.py b/rainbowstream/py3patch.py
index 4ba959110dd8a202f8c410cda9064317c5e3822b..7e71e2a04fa897f282e4e53bf0a0a0c0c42d3c89 100644 (file)
--- a/rainbowstream/py3patch.py
+++ b/rainbowstream/py3patch.py
@@ -1,17 +1,13 @@
 import sys
 
 # Library compatibility
-# StringIO module
-try:
-    from StringIO import StringIO, BytesIO
-except:
-    from io import StringIO, BytesIO
-
-# HTMLParser module
 if sys.version[0] == "2":
     from HTMLParser import HTMLParser
+    from urllib2 import URLError
 else:
     from html.parser import HTMLParser
+    from urllib.error import URLError
+
 unescape = HTMLParser().unescape
 # According to https://github.com/python/cpython/blob/master/Lib/html/parser.py#L547 ,
 # in python 3.5 maybe I should use