Oops, left debugging cruft on there.

[z_archive/twitter.git] / twitter / stream_example.py

diff --git a/twitter/stream_example.py b/twitter/stream_example.py
index ac0c8b7350255159c563062b06aad80aba9ebe67..8869c15f8515adec138e35f5f1b7ce01b46009d7 100644 (file)
--- a/twitter/stream_example.py
+++ b/twitter/stream_example.py
@@ -7,7 +7,7 @@ from __future__ import print_function
 
 import argparse
 
-from twitter.stream import TwitterStream, Timeout
+from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
 from twitter.oauth import OAuth
 from twitter.util import printNicely
 
@@ -22,30 +22,38 @@ def parse_arguments():
     parser.add_argument('-cs', '--consumer-secret', required=True, help='The Twitter Consumer Secret.')
     parser.add_argument('-us', '--user-stream', action='store_true', help='Connect to the user stream endpoint.')
     parser.add_argument('-ss', '--site-stream', action='store_true', help='Connect to the site stream endpoint.')
-    parser.add_argument('-to', '--timeout', help='Timeout for the stream (seconds)')
-    parser.add_argument('-nb', '--no-block', action='store_true', help='Set stream to non-blocking')
-    parser.add_argument('-tt', '--track-keywords', help='Search the stream for specific text')
+    parser.add_argument('-to', '--timeout', help='Timeout for the stream (seconds).')
+    parser.add_argument('-ht', '--heartbeat-timeout', help='Set heartbeat timeout.', default=90)
+    parser.add_argument('-nb', '--no-block', action='store_true', help='Set stream to non-blocking.')
+    parser.add_argument('-tt', '--track-keywords', help='Search the stream for specific text.')
     return parser.parse_args()
 
 def main():
     args = parse_arguments()
 
-    if not all((args.token, args.token_secret, args.consumer_key, args.consumer_secret)):
-        print(__doc__)
-        return 2
-
     # When using twitter stream you must authorize.
     auth = OAuth(args.token, args.token_secret, args.consumer_key, args.consumer_secret)
+
+    # These arguments are optional:
+    stream_args = dict(
+        timeout=args.timeout,
+        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', timeout=args.timeout, block=not args.no_block)
-        tweet_iter = stream.user()
+        stream = TwitterStream(auth=auth, domain='userstream.twitter.com', **stream_args)
+        tweet_iter = stream.user(**query_args)
     elif args.site_stream:
-        stream = TwitterStream(auth=auth, domain='sitestream.twitter.com', timeout=args.timeout, block=not args.no_block)
-        tweet_iter = stream.site()
+        stream = TwitterStream(auth=auth, domain='sitestream.twitter.com', **stream_args)
+        tweet_iter = stream.site(**query_args)
     else:
-        stream = TwitterStream(auth=auth, timeout=args.timeout, block=not args.no_block)
+        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()
 
@@ -57,8 +65,14 @@ def main():
             printNicely("-- None --")
         elif tweet is Timeout:
             printNicely("-- Timeout --")
+        elif tweet is HeartbeatTimeout:
+            printNicely("-- Heartbeat Timeout --")
+        elif tweet is Hangup:
+            printNicely("-- Hangup --")
         elif tweet.get('text'):
             printNicely(tweet['text'])
+        else:
+            printNicely("-- Some data: " + str(tweet))
 
 if __name__ == '__main__':
     main()