Merge pull request #20 from bnrubin/master

[rainbowstream.git] / rainbowstream / py3patch.py

diff --git a/rainbowstream/py3patch.py b/rainbowstream/py3patch.py
index 543b9a1462a27ef2fb5cb7100058159654b32ea4..604298411069fd7f6c0a678f09928c24c38ba004 100644 (file)
--- a/rainbowstream/py3patch.py
+++ b/rainbowstream/py3patch.py
@@ -9,11 +9,16 @@ try:
 except:
     from io import StringIO, BytesIO
 
+# HTMLParser module
+
+try:
+    from HTMLParser import HTMLParser
+except:
+    from html.parser import HTMLParser
+
 # raw_input and map functiion behaviour
-if sys.version[0]=="3": 
+if sys.version[0] == "3":
     raw_input = input
-    lmap = lambda f,a: list(map(f,a))
+    lmap = lambda f, a: list(map(f, a))
 else:
-    lmap = lambda f,a: map(f,a)
-
-
+    lmap = lambda f, a: map(f, a)