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