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

FIX job_id直接获取extra_data

This commit is contained in:
康宇佳 2025-03-03 11:45:14 +08:00
parent 0435df9a84
commit ffd0ca4ee2
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
__init__.py
View File

@ -353,11 +353,11 @@ class LogToDB:
CATEGORY = "不忘科技-自定义节点🚩" CATEGORY = "不忘科技-自定义节点🚩"
def log2db(self, job_id, log, status, unique_id): def log2db(self, log, status, unique_id):
job_id = int(job_id)
# 获取comfy服务器队列信息 # 获取comfy服务器队列信息
(_, prompt_id, prompt, extra_data, outputs_to_execute) = next( (_, prompt_id, prompt, extra_data, outputs_to_execute) = next(
iter(server.PromptServer.instance.prompt_queue.currently_running.values())) iter(server.PromptServer.instance.prompt_queue.currently_running.values()))
job_id = extra_data["client_id"]
engine = create_engine( engine = create_engine(
"mysql+pymysql://root:*k3&5xxG6oqHJM@sh-cdb-1xspb808.sql.tencentcdb.com:28795/comfy", "mysql+pymysql://root:*k3&5xxG6oqHJM@sh-cdb-1xspb808.sql.tencentcdb.com:28795/comfy",
echo=True echo=True
@ -556,6 +556,7 @@ class VodToLocalNode:
def execute(self, file_id, sub_app_id): def execute(self, file_id, sub_app_id):
# 调用下载逻辑 # 调用下载逻辑
local_path = self.download_vod(file_id, sub_app_id) local_path = self.download_vod(file_id, sub_app_id)
print(f"下载成功: {local_path}")
return (local_path,) return (local_path,)
def _get_download_url(self, file_id, sub_app_id): def _get_download_url(self, file_id, sub_app_id):
@ -593,6 +594,7 @@ class VodToLocalNode:
返回本地文件路径 返回本地文件路径
""" """
media_url = self._get_download_url(file_id=file_id, sub_app_id=sub_app_id) media_url = self._get_download_url(file_id=file_id, sub_app_id=sub_app_id)
print(f"download from url: {media_url}")
# 生成一个临时目录路径名并创建该目录 # 生成一个临时目录路径名并创建该目录
self.create_directory( self.create_directory(
os.path.join( os.path.join(

5
config.yaml
View File

@ -1,5 +0,0 @@
secret_id: AKID63qSzJRSFyUowqa6RaVCh182jCUZS771
secret_key: kP1ThjZDfvynQzVr9QaXHqBJOHzVowqX
region: ap-shanghai
bucket: bwkj-cos-1324682537
subfolder: test