Code to download files from google drive to colab

Code:

def download_from_google_drive(file_name_prefix):
# 1. Authenticate and create the PyDrive client.
auth.authenticate_user()
gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.get_application_default()
drive = GoogleDrive(gauth)

# choose a local (colab) directory to store the data.
local_download_path = os.path.expanduser('~/data')
try:
os.makedirs(local_download_path)
except: pass

# 2. Auto-iterate using the query syntax
#    https://developers.google.com/drive/v2/web/search-parameters file_list = drive.ListFile(
{'q': "title contains '%s'" % (file_name_prefix) }).GetList()

files_dict = {}
for f in file_list:
# 3. Create & download by id.
print('title: %s, id: %s' % (f['title'], f['id']))
fname = os.path.join(local_download_path, f['title'])
if not os.path.exists(fname):
print('downloading to {}'.format(fname))
f_ = drive.CreateFile({'id': f['id']})
f_.GetContentFile(fname)
print('Download Completed!')
files_dict[ f['title'] ] = fname

# with open(fname, 'r') as f:
#   print(f.read())
return files_dict, local_download_path

References: