suggest fix for #20

diff --git a/rainbowstream/draw.py b/rainbowstream/draw.py
index f23e4cacd70d8ab9558573289590a924673e3a8e..4360d469c6b859e948f930f817cffa86787e4dca 100644 (file)
--- a/rainbowstream/draw.py
+++ b/rainbowstream/draw.py
@@ -20,11 +20,6 @@ db = RainbowDB()
 g = {}
 
 
-def unescape(s):
-    p = HTMLParser()
-    return p.unescape(s)
-
-
 def init_cycle():
     """
     Init the cycle
@@ -163,11 +158,12 @@ def draw(t, iot=False, keyword=None, check_semaphore=False, fil=[], ig=[]):
 
     # Pull extended retweet text
     try:
-        text = 'RT @{0}: {1}'.format(t['retweeted_status']['user']['screen_name'],
-                                     t['retweeted_status']['text'])
+        text = 'RT @' + t['retweeted_status']['user']['screen_name'] + ': ' +\
+            t['retweeted_status']['text']
     except:
         pass
 
+    # Unescape HTML character
     text = unescape(text)
 
     # Get expanded url

diff --git a/rainbowstream/py3patch.py b/rainbowstream/py3patch.py
index 604298411069fd7f6c0a678f09928c24c38ba004..83e5f0a1845bc4d781d2192a10f66ba71c77e208 100644 (file)
--- a/rainbowstream/py3patch.py
+++ b/rainbowstream/py3patch.py
@@ -13,8 +13,11 @@ except:
 
 try:
     from HTMLParser import HTMLParser
+    def unescape(s):
+        p = HTMLParser()
+        return p.unescape(s)
 except:
-    from html.parser import HTMLParser
+    from html import unescape
 
 # raw_input and map functiion behaviour
 if sys.version[0] == "3":