X-Git-Url: https://vcs.fsf.org/?p=rainbowstream.git;a=blobdiff_plain;f=rainbowstream%2Fpy3patch.py;h=7e71e2a04fa897f282e4e53bf0a0a0c0c42d3c89;hp=543b9a1462a27ef2fb5cb7100058159654b32ea4;hb=e449db3b899ef5ecf0c41e9176504d6cec5fd55a;hpb=77f1d210dfd44f5b6c33cc58150b633643621bb0 diff --git a/rainbowstream/py3patch.py b/rainbowstream/py3patch.py index 543b9a1..7e71e2a 100644 --- a/rainbowstream/py3patch.py +++ b/rainbowstream/py3patch.py @@ -1,19 +1,28 @@ -""" - Python 3 supports -""" import sys -# StringIO module -try: - from StringIO import StringIO, BytesIO -except: - 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)) +# Library compatibility +if sys.version[0] == "2": + from HTMLParser import HTMLParser + from urllib2 import URLError else: - lmap = lambda f,a: map(f,a) + 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 +# from html import unescape +# but it is a far-future story:) +# Function compatibility +# xrange, raw_input, map ,unicde +if sys.version[0] == "2": + lmap = lambda f, a: map(f, a) + str2u = lambda x: x.decode('utf-8') + u2str = lambda x: x.encode('utf-8') +else: + xrange = range + raw_input = input + lmap = lambda f, a: list(map(f, a)) + str2u = u2str = lambda x: x