mytwitter.py

__author__ = 'jpatdalton'


import twitter
import columns


access_token = ''
access_token_secret = ''
consumer_secret = ''
consumer_key = ''


def get_twitter_stats(artists, worksheet, end):

    api = twitter.Api(consumer_key=consumer_key,
                          consumer_secret=consumer_secret,
                          access_token_key=access_token,
                          access_token_secret=access_token_secret)
    col = columns.twitter
    cell_list_likes = worksheet.range(col+'2:' + col+end)
    for n in xrange(len(artists)):
        name = artists[n]
        try:
            result = api.GetUsersSearch(name,1,1)
            user_id = result[0].id
            user = api.GetUser(user_id)
            if user.verified:
                followers = user.followers_count
            else:
                print 'No verified Twitter account for ' + name
                followers = 'No verified account'
            cell_list_likes[n].value = followers
        except Exception, e:
            print e, 'Twitter Exception on artist [' + name + ']'

    worksheet.update_cells(cell_list_likes)


def get_followers(user_id):
    followers = 0
    try:
        api = twitter.Api(consumer_key=consumer_key,
                              consumer_secret=consumer_secret,
                              access_token_key=access_token,
                              access_token_secret=access_token_secret)
        user = api.GetUser(user_id)
        followers = user.followers_count
    except Exception, e:
        print e, 'Error in getting followers twitter', user_id
    return followers

def get_id(name):
    api = twitter.Api(consumer_key=consumer_key,
                          consumer_secret=consumer_secret,
                          access_token_key=access_token,
                          access_token_secret=access_token_secret)
    the_id = 'None'
    try:
        result = api.GetUsersSearch(name,1,1)
        user_id = result[0].id
        user = api.GetUser(user_id)
        if user.verified:
            the_id = user_id
    except Exception, e:
        print e, 'Cant get Twitter id for: ' + name
    return the_id

'''
result = api.GetUsersSearch("HIBtheBomb",1,1)
result[0].name
user_id = result[0].id
user = api.GetUser(user_id)
user.followers_count
'''