Allow alternate rate limit notifications

tweepy · Nov 8, 2016 · c713b4f · c713b4f
1 parent 5fa066b
commit c713b4f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tweepy/binder.py b/tweepy/binder.py
@@ -160,7 +160,10 @@ def execute(self):
                                 sleep_time = self._reset_time - int(time.time())
                                 if sleep_time > 0:
                                     if self.wait_on_rate_limit_notify:
-                                        log.warning("Rate limit reached. Sleeping for: %d" % sleep_time)
+                                        if callable(self.wait_on_rate_limit_notify):
+                                            self.wait_on_rate_limit_notify(sleep_time)
+                                        else:
+                                            log.warning("Rate limit reached. Sleeping for: %d" % sleep_time)
                                     time.sleep(sleep_time + 5)  # sleep for few extra sec
 
                 # if self.wait_on_rate_limit and self._reset_time is not None and \