X-Git-Url: https://vcs.fsf.org/?p=rainbowstream.git;a=blobdiff_plain;f=rainbowstream%2Fpy3patch.py;h=adfbff281ebe64f1bb52b1b6fa7aa792cfddba48;hp=83e5f0a1845bc4d781d2192a10f66ba71c77e208;hb=3f6bfc205123dfb813405ef72ff38d9f67590565;hpb=18df6e7fa892cf85dda074a7044ce586ecf17c77

diff --git a/rainbowstream/py3patch.py b/rainbowstream/py3patch.py
index 83e5f0a..adfbff2 100644
--- a/rainbowstream/py3patch.py
+++ b/rainbowstream/py3patch.py
@@ -10,18 +10,19 @@ except:
     from io import StringIO, BytesIO
 
 # HTMLParser module
-
-try:
+if sys.version[0] == "2":
     from HTMLParser import HTMLParser
-    def unescape(s):
-        p = HTMLParser()
-        return p.unescape(s)
-except:
-    from html import unescape
+else:
+    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 functiion behaviour
-if sys.version[0] == "3":
+# 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))
-else:
-    lmap = lambda f, a: map(f, a)