Merge pull request tweepy#218 from inactivist/streaming-fixes

joshthecoder · joshthecoder · commit bda2cf3c4201 · 2012-11-23T19:07:16.000-08:00
Add stall_warnings parameter support to streaming endpoint
diff --git a/tweepy/streaming.py b/tweepy/streaming.py
@@ -135,9 +135,8 @@ def _run(self):
             raise
 
     def _data(self, data):
-        for d in [dt for dt in data.split('\n') if dt]:
-            if self.listener.on_data(d) is False:
-                self.running = False
+        if self.listener.on_data(data) is False:
+            self.running = False
 
     def _read_loop(self, resp):
 
@@ -208,7 +207,8 @@ def sample(self, count=None, async=False):
             self.url += '&count=%s' % count
         self._start(async)
 
-    def filter(self, follow=None, track=None, async=False, locations=None, count = None):
+    def filter(self, follow=None, track=None, async=False, locations=None, 
+        count = None, stall_warnings=False):
         self.parameters = {}
         self.headers['Content-type'] = "application/x-www-form-urlencoded"
         if self.running:
@@ -223,6 +223,8 @@ def filter(self, follow=None, track=None, async=False, locations=None, count = N
             self.parameters['locations'] = ','.join(['%.2f' % l for l in locations])
         if count:
             self.parameters['count'] = count
+        if stall_warnings:
+            self.parameters['stall_warnings'] = stall_warnings
         self.body = urlencode_noplus(self.parameters)
         self.parameters['delimited'] = 'length'
         self._start(async)
