def client(self):
if not self._client.GetClientLoginToken():
try:
self._client.ClientLogin(
self.credential.username,
self.credential.password,
self._client.source,
)
except gdata.service.CaptchaRequired:
sys.stdout.write('Please visit ' + self._client.captcha_url)
answer = self.input_getter('Answer to the challenge? ')
self._client.email = self.credential.username
self._client.password = self.credential.password
self._client.ClientLogin(
self.credential.username,
self.credential.password,
self._client.source,
captcha_token=self._client.captcha_token,
captcha_response=answer,
)
except gdata.service.BadAuthentication:
raise errors.InitError('Users credential were unrecognized')
except gdata.service.Error:
raise errors.InitError('Login Error')
return self._client
def _read(self):
from gdata.docs.service import DocumentQuery
title_query = DocumentQuery(categories=[self.collection])
title_query['title'] = self._get_doc_title()
title_query['title-exact'] = 'true'
docs = self.client.QueryDocumentListFeed(title_query.ToUri())
if not docs.entry:
if self.can_create:
docs_entry = None
keyring_dict = {}
else:
raise errors.InitError(
'%s not found in %s and create not permitted'
% (self._get_doc_title(), self.collection)
)
else:
docs_entry = docs.entry[0]
file_contents = ''
try:
url = docs_entry.content.src
url += '&exportFormat=txt'
server_response = self.client.request('GET', url)
if server_response.status != 200:
raise errors.InitError(
'Could not read existing Google Docs keyring'
)
file_contents = server_response.read()
if file_contents.startswith(codecs.BOM_UTF8):
file_contents = file_contents[len(codecs.BOM_UTF8) :]
keyring_dict = pickle.loads(
base64.urlsafe_b64decode(file_contents.decode('string-escape'))
)
except pickle.UnpicklingError as ex:
raise errors.InitError(
'Could not unpickle existing Google Docs keyring', ex
)
except TypeError as ex:
raise errors.InitError(
'Could not decode existing Google Docs keyring', ex
)
return docs_entry, keyring_dict