# -*- coding: utf-8 -*-
import tweepy
from tweepy import OAuthHandler
from tweepy.api import API
from keys import keys
import json


CONSUMER_KEY = keys['CONSUMER_KEY']
CONSUMER_SECRET = keys['CONSUMER_SECRET']
ACCESS_KEY = keys['ACCESS_KEY']
ACCESS_SECRET = keys['ACCESS_SECRET']

auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
auth.set_access_token(ACCESS_KEY, ACCESS_SECRET)
api = tweepy.API(auth)
mainMenu = (CONSUMER_KEY, CONSUMER_SECRET, ACCESS_KEY, ACCESS_SECRET)
s2s = (CONSUMER_KEY, CONSUMER_SECRET, ACCESS_KEY, ACCESS_SECRET, auth, api)


def userInput():
    print '-------MENU-------'
    option = int(raw_input('[1]: Get info for specific user\n[2]: Listen for specific tweets\n[3]: NOTHING\n------------------\n'))
    if option == 1:
        username = raw_input('\n@')
        data = api.get_user(username)
        twt = data.status.text
        print 'Getting statistics for %s...' % username
        print 'Followers: ' + str(data.followers_count)
        print 'Tweets: ' + str(data.statuses_count)
        print 'Favorites: ' + str(data.favourites_count)
        print 'Friends: ' + str(data.friends_count)
        print 'Appears on ' + str(data.listed_count) + ' lists'
        print 'Latest tweet: ' +str(twt)
    elif option == 2:
        class s2s(tweepy.StreamListener):
            def __init__(self, api=None):
                self.api = api or API()
                self.n = 0
                self.m = 10
            def on_status(self, status):
                print status.text.encode('utf8')
                self.n = self.n+1
                if self.n < self.m:return True
                else:
                    print '\ntweets streamed: \n'+str(self.n)
                    return False
        buzzwords = raw_input('\nInput words to stream.\n')
        stream = tweepy.streaming.Stream(auth, s2s())
        stream.filter(track= buzzwords, languages=['en'])
        s2s()
    elif option == 3:
        print 'test'
    else:
        print "Error"
userInput()