USAGE
- twitter-stream-example <username> <password>
+ stream-example -t <token> -ts <token_secret> -ck <consumer_key> -cs <consumer_secret>
"""
from __future__ import print_function
-import sys
+import argparse
-from .stream import TwitterStream
-from .auth import UserPassAuth
-from .util import printNicely
+from twitter.stream import TwitterStream
+from twitter.oauth import OAuth
+from twitter.util import printNicely
-def main(args=sys.argv[1:]):
- if not args[1:]:
- print(__doc__)
- return 1
- # When using twitter stream you must authorize. UserPass or OAuth.
- stream = TwitterStream(auth=UserPassAuth(args[0], args[1]))
+def parse_arguments():
+
+ parser = argparse.ArgumentParser()
+
+ parser.add_argument('-t', '--token', help='The Twitter Access Token.')
+ parser.add_argument('-ts', '--token_secret', help='The Twitter Access Token Secret.')
+ parser.add_argument('-ck', '--consumer_key', help='The Twitter Consumer Key.')
+ parser.add_argument('-cs', '--consumer_secret', 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.')
+
+ return parser.parse_args()
+
+## parse_arguments()
+
+
+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)
+ if args.user_stream:
+ stream = TwitterStream(auth=auth, domain='userstream.twitter.com')
+ tweet_iter = stream.user()
+ elif args.site_stream:
+ stream = TwitterStream(auth=auth, domain='sitestream.twitter.com')
+ tweet_iter = stream.site()
+ else:
+ stream = TwitterStream(auth=auth)
+ tweet_iter = stream.statuses.sample()
# Iterate over the sample stream.
- tweet_iter = stream.statuses.sample()
for tweet in tweet_iter:
# You must test that your tweet has text. It might be a delete
# or data message.
if tweet.get('text'):
printNicely(tweet['text'])
+
+## main()
+
+if __name__ == '__main__':
+ main()