reupload

env/lib/python3.6/site-packages/twitter/stream_example.py

@@ -0,0 +1,92 @@
+"""
+Example program for the Stream API. This prints public status messages
+from the "sample" stream as fast as possible. Use -h for help.
+"""
+
+from __future__ import print_function
+
+import argparse
+
+from twitter.stream import TwitterStream, Timeout, HeartbeatTimeout, Hangup
+from twitter.oauth import OAuth
+from twitter.util import printNicely
+
+
+def parse_arguments():
+
+    parser = argparse.ArgumentParser(description=__doc__ or "")
+
+    parser.add_argument('-t',  '--token', required=True,
+                        help='The Twitter Access Token.')
+    parser.add_argument('-ts', '--token-secret', required=True,
+                        help='The Twitter Access Token Secret.')
+    parser.add_argument('-ck', '--consumer-key', required=True,
+                        help='The Twitter Consumer Key.')
+    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('-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()
+
+    # 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',
+                               **stream_args)
+        tweet_iter = stream.user(**query_args)
+    elif args.site_stream:
+        stream = TwitterStream(auth=auth, domain='sitestream.twitter.com',
+                               **stream_args)
+        tweet_iter = stream.site(**query_args)
+    else:
+        stream = TwitterStream(auth=auth, **stream_args)
+        if args.track_keywords:
+            tweet_iter = stream.statuses.filter(**query_args)
+        else:
+            tweet_iter = stream.statuses.sample()
+
+    # Iterate over the sample stream.
+    for tweet in tweet_iter:
+        # You must test that your tweet has text. It might be a delete
+        # or data message.
+        if tweet is None:
+            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()