# -*- coding: utf-8 -*-
import time
import tweepy
from tweepy import OAuthHandler
from tweepy import Stream
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)
stream2screen = (CONSUMER_KEY, CONSUMER_SECRET, ACCESS_KEY, ACCESS_SECRET, auth, api)

def userInput():
    print "Username to look up"
    option= int(input("1: Get info for specific user\n2: NOTHING\n3: NOTHING\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 'Tweet: ' +str(twt)
    elif option == 2:
        print "test_1"
    elif option == 3:
        print "test_2"
    else:
        print "Error"
userInput()


class stream2screen(tweepy.StreamListener):
    def __init__(self, api=None):
        self.api = api or API()
        self.n = 0
        self.m = 1
    def on_status(self, status):
       print status.text.encode('utf8')
       self.n = self.n+1
       if self.n < self.m: return True
       else:
           print 'tweets = '+str(self.n)
           return False

stream = tweepy.streaming.Stream(auth, stream2screen())
stream.filter(track=[''], languages=['en'])

stream2screen()