bowong
/
ComfyUI-CustomNode
6
0
Fork 0
Code Pull Requests Releases Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes/s3.py
View File

@ -102,7 +102,6 @@ class S3UploadURL:
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"s3_bucket": ("STRING", {"default": "bw-comfyui-output"}),
"path": ("STRING", {"multiline": True}), "path": ("STRING", {"multiline": True}),
"subfolder": ("STRING", {"default": "test"}), "subfolder": ("STRING", {"default": "test"}),
} }
@ -114,7 +113,8 @@ class S3UploadURL:
FUNCTION = "upload" FUNCTION = "upload"
CATEGORY = "不忘科技-自定义节点🚩/S3" 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}") loguru.logger.info(f"S3 UPLOAD {path} to {s3_bucket}/{subfolder}")
try: try:
if "aws_key_id" in list(os.environ.keys()): if "aws_key_id" in list(os.environ.keys()):
@ -138,7 +138,7 @@ class S3UploadURL:
client.upload_file(path, s3_bucket, dest_key) client.upload_file(path, s3_bucket, dest_key)
except Exception as e: except Exception as e:
raise Exception(f"S3上传失败！ bucket {s3_bucket}; local_path {path}; subfolder {subfolder}") 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,) return (url,)