def _set_account_info(self):
with session_scope() as db_session:
account = db_session.query(Account).get(self.account_id)
#self.full_name = account.full_name
self.full_name = 'TEST'
self.email_address = account.email_address
self.provider = account.provider
self.auth_type = AUTH_TYPES.get(account.provider)
if self.auth_type == 'OAuth':
# Refresh OAuth token if need be
account = verify_imap_account(db_session, account)
self.o_access_token = account.o_access_token
else:
assert self.auth_type == 'Password'
self.password = account.password
def _get_google_client(self):
"""Return the Google API client."""
# TODO(emfree) figure out a better strategy for refreshing OAuth
# credentials as needed
with session_scope() as db_session:
try:
account = db_session.query(ImapAccount).get(self.account_id)
account = verify_imap_account(db_session, account)
two_legged_oauth_token = gdata.gauth.OAuth2Token(
client_id=GOOGLE_OAUTH_CLIENT_ID,
client_secret=GOOGLE_OAUTH_CLIENT_SECRET,
scope=OAUTH_SCOPE,
user_agent=SOURCE_APP_NAME,
access_token=account.o_access_token,
refresh_token=account.o_refresh_token,
)
google_client = gdata.contacts.client.ContactsClient(source=SOURCE_APP_NAME)
google_client.auth_token = two_legged_oauth_token
return google_client
except gdata.client.BadAuthentication:
self.log.error("Invalid user credentials given")
return None
