google_utils.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
Created on 2013-11-25

@author: Martin H. Bramwell
'''
from persistence import store, close_store, make_store, reopen_store
from gspread import login, authorize

store = None
if not store:
    store = make_store()

expr = "Missing or misformed configuration file(s)."
msg = "Either a file named 'creds_oa.py' or a file named 'creds_up.py' must be"
msg = " present in order to authenticate your connection to Google."

credentials = {}
try:
    credentials = store['credentials']
    print "Will connect with shelved credentials"
except KeyError:
    try:
        from creds_oa import credentials
        print "Will try with OAuth credentials"
    except ImportError:
        try:
            from creds_up import credentials
            print "Will try with UID/PWD credentials"
        except ImportError:
            print "Configuration error: {} {}".format(expr, msg)
            exit(-1)

    store['credentials'] = credentials
    reopen_store()
    print "The creds_*.py file(s) may now be deleted."

OAUTH = "oauth"
UID_PWD = "uidpwd"


def getPumpCredentials(wkbk):

    shtCreds = wkbk.worksheet("Creds")
    lstlstCreds = shtCreds.get_all_values()

    creds = {t[0]: t[1] for t in lstlstCreds}

    return creds


class Google(object):

    connection = None

    def __init__(self):

        global connection

        print "Getting connection."
        if credentials['cred_type'] == OAUTH:
            connection = authorize(
                credentials['access_token'], credentials['key_ring'])
            print "Will authorize."
        elif credentials['cred_type'] == UID_PWD:
            connection = login(credentials['uid'], credentials['pwd'])
            print "Will log in."
        else:
            print "Configuration error: {} {}".format(expr, msg)
            exit(-1)

    def connect(self):
        return connection