FIX job_id直接获取extra_data
This commit is contained in:
parent
0435df9a84
commit
ffd0ca4ee2
|
|
@ -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(
|
||||
|
|
|
|||
|
|
@ -1,5 +0,0 @@
|
|||
secret_id: AKID63qSzJRSFyUowqa6RaVCh182jCUZS771
|
||||
secret_key: kP1ThjZDfvynQzVr9QaXHqBJOHzVowqX
|
||||
region: ap-shanghai
|
||||
bucket: bwkj-cos-1324682537
|
||||
subfolder: test
|
||||
Loading…
Reference in New Issue