下载批量文件drom Google Drive与调整的父文件夹

import os
import io
from google.oauth2 import service_account
from googleapiclient.discovery import build
from googleapiclient.http import MediaIoBaseDownload

# 设置凭证文件路径
SCOPES = ['https://www.googleapis.com/auth/drive']
SERVICE_ACCOUNT_FILE = 'path/to/your/service_account.json'

# 认证并构建服务
creds = service_account.Credentials.from_service_account_file(
    SERVICE_ACCOUNT_FILE, scopes=SCOPES)
service = build('drive', 'v3', credentials=creds)

def download_file(file_id, file_name, parent_folder_id):
    request = service.files().get_media(fileId=file_id)
    fh = io.FileIO(file_name, 'wb')
    downloader = MediaIoBaseDownload(fh, request)
    done = False
    while done is False:
        status, done = downloader.next_chunk()
        print(f"Download {int(status.progress() * 100)}%.")

    # 更新文件的父文件夹
    file = service.files().update(
        fileId=file_id,
        addParents=parent_folder_id,
        removeParents='root'  # 假设文件原本在根目录
    ).execute()
    print(f"Updated file {file_name} with new parent folder.")

def list_files_in_folder(folder_id):
    results = service.files().list(
        q=f"'{folder_id}' in parents",
        fields="nextPageToken, files(id, name)"
    ).execute()
    items = results.get('files', [])
    return items

# 示例用法
folder_id = 'your-target-folder-id'
new_parent_folder_id = 'new-parent-folder-id'

files = list_files_in_folder(folder_id)
for file in files:
    download_file(file['id'], file['name'], new_parent_folder_id)