This commit is contained in:
Keika Hatanaka
2025-02-20 14:37:56 +09:00
commit 9b85df8b40
2 changed files with 76 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
.env
Executable
+75
View File
@@ -0,0 +1,75 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import argparse
from dotenv import load_dotenv
import boto3
import mimetypes
import uuid
# .envファイルから環境変数を読み込む
load_dotenv()
# .envファイルからCloudflare R2のエンドポイントを読み込む
ENDPOINT = os.getenv('R2_ENDPOINT')
def upload_to_r2(image_path):
"""
画像をCloudflare R2にアップロードする。
Args:
image_path (str): アップロードする画像のパス。
Returns:
str: アップロードされた画像のURL。
int: アップロードに失敗した場合のエラーコード。
"""
try:
# R2のクライアントを作成
s3 = boto3.client(
's3',
endpoint_url=ENDPOINT,
aws_access_key_id=os.getenv('R2_ACCESS_KEY_ID'),
aws_secret_access_key=os.getenv('R2_SECRET_ACCESS_KEY'),
config=boto3.session.Config(signature_version='s3v4')
)
# ランダムなファイル名を生成(拡張子は元のまま)
random_filename = str(uuid.uuid4()) + os.path.splitext(image_path)[1]
# 画像のContentTypeをファイル名から推測
content_type, _ = mimetypes.guess_type(image_path)
# 画像をアップロード
with open(image_path, 'rb') as f:
bucket_name = os.getenv('R2_BUCKET_NAME')
s3.put_object(Bucket=bucket_name, Key=random_filename, Body=f, ContentType=content_type)
# アップロードされた画像のURLを返す
return f"{os.getenv('R2_CUSTOM_URL')}{random_filename}"
except Exception as e:
print(f"Error uploading image: {e}")
return None
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Upload an image to Cloudflare R2.')
parser.add_argument('--image-path', required=True, help='Path to the image file.')
args = parser.parse_args()
image_path = args.image_path
# 画像が存在するか確認
if not os.path.exists(image_path):
print(f"Error: Image not found at {image_path}")
exit(1)
# R2に画像をアップロード
image_url = upload_to_r2(image_path)
if image_url:
print(f"{image_url}")
else:
print("Image upload failed.")