Handle content-type header charset value for streaming API

[twitter-api-cdsw] / tweepy / error.py

diff --git a/tweepy/error.py b/tweepy/error.py
index 753e2fe676cf2f581a9fe032a71d9c4d638cc471..7827029a55236e7becbc4e445994934c645248c5 100644 (file)
--- a/tweepy/error.py
+++ b/tweepy/error.py
@@ -2,14 +2,30 @@
 # Copyright 2009-2010 Joshua Roesslein
 # See LICENSE for details.
 
+from __future__ import print_function
+
+import six
+
 class TweepError(Exception):
     """Tweepy exception"""
 
     def __init__(self, reason, response=None):
-        self.reason = unicode(reason)
+        self.reason = six.text_type(reason)
         self.response = response
         Exception.__init__(self, reason)
 
     def __str__(self):
         return self.reason
 
+def is_rate_limit_error_message(message):
+    """Check if the supplied error message belongs to a rate limit error."""
+    return isinstance(message, list) \
+        and len(message) > 0 \
+        and 'code' in message[0] \
+        and message[0]['code'] == 88
+
+class RateLimitError(TweepError):
+    """Exception for Tweepy hitting the rate limit."""
+    # RateLimitError has the exact same properties and inner workings
+    # as TweepError for backwards compatibility reasons.
+    pass