FIX 修复上传到CDN接口

This commit is contained in:
kyj@bowong.ai 2025-06-13 12:50:18 +08:00
parent cc38a14e14
commit 5a220bc39a
1 changed files with 3 additions and 3 deletions

View File

@ -102,7 +102,6 @@ class S3UploadURL:
def INPUT_TYPES(s):
return {
"required": {
"s3_bucket": ("STRING", {"default": "bw-comfyui-output"}),
"path": ("STRING", {"multiline": True}),
"subfolder": ("STRING", {"default": "test"}),
}
@ -114,7 +113,8 @@ class S3UploadURL:
FUNCTION = "upload"
CATEGORY = "不忘科技-自定义节点🚩/S3"
def upload(self, s3_bucket, path, subfolder):
def upload(self, path, subfolder):
s3_bucket = "modal-media-cache"
loguru.logger.info(f"S3 UPLOAD {path} to {s3_bucket}/{subfolder}")
try:
if "aws_key_id" in list(os.environ.keys()):
@ -138,7 +138,7 @@ class S3UploadURL:
client.upload_file(path, s3_bucket, dest_key)
except Exception as e:
raise Exception(f"S3上传失败 bucket {s3_bucket}; local_path {path}; subfolder {subfolder}")
url = f"https://{s3_bucket}.s3.dualstack.ap-northeast-2.amazonaws.com/{dest_key}"
url = f"https://cdn.roasmax.cn/{dest_key}"
return (url,)