def download(self):
# Get a file
drive_service = TransportVehicle.build_drive_service()
try:
file = drive_service.files().get(fileId=self.file_id).execute()
except AccessTokenCredentialsError as a:
print "Download try:", a
creds = TransportVehicle.manual_authorization()
drive_service = TransportVehicle.build_drive_service(credentials=creds)
file = drive_service.files().get(fileId=self.file_id).execute()
url = file.get('downloadUrl')
if url:
resp, content = drive_service._http.request(url)
if resp.status == 200:
f = open(self.given_name,'w+')
f.write(content)
f.close()
def upload(self):
# Path to the file to upload
drive_service = TransportVehicle.build_drive_service()
media_body = MediaFileUpload(self.file_path, mimetype='text/plain', resumable=True)
body = {
'title': self.given_name,
'description': 'A test document',
'mimeType': 'text/plain'
}
try:
file = drive_service.files().insert(body=body, media_body=media_body).execute()
except Exception as a:
print "Upload try:", a
creds = TransportVehicle.manual_authorization()
drive_service = TransportVehicle.build_drive_service(credentials=creds)
file = drive_service.files().insert(body=body, media_body=media_body).execute()
return file['id']
