X-Git-Url: https://jfr.im/git/z_archive/twitter.git/blobdiff_plain/03d8511cd4f14723ffe3e78ff92cbcbbea8080df..1ff502360f64181a10677a84737cf1aaea0ec4f8:/twitter/stream_example.py

diff --git a/twitter/stream_example.py b/twitter/stream_example.py
index 0996ea1..8869c15 100644
--- a/twitter/stream_example.py
+++ b/twitter/stream_example.py
@@ -40,16 +40,20 @@ def main():
         block=not args.no_block,
         heartbeat_timeout=args.heartbeat_timeout)
 
+    query_args = dict()
+    if args.track_keywords:
+        query_args['track'] = args.track_keywords
+
     if args.user_stream:
         stream = TwitterStream(auth=auth, domain='userstream.twitter.com', **stream_args)
-        tweet_iter = stream.user()
+        tweet_iter = stream.user(**query_args)
     elif args.site_stream:
         stream = TwitterStream(auth=auth, domain='sitestream.twitter.com', **stream_args)
-        tweet_iter = stream.site()
+        tweet_iter = stream.site(**query_args)
     else:
         stream = TwitterStream(auth=auth, **stream_args)
         if args.track_keywords:
-            tweet_iter = stream.statuses.filter(track=args.track_keywords)
+            tweet_iter = stream.statuses.filter(**query_args)
         else:
             tweet_iter = stream.statuses.sample()