使用Python通过Google Drive API下载文件时遇到的问题

Question

因此，我尝试使用Python和Google Drive API将Google Drive文件夹中的所有文件下载到计算机的文件夹中。但是由于某些原因，下载的文件似乎已损坏，并且大小仅为零字节。我尝试了两种不同的解决方案，但它们都不起作用。有人可以告诉我我在做什么错吗？

from __future__ import print_function
import pickle
from pathlib import Path
import requests
from googleapiclient.discovery import build
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
from googleapiclient.http import MediaIoBaseDownload
import io, os

def dl_file(g_drive, folderID):
    folder_path = Path("<folder path>")
    files_obj = g_drive.files()
    files_list = files_obj.list(pageSize=100,
                             fields="nextPageToken, files(name, id, webContentLink, parents)").execute()
    items = files_list['files']
    if not items:
        print('No files found.')
    else:
        for item in items:
            if item['parents'] == [folderID]:
                file_name = item['name']

                #Method 1
                fh = io.BytesIO()
                downloader = MediaIoBaseDownload(fh, files_obj.get_media(fileId=item['id']))
                done = False
                while done is False:
                    status, done = downloader.next_chunk()
                    print("Download %d%%." % int(status.progress() * 100))
                with io.open(f"{folder_path.as_posix()}/{file_name}", 'wb') as f:
                    f.seek(0)
                    f.write(fh.read())

                #Method 2
                url = item['webContentLink']
                r = requests.get(url, stream=True)
                if r.ok:
                    print(f"Saving to '{folder_path.name}/'")
                    with open(f"{folder_path.as_posix()}/{file_name}", 'wb') as f:
                        for chunk in r.iter_content(chunk_size=1024 * 8):
                            if chunk:
                                f.write(chunk)
                                f.flush()
                                os.fsync(f.fileno())
                else:  # HTTP status code 4XX/5XX
                    print("Download failed: status code {}\n{}".format(r.status_code, r.text))

g_drive = build('drive', 'v3', credentials=creds)
dl_file(g_drive, "<folder_ID>")

Answer 1

我相信您的目标如下。

• 您要在Google云端硬盘的特定文件夹中下载不是Google文档的文件。
• 您想使用python实现。
• 您已经能够使用Drive API下载文件。
  • 您尝试了#Method 1和#Method 2的2种方法。
  • #Method 1使用googleapis。
  • [#Method 2使用requests。

为此，这个答案如何？

修改点：

• 在此修改中，我想使用googleapis修改#Method 1。
• 从特定文件夹下载文件时，我建议使用“文件：列表”方法直接从特定文件夹中检索Google Docs以外的文件列表。在这种情况下，将使用搜索查询。

当这些内容反映到您的脚本时，它将变成如下。

修改的脚本：

请如下修改dl_file。

def dl_file(g_drive, folderID):
    folder_path = Path("<folder path>")
    files_obj = g_drive.files()
    files_list = files_obj.list(pageSize=100,
                                q="'" + folderID + "' in parents and not mimeType contains 'application/vnd.google-apps'",
                                fields="nextPageToken, files(name, id, webContentLink, parents)").execute()
    items = files_list['files']
    if not items:
        print('No files found.')
    else:
        for item in items:
            file_name = item['name']
            request = g_drive.files().get_media(fileId=item['id'])
            fh = io.FileIO(folder_path.as_posix() + '/' + file_name, mode='wb')
            downloader = MediaIoBaseDownload(fh, request)
            print('Download of ' + file_name)
            done = False
            while done is False:
                status, done = downloader.next_chunk()
                print('Download %d%%.' % int(status.progress() * 100))

• 运行上述脚本时，下载的文件将创建到<folder path>的目录中。

参考：

• Files: list
• Search for files and folders