import time

from getpass import getpass

from textwrap import TextWrapper

import tweepy

import webbrowser

CONSUMER_KEY = "54ThDZhpEjokcMgHJOMnQA"

CONSUMER_SECRET = "wUoL9UL2T87tfc97R0Dff2EaqRzpJ5XGdmaN2XK3udA"

class StreamWatcherListener(tweepy.StreamListener):

    status_wrapper = TextWrapper(width=60, initial_indent='    ', subsequent_indent='    ')

    def on_status(self, status):

        try:

            print self.status_wrapper.fill(status.text)

            print '\n %s  %s  via %s\n' % (status.author.screen_name, status.created_at, status.source)

        except:

            # Catch any unicode errors while printing to console

            # and just ignore them to avoid breaking application.

            pass

    def on_error(self, status_code):

        print 'An error has occured! Status code = %s' % status_code

        return True  # keep stream alive

    def on_timeout(self):

        print 'Snoozing Zzzzzz'

def main():

    auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)

    auth_url = auth.get_authorization_url()

    print 'Please authorize: ' + auth_url

    webbrowser.open(auth_url)

    # Prompt for login credentials and setup stream object

    verifier = raw_input('PIN: ').strip()

    auth.get_access_token(verifier)

    stream = tweepy.Stream(auth, StreamWatcherListener(), timeout=None)

    # Prompt for mode of streaming

    valid_modes = ['sample', 'filter']

    while True:

        mode = raw_input('Mode? [sample/filter] ')

        if mode in valid_modes:

            break

        print 'Invalid mode! Try again.'

    if mode == 'sample':

        stream.sample()

    elif mode == 'filter':

        follow_list = raw_input('Users to follow (comma separated): ').strip()

        track_list = raw_input('Keywords to track (comma seperated): ').strip()

        if follow_list:

            follow_list = [u for u in follow_list.split(',')]

        else:

            follow_list = None

        if track_list:

            track_list = [k for k in track_list.split(',')]

        else:

            track_list = None

        stream.filter(follow_list, track_list)

if __name__ == '__main__':

    try:

        main()

    except KeyboardInterrupt:

        print '\nGoodbye!'