fix serious bug in import HTMLParser in Python > 3.4
[rainbowstream.git] / rainbowstream / py3patch.py
index e1eec6e..adfbff2 100644 (file)
@@ -5,15 +5,24 @@ import sys
 
 # StringIO module
 try:
-    from StringIO import StringIO
+    from StringIO import StringIO, BytesIO
 except:
-    from io import StringIO
+    from io import StringIO, BytesIO
 
-# raw_input and map functiion behaviour
-if sys.version[0]=="3": 
-    raw_input = input
-    lmap = lambda f,a: list(map(f,a))
+# HTMLParser module
+if sys.version[0] == "2":
+    from HTMLParser import HTMLParser
 else:
-    lmap = lambda f,a: map(f,a)
-
+    from html.parser import HTMLParser
+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
+# from html import unescape
+# but it is a far-future story:)
 
+# raw_input and map function behaviour
+if sys.version[0] == "2":
+    lmap = lambda f, a: map(f, a)
+else:
+    raw_input = input
+    lmap = lambda f, a: list(map(f, a))