Simple Twython streaming search

main.py

"""
See Streaming API: https://twython.readthedocs.io/en/latest/usage/streaming_api.html

"""
import os
import sys
from twython import TwythonStreamer

class TweetPrinter(TwythonStreamer):
    def on_success(self, data):
        if 'text' in data:
            print("\n--- @{} --------".format(data['user']['screen_name']))
            print(data['text'])

    def on_error(self, status_code, data):
        print(status_code)

        
if __name__ == "__main__":
    if len(sys.argv) == 2:
        stream = MyStreamer(
            os.environ['TWITTER_CONSUMER_KEY'],
            os.environ['TWITTER_CONSUMER_SECRET'],
            os.environ['TWITTER_OAUTH_TOKEN'],
            os.environ['TWITTER_OAUTH_TOKEN_SECRET']
        )
        stream.statuses.filter(track=sys.argv[1])
    else:
        print("USAGE: python main.py <search>")