X-Git-Url: https://jfr.im/git/z_archive/twitter.git/blobdiff_plain/89f30c21b39702d330070ed54d2d82377501f472..794d6d442f9c150766568fd1626be46735829c85:/twitter/stream_example.py

diff --git a/twitter/stream_example.py b/twitter/stream_example.py
index 0996ea1..4e5184c 100644
--- a/twitter/stream_example.py
+++ b/twitter/stream_example.py
@@ -9,9 +9,9 @@ import argparse
 
 from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
 from twitter.oauth import OAuth
+from twitter.oauth2 import OAuth2, read_bearer_token_file
 from twitter.util import printNicely
 
-
 def parse_arguments():
 
     parser = argparse.ArgumentParser(description=__doc__ or "")
@@ -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()