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:

Search for Files and Team Drives
Files: list