mirror of
https://github.com/BlueSkyXN/WorkerJS_CloudFlare_ImageBed.git
synced 2024-11-16 03:32:26 +08:00
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
|
import sys
|
|||
|
|
import os
|
|||
|
|
import boto3
|
|||
|
|
from botocore.client import Config
|
|||
|
|
from botocore.exceptions import ClientError
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
# Filebase的S3兼容的API端点和凭据
|
|||
|
|
endpoint_url = 'https://s3.filebase.com'
|
|||
|
|
access_key = ''
|
|||
|
|
secret_key = ''
|
|||
|
|
|
|||
|
|
# 获取命令行参数
|
|||
|
|
if len(sys.argv) != 2:
|
|||
|
|
print("Usage: python upload_to_filebase.py <local_file_path>")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
local_file_path = sys.argv[1]
|
|||
|
|
# Filebase的S3兼容存储桶名称
|
|||
|
|
bucket_name = ''
|
|||
|
|
|
|||
|
|
# 使用 os.path.basename() 只提取文件名
|
|||
|
|
file_name, file_extension = os.path.splitext(os.path.basename(local_file_path))
|
|||
|
|
|
|||
|
|
# 生成当前时间戳，格式为：年-月-日_时-分-秒
|
|||
|
|
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
|||
|
|
|
|||
|
|
# 使用文件名 + 时间戳 作为 S3 文件名
|
|||
|
|
s3_file_key = f"{file_name}_{timestamp}{file_extension}"
|
|||
|
|
|
|||
|
|
# 创建S3客户端，增加超时设置
|
|||
|
|
try:
|
|||
|
|
print("Creating S3 client...")
|
|||
|
|
s3 = boto3.client(
|
|||
|
|
's3',
|
|||
|
|
aws_access_key_id=access_key,
|
|||
|
|
aws_secret_access_key=secret_key,
|
|||
|
|
endpoint_url=endpoint_url,
|
|||
|
|
config=Config(signature_version='s3v4', connect_timeout=10, read_timeout=10)
|
|||
|
|
)
|
|||
|
|
print("S3 client created successfully.")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Error creating S3 client: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 上传文件
|
|||
|
|
def upload_file_to_s3():
|
|||
|
|
try:
|
|||
|
|
print(f"Starting to upload '{local_file_path}' to bucket '{bucket_name}'...")
|
|||
|
|
s3.upload_file(local_file_path, bucket_name, s3_file_key)
|
|||
|
|
print(f"File '{local_file_path}' successfully uploaded to '{bucket_name}/{s3_file_key}'")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Error uploading file: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 获取文件的元数据，包括CID
|
|||
|
|
def get_file_metadata(bucket_name, s3_file_key):
|
|||
|
|
try:
|
|||
|
|
print(f"Fetching metadata for file '{s3_file_key}' in bucket '{bucket_name}'...")
|
|||
|
|
response = s3.head_object(Bucket=bucket_name, Key=s3_file_key)
|
|||
|
|
metadata = response.get('Metadata', {})
|
|||
|
|
|
|||
|
|
if metadata:
|
|||
|
|
print("File metadata:", metadata)
|
|||
|
|
cid = metadata.get('cid', None) # 假设 CID 被存储为元数据的一部分
|
|||
|
|
if cid:
|
|||
|
|
print(f"CID: {cid}")
|
|||
|
|
else:
|
|||
|
|
print("CID not found in metadata.")
|
|||
|
|
else:
|
|||
|
|
print("No custom metadata found for the file.")
|
|||
|
|
except ClientError as e:
|
|||
|
|
print(f"Error fetching metadata: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 测试连接是否正常
|
|||
|
|
def test_s3_connection():
|
|||
|
|
try:
|
|||
|
|
print("Testing S3 connection by listing buckets...")
|
|||
|
|
s3.list_buckets()
|
|||
|
|
print("S3 connection successful.")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Error in connecting to S3: {e}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# 执行上传文件和获取CID的流程
|
|||
|
|
test_s3_connection() # 首先测试连接
|
|||
|
|
upload_file_to_s3() # 上传文件
|
|||
|
|
get_file_metadata(bucket_name, s3_file_key) # 获取元数据
|