DownloadFiles.py

Go to the documentation of this file.

from __future__ import print_function
 
import io
import os
import google.auth
from google.oauth2.credentials import Credentials
from googleapiclient.discovery import build
from googleapiclient.errors import HttpError
from googleapiclient.http import MediaIoBaseDownload
from urllib.request import urlretrieve
 
 
def download_file_or_folder(file_id, path):
    """
    Downloads a file or a folder from Google Drive
    
    Args:
        file_id (str): ID of the folder/file to download
        path (str): Local path where file(s) will be downloaded
    Returns : Nothing
    """
    # creds, project_id = google.auth.default()
    # dir_path = os.path.dirname("glider/")
    # base_path = os.path.basename(os.getcwd())
    # full_path = dir_path+"/"+base_path
    full_path = __file__
    rel_path = full_path.split("/")[:-1]
    rel_path = "/".join(rel_path)
    try:
        # create drive api client
        SCOPES = ["https://www.googleapis.com/auth/drive", "https://www.googleapis.com/auth/drive.file", "https://www.googleapis.com/auth/drive.readonly", 
        "https://www.googleapis.com/auth/drive.metadata.readonly", "https://www.googleapis.com/auth/drive.appdata", "https://www.googleapis.com/auth/drive.metadata", 
        "https://www.googleapis.com/auth/drive.photos.readonly"]
        # SCOPES = ['https://www.googleapis.com/auth/spreadsheets', "https://www.googleapis.com/auth/drive.file", "https://www.googleapis.com/auth/drive"]
        creds = Credentials.from_authorized_user_file(rel_path+"/token.json", SCOPES)
        service = build("drive", "v3", credentials=creds)
 
        file = service.files().get(fileId=file_id, fields="name, mimeType").execute()
        file_name = file["name"]
        mime_type = file["mimeType"]
        new_path = os.path.join(path, file_name)
        if mime_type == "application/vnd.google-apps.folder":
            # If it's a folder, create a corresponding local directory and download its contents
            if not os.path.exists(new_path):
                os.makedirs(new_path)
            results = service.files().list(q="'{}' in parents".format(file_id), fields="nextPageToken, files(id, name)").execute()
            items = results.get("files", [])
            for item in items:
                download_file_or_folder(item['id'], new_path)
        else:
            # If it's a file, download it to the local directory
            request = service.files().get_media(fileId=file_id)
            fh = io.BytesIO()
            downloader = MediaIoBaseDownload(fh, request)
            done = False
            while done is False:
                status, done = downloader.next_chunk()
                print("Download %d%%." % int(status.progress() * 100))
            fh.seek(0)
 
            with open(new_path, "wb") as f:
                f.write(fh.read())
                f.close()
 
    except HttpError as error:
        print("An error occurred: {}".format(error))
        raise Exception("Some files can not be downloaded")