Extend on_data method by including a conditional to process warning messages and...
authorjoausaga <jorgesaldivar@gmail.com>
Wed, 12 Nov 2014 11:05:52 +0000 (12:05 +0100)
committerjoausaga <jorgesaldivar@gmail.com>
Wed, 12 Nov 2014 11:05:52 +0000 (12:05 +0100)
tweepy/streaming.py

index 736cffafb968b7f40c8653d2eb1986ef849d1ae0..b27e8f394206e912260b406ecb70d3ff37b28c1b 100644 (file)
@@ -68,6 +68,9 @@ class StreamListener(object):
         elif 'disconnect' in data:
             if self.on_disconnect(data['disconnect']) is False:
                 return False
+        elif 'warning' in data:
+            if self.on_warning(data['warning']) is False:
+                return False
         else:
             logging.error("Unknown message type: " + str(raw_data))
 
@@ -99,7 +102,7 @@ class StreamListener(object):
         return
 
     def on_limit(self, track):
-        """Called when a limitation notice arrvies"""
+        """Called when a limitation notice arrives"""
         return
 
     def on_error(self, status_code):
@@ -117,6 +120,10 @@ class StreamListener(object):
         https://dev.twitter.com/docs/streaming-apis/messages#Disconnect_messages_disconnect
         """
         return
+    
+    def on_warning(self, notice):
+        """Called when a disconnection warning message arrives"""
+        return
 
 
 class ReadBuffer(object):