Implement StreamListener.on_connect callback.

joshthecoder · joshthecoder · commit 5e06813723ad · 2013-05-18T23:58:26.000-07:00
diff --git a/tweepy/streaming.py b/tweepy/streaming.py
@@ -22,6 +22,15 @@ class StreamListener(object):
     def __init__(self, api=None):
         self.api = api or API()
 
+    def on_connect(self):
+        """Called once connected to streaming server.
+
+        This will be invoked once a successful response
+        is received from the server. Allows the listener
+        to perform some work prior to entering the read loop.
+        """
+        pass
+
     def on_data(self, data):
         """Called when raw data is received from connection.
 
@@ -114,6 +123,7 @@ def _run(self):
                     sleep(self.retry_time)
                 else:
                     error_counter = 0
+                    self.listener.on_connect()
                     self._read_loop(resp)
             except timeout:
                 if self.listener.on_timeout() == False:
