Twitter Error display bug in Python 3

[rainbowstream.git] / rainbowstream / util.py

diff --git a/rainbowstream/util.py b/rainbowstream/util.py
index 232845c..e8d3c74 100644 (file)
--- a/rainbowstream/util.py
+++ b/rainbowstream/util.py
@@ -1,16 +1,20 @@
 import json
 
 from twitter.util import printNicely
 import json
 
 from twitter.util import printNicely

+from .colors import *

 
 

-from .colors import magenta

 
 

-
-def printTwitterErrors(twitterException):
+def detail_twitter_error(twitterException):
+    """
+    Display Twitter Errors nicely
+    """

     try:
     try:

-        loadedJson = json.loads(twitterException.response_data)
+        # twitterException.response_data can be byte string on Python 3
+        # or nornal dict on Python 2
+        loadedJson = json.loads(twitterException.response_data.decode('utf8'))

         for m in loadedJson.get('errors', dict()):
         for m in loadedJson.get('errors', dict()):

-            printNicely(
-                magenta("Error " + str(m.get('code')) + ": " + m.get('message')))
-    except ValueError:
-        printNicely(
-                magenta( "Error: " + twitterException.response_data))
+            info = "Error " + str(m.get('code')) + ": " + m.get('message')
+            printNicely(yellow(info))
+    except:
+        info = "Error: " + twitterException.response_data.decode('utf8')
+        printNicely(yellow(info))