如何使用 Python 将文件上传到 Google Drive

Python 2月前 21

本文将介绍如何使用 Python 编程语言将文件上传到 Google Drive。我们将使用 Google Drive API 和 Python 的相关库来实现这一功能。文章中将详细说明如何设置 Google Drive API，编写 Python 代码以上传文件，并提供一些注意事项以确保顺利完成上传过程。

1. 设置 Google Drive API

在开始之前，我们需要进行一些准备工作来设置 Google Drive API。请按照以下步骤进行操作：

创建一个 Google Cloud 项目：在 Google Cloud 控制台上创建一个新的项目。
启用 Google Drive API：在项目的 API 和服务页面中启用 Google Drive API。
创建凭据：在 API 和服务页面的凭据部分创建一个新的 OAuth 2.0 客户端 ID。选择“桌面应用程序”作为应用类型，并记录生成的客户端 ID 和客户端密钥。
下载凭据文件：下载生成的凭据文件（JSON 格式），并将其保存在你的项目文件夹中。

2. 安装必要的库

在 Python 中上传文件到 Google Drive 需要使用 Google Drive API 的 Python 客户端库。使用以下命令安装所需的库：

pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib

3. 编写 Python 代码

现在我们可以开始编写 Python 代码来实现文件上传功能。以下是一个示例代码：

import os
from googleapiclient.discovery import build
from google.oauth2.credentials import Credentials
from google_auth_oauthlib.flow import InstalledAppFlow

# 设置凭据文件路径
credentials_path = 'path/to/credentials.json'

# 设置要上传的文件路径
file_path = 'path/to/file.txt'

# 设置要上传到的 Google Drive 文件夹 ID
folder_id = 'your_folder_id'

# 定义上传文件的函数
def upload_file(credentials_path, file_path, folder_id):
    # 加载凭据
    creds = Credentials.from_authorized_user_file(credentials_path)

    # 构建 Google Drive 服务
    service = build('drive', 'v3', credentials=creds)

    # 创建文件元数据
    file_metadata = {
        'name': os.path.basename(file_path),
        'parents': [folder_id]
    }

    # 上传文件
    media = MediaFileUpload(file_path)
    file = service.files().create(
        body=file_metadata,
        media_body=media,
        fields='id'
    ).execute()

    print('文件已上传，文件 ID：', file.get('id'))

# 调用上传文件函数
upload_file(credentials_path, file_path, folder_id)

请确保将 credentials_path、file_path 和 folder_id 替换为你自己的凭据文件路径、要上传的文件路径和目标文件夹的 ID。