def get_credentials(self):
if self.storage is None:
self.storage = Storage(utils.GDRIVE_CREDS)
self.credentials = self.storage.get()
return self.credentials
class GoogleDrive(object):
"""Wrapper to the google drive api"""
def __init__(self):
self.storage = None
self.credentials = None
self.http = None
self.service = None
self.get_credentials()
def get_credentials(self):
if self.storage is None:
self.storage = Storage(utils.GDRIVE_CREDS)
self.credentials = self.storage.get()
return self.credentials
def test_auth(self, file_id="root"):
try:
root = self.get_service().files().get(fileId=file_id).execute()
except:
log.exception("Couldn't connect to Google drive")
return False
else:
return True
def upload_file(self, filepath, filename, parent="root"):
media_body = MediaFileUpload(filepath, resumable=True, mimetype="", chunksize=-1)
body = {
'title': filename,
'parents': [{'id': parent}],
'mimeType': ""
}
return self.get_service().files().insert(body=body, media_body=media_body).execute()
def check_file_exists(self, filename, parent="root"):
children = self.get_service().children().list(folderId=parent, q="title = '%s'" % filename.replace("'", "\\'")).execute()
items = children.get('items', [])
original = None
for child in items:
child_file = self.get_service().files().get(fileId=child["id"]).execute()
if original is None:
original = child_file
else:
o = time.strptime(original["createdDate"], "%Y-%m-%dT%H:%M:%S.%fZ")
t = time.strptime(child_file["createdDate"], "%Y-%m-%dT%H:%M:%S.%fZ")
if t < o:
original = child_file
if original is None:
return False
else:
return original
def need_to_upload(self, filename, folder_id, size_bytes):
if size_bytes <= 0:
return False
myfile = self.check_file_exists(filename, parent=folder_id)
if myfile and int(myfile["fileSize"]) == size_bytes:
return False
elif myfile:
log.debug("Filesize incorrect deleting %s", filename)
self.get_service().files().delete(fileId=myfile["id"])
return True
else:
return True
def get_folder_byname(self, foldername, parent="root", createnotfound=False):
children = self.get_service().children().list(folderId=parent,
q="title = '%s' and mimeType = 'application/vnd.google-apps.folder'" % foldername.replace("'", "\\'")).execute()
items = children.get('items', [])
original = None
for child in items:
child_file = self.get_service().files().get(fileId=child["id"]).execute()
if original is None:
original = child_file
else:
o = time.strptime(original["createdDate"], "%Y-%m-%dT%H:%M:%S.%fZ")
t = time.strptime(child_file["createdDate"], "%Y-%m-%dT%H:%M:%S.%fZ")
if t < o:
original = child_file
if original is None:
if createnotfound:
log.debug("No items by the name of %s found. Creating", foldername)
body = {
'title': foldername,
'mimeType':'application/vnd.google-apps.folder'
}
if parent != "root":
body['parents'] = [{'id':parent}]
return self.get_service().files().insert(body=body).execute()
else:
log.debug("No items by the name of %s found", foldername)
return False
else:
return original
@property
def token_expired(self):
if self.credentials is None:
return True
return self.credentials.access_token_expired
def auth(self, promptForAuth, launchBrowser):
"""Authorize an http client, asking the user if required.
"""
if self.token_expired:
if self.can_token_refresh:
log.debug("Trying to refresh auth token")
try:
self.credentials.refresh(self.http)
except:
log.debug("Auth token refresh failed. New auth needed")
self.credentials = None
if promptForAuth:
self.get_service(True)
else:
raise
elif promptForAuth:
flow = flow_from_clientsecrets(utils.GDRIVE_SECRETS, scope='https://www.googleapis.com/auth/drive')
flow.redirect_uri = "urn:ietf:wg:oauth:2.0:oob"
flow.params['access_type'] = 'offline'
flow.params['approval_prompt'] = 'force'
url = flow.step1_get_authorize_url()
log.info(url)
if launchBrowser:
webbrowser.open(url)
code = raw_input("Enter token ")
self.credentials = flow.step2_exchange(code)
else:
raise AccessTokenRefreshError("Access token unable to refresh please re-authenticate")
log.info("Storing Google credentials")
self.storage.put(self.credentials)
self.credentials.authorize(self.http)
@property
def can_token_refresh(self):
return self.credentials is not None and self.credentials.refresh_token is not None
def get_service(self, promptForAuth=False, launchBrowser=False):
if self.service is None or self.token_expired:
if self.http is None:
self.http = httplib2.Http()
self.auth(promptForAuth, launchBrowser)
self.service = build('drive', 'v2', http=self.http)
return self.service
