From ffd0ca4ee2bbfa9042d10c85258ef36ccc69462d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BA=B7=E5=AE=87=E4=BD=B3?= Date: Mon, 3 Mar 2025 11:45:14 +0800 Subject: [PATCH] =?UTF-8?q?FIX=20job=5Fid=E7=9B=B4=E6=8E=A5=E8=8E=B7?= =?UTF-8?q?=E5=8F=96extra=5Fdata?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- __init__.py | 6 ++++-- config.yaml | 5 ----- 2 files changed, 4 insertions(+), 7 deletions(-) delete mode 100644 config.yaml diff --git a/__init__.py b/__init__.py index 990c662..f47a4dc 100644 --- a/__init__.py +++ b/__init__.py @@ -353,11 +353,11 @@ class LogToDB: CATEGORY = "不忘科技-自定义节点🚩" - def log2db(self, job_id, log, status, unique_id): - job_id = int(job_id) + def log2db(self, log, status, unique_id): # 获取comfy服务器队列信息 (_, prompt_id, prompt, extra_data, outputs_to_execute) = next( iter(server.PromptServer.instance.prompt_queue.currently_running.values())) + job_id = extra_data["client_id"] engine = create_engine( "mysql+pymysql://root:*k3&5xxG6oqHJM@sh-cdb-1xspb808.sql.tencentcdb.com:28795/comfy", echo=True @@ -556,6 +556,7 @@ class VodToLocalNode: def execute(self, file_id, sub_app_id): # 调用下载逻辑 local_path = self.download_vod(file_id, sub_app_id) + print(f"下载成功: {local_path}") return (local_path,) 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) + print(f"download from url: {media_url}") # 生成一个临时目录路径名并创建该目录 self.create_directory( os.path.join( diff --git a/config.yaml b/config.yaml deleted file mode 100644 index 40360e3..0000000 --- a/config.yaml +++ /dev/null @@ -1,5 +0,0 @@ -secret_id: AKID63qSzJRSFyUowqa6RaVCh182jCUZS771 -secret_key: kP1ThjZDfvynQzVr9QaXHqBJOHzVowqX -region: ap-shanghai -bucket: bwkj-cos-1324682537 -subfolder: test \ No newline at end of file