vine.py

__author__ = 'jpatdalton'


import requests
import columns

def get_page_id(name):
    api_endpoint = 'https://api.vineapp.com/users/search/' + name
    #api_endpoint = "https://graph.facebook.com/search?q=" + name + "&type=page" + '&access_token='+APP_ID+'|'+APP_SECRET
    data = requests.get(api_endpoint).json()["data"]
    return data["records"][0]["userId"]

def get_user_data(id):
    the_data = 0
    try:
        user_endpoint = 'https://api.vineapp.com/users/profiles/' + str(id)
        data = requests.get(user_endpoint).json()["data"]
        if data["verified"]:
            the_data = data['followerCount']
        else:
            the_data = 'Not Verified'
    except Exception, e:
        print e, 'Vine get user data', id
    return the_data


def get_vine_stats(artists, worksheet, end):
    col = columns.vine
    cell_list_likes = worksheet.range(col+'2:'+col+end)
    for n in xrange(len(artists)):
        name = artists[n]
        try:
            page_id = get_page_id(name)
            data = get_user_data(page_id)
            cell_list_likes[n].value = str(data)
        except Exception, e:
            print e, 'Vine access error ['+name+']'

    worksheet.update_cells(cell_list_likes)

def get_id(name):
    the_id = 'None'
    try:
        user_id = get_page_id(name)
        user_endpoint = 'https://api.vineapp.com/users/profiles/' + str(user_id)
        data = requests.get(user_endpoint).json()["data"]
        if data["verified"]:
            the_id = user_id
    except Exception, e:
        print e, 'Cant get id for Vine name: ' + name
    return the_id