- stream = TwitterStream(auth=OAuth(args.token, args.token_secret, args.consumer_key, args.consumer_secret))
+ 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, timeout=60.0)
+ tweet_iter = stream.statuses.sample()