Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
Download file from shared Google Drive link.
from __future__ import print_function
import os
import requests
URL = "https://docs.google.com/uc?export=download"
ID = "" # the file ID appeared in the shared link
DATAFILE = "" # the name of the file to be downloaded
if __name__ == '__main__':
file_dir = os.path.dirname(os.path.realpath(__file__))
if not os.path.exists(os.path.join(file_dir, DATAFILE)):
print("dowloading file {} ...".format(DATAFILE))
get_file(URL, ID, os.path.join(file_dir, DATAFILE))
print("done, saved to {}".format(file_dir))
else:
print("file {} already existed".format(DATAFILE))
def get_file(url, id, destination):
session = requests.Session()
response = session.get(url, params = { 'id' : id }, stream = True)
token = get_confirm_token(response)
if token:
params = { 'id' : id, 'confirm' : token }
response = session.get(url, params = params, stream = True)
save_response_content(response, destination)
def get_confirm_token(response):
for key, value in response.cookies.items():
if key.startswith('download_warning'):
return value
return None
def save_response_content(response, destination):
CHUNK_SIZE = 32768
if not os.path.exists(os.path.dirname(destination)):
os.makedirs(os.path.dirname(destination))
with open(destination, "wb") as f:
for chunk in response.iter_content(CHUNK_SIZE):
if chunk: # filter out keep-alive new chunks
f.write(chunk)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment