[Python] Google Driveに任意のファイルをアップロードする方法

pip install google-auth

pip install google-api-python-client

pip install google-auth-oauthlib

import os
import google.auth
from google.auth.transport.requests import Request
from google_auth_oauthlib.flow import InstalledAppFlow
# スコープの設定
SCOPES = ['https://www.googleapis.com/auth/drive.file']
def get_credentials():
    creds = None
    # トークンファイルが存在する場合は、そこから認証情報を取得
    if os.path.exists('token.json'):
        creds = google.auth.load_credentials_from_file('token.json')[0]
    # 認証情報が無い場合は新たに取得
    if not creds or not creds.valid:
        if creds and creds.expired and creds.refresh_token:
            creds.refresh(Request())
        else:
            flow = InstalledAppFlow.from_client_secrets_file('credentials.json', SCOPES)
            creds = flow.run_local_server(port=0)
        # 認証情報をファイルに保存
        with open('token.json', 'w') as token:
            token.write(creds.to_json())
    return creds
# 認証情報を取得
credentials = get_credentials()

from googleapiclient.discovery import build
# Google Drive APIのサービスを作成
drive_service = build('drive', 'v3', credentials=credentials)
# アップロードするファイルのメタデータを設定
file_metadata = {
    'name': 'example.txt',  # アップロードするファイル名
    'mimeType': 'text/plain'  # MIMEタイプ
}

# 例: 画像ファイルのMIMEタイプを指定
file_metadata = {
    'name': 'image.jpg',
    'mimeType': 'image/jpeg'
}

from googleapiclient.http import MediaFileUpload
# アップロードするファイルのパス
file_path = 'path/to/your/example.txt'
# ファイルの内容を指定
media = MediaFileUpload(file_path, mimetype='text/plain')
# ファイルをアップロード
file = drive_service.files().create(body=file_metadata, media_body=media, fields='id').execute()
print('ファイルがアップロードされました。ファイルID:', file.get('id'))

def progress_callback(current, total):
    print(f'アップロード中: {current / total * 100:.2f}%')
# ファイルをアップロード(進行状況を表示)
media = MediaFileUpload(file_path, mimetype='text/plain', chunksize=1024*1024, resumable=True)
request = drive_service.files().create(body=file_metadata, media_body=media, fields='id')
# アップロードの進行状況を確認
response = None
while response is None:
    status, response = request.next_chunk()
    if status:
        print(f'アップロード中: {int(status.progress() * 100)}%')
print('ファイルがアップロードされました。ファイルID:', response.get('id'))

https://drive.google.com/drive/folders/フォルダID

# アップロードするファイルのメタデータを設定
folder_id = 'YOUR_FOLDER_ID'  # 取得したフォルダIDをここに入力
file_metadata = {
    'name': 'example_in_folder.txt',  # アップロードするファイル名
    'parents': [folder_id],  # 親フォルダのIDを指定
    'mimeType': 'text/plain'  # MIMEタイプ
}
# アップロードするファイルのパス
file_path = 'path/to/your/example_in_folder.txt'
# ファイルの内容を指定
media = MediaFileUpload(file_path, mimetype='text/plain')
# フォルダにファイルをアップロード
file = drive_service.files().create(body=file_metadata, media_body=media, fields='id').execute()
print('フォルダにファイルがアップロードされました。ファイルID:', file.get('id'))

# アップロードしたファイルのID
file_id = 'YOUR_FILE_ID'  # アップロードしたファイルのIDをここに入力
# 共有リンクを取得するための権限を設定
permission = {
    'type': 'anyone',  # 誰でもアクセス可能
    'role': 'reader'   # 読み取り専用
}
# 権限を追加
drive_service.permissions().create(fileId=file_id, body=permission).execute()
# 共有リンクの生成
shared_link = f'https://drive.google.com/file/d/{file_id}/view?usp=sharing'
print('共有リンク:', shared_link)

# 特定のユーザーのメールアドレス
user_email = 'user@example.com'  # 共有したいユーザーのメールアドレスをここに入力
# ユーザーに編集権限を付与
permission = {
    'type': 'user',
    'role': 'writer',
    'emailAddress': user_email
}
# 権限を追加
drive_service.permissions().create(fileId=file_id, body=permission).execute()
print(f'{user_email}に編集権限を付与しました。')

# 特定のユーザーのメールアドレス
user_email = 'user@example.com'  # 共有したいユーザーのメールアドレスをここに入力
# ユーザーに読み取り専用権限を付与
permission = {
    'type': 'user',
    'role': 'reader',
    'emailAddress': user_email
}
# 権限を追加
drive_service.permissions().create(fileId=file_id, body=permission).execute()
print(f'{user_email}に読み取り専用権限を付与しました。')

import os
# アップロードするフォルダのパス
folder_path = 'path/to/your/folder'
folder_id = 'YOUR_FOLDER_ID'  # アップロード先のフォルダID
# フォルダ内のすべてのテキストファイルをアップロード
for filename in os.listdir(folder_path):
    if filename.endswith('.txt'):
        file_metadata = {
            'name': filename,
            'parents': [folder_id],
            'mimeType': 'text/plain'
        }
        file_path = os.path.join(folder_path, filename)
        media = MediaFileUpload(file_path, mimetype='text/plain')
        file = drive_service.files().create(body=file_metadata, media_body=media, fields='id').execute()
        print(f'ファイル {filename} がアップロードされました。ファイルID: {file.get("id")}')

# 更新するファイルのID
file_id = 'YOUR_FILE_ID'  # 更新したいファイルのIDをここに入力
new_file_path = 'path/to/your/new_example.txt'  # 新しいファイルのパス
# 新しいファイルのメタデータを設定
file_metadata = {
    'name': 'updated_example.txt',  # 新しいファイル名
}
# 新しいファイルの内容を指定
media = MediaFileUpload(new_file_path, mimetype='text/plain')
# ファイルを更新
file = drive_service.files().update(fileId=file_id, body=file_metadata, media_body=media).execute()
print('ファイルが更新されました。ファイルID:', file.get('id'))

# フォルダ内のファイルを一覧取得
query = f"'{folder_id}' in parents"
results = drive_service.files().list(q=query, fields="files(id, name)").execute()
items = results.get('files', [])
if not items:
    print('フォルダ内にファイルはありません。')
else:
    print('フォルダ内のファイル:')
    for item in items:
        print(f'ファイル名: {item["name"]}, ファイルID: {item["id"]}')

# 削除するファイルのID
file_id = 'YOUR_FILE_ID'  # 削除したいファイルのIDをここに入力
# ファイルを削除
drive_service.files().delete(fileId=file_id).execute()
print('ファイルが削除されました。')

import io
from googleapiclient.http import MediaIoBaseDownload
# ダウンロードするファイルのID
file_id = 'YOUR_FILE_ID'  # ダウンロードしたいファイルのIDをここに入力
# ファイルの情報を取得
request = drive_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(f'ダウンロード中: {int(status.progress() * 100)}%')
# ダウンロードしたファイルを保存
with open('downloaded_file.txt', 'wb') as f:
    f.write(fh.getbuffer())
print('ファイルがダウンロードされました。')