From c25e575ee2f2c6489f5e74ee78c2be7b1a686f83 Mon Sep 17 00:00:00 2001
From: dongchangxi <458593490@qq.com>
Date: Tue, 17 Jun 2025 22:21:05 +0800
Subject: [PATCH] 1111

---
 comfyui/main.py       | 9 ++++-----
 comfyui/util/utils.py | 2 +-
 2 files changed, 5 insertions(+), 6 deletions(-)

diff --git a/comfyui/main.py b/comfyui/main.py
index 0c8cc08..1658dbb 100644
--- a/comfyui/main.py
+++ b/comfyui/main.py
@@ -50,7 +50,7 @@ def main(pid,orderId,styleNumber):
 
         #检测是否完成数据，一个死循环
         while True:
-            boolV = checkIsFinish(driver, pid, orderId)
+            boolV = checkIsFinish(driver, pid, orderId,styleNumber)
             if boolV:
                 break
             else:
@@ -63,7 +63,6 @@ def main(pid,orderId,styleNumber):
                 time.sleep(5)
                 continue
 
-
     except Exception as e:
         print(f"发生错误：{e}")
         utils.notify(f"发生错误：{e}")
@@ -169,7 +168,7 @@ if __name__ == '__main__':
     outputFolder = "C:\StableDiffusion_xiaoyu_V3\ComfyUI-aki-v1.6-XY\ComfyUI\output"
     jsonFilePath = "C:\StableDiffusion_xiaoyu_V3\ComfyUI-aki-v1.6-XY\ComfyUI\pysssss-workflows"
     # 设置浏览器驱动路径（以Chrome为例）
-    driver_path = 'D:\chrome\chromedriver-win64\chromedriver.exe'  # 替换为你的chromedriver路径
+    driver_path = 'D:\chromedriver-win64\chromedriver.exe'  # 替换为你的chromedriver路径
 
     #检查文件是否存在，不存在就创建， 测试使用
     if not os.path.exists(inputFolder):
@@ -195,12 +194,12 @@ if __name__ == '__main__':
         r = utils.create_redis_connection()
         while True:
             try:
-                if r.llen('model:badge_cartoon') == 0:
+                if r.llen('model:badge_cartoon_test') == 0:
                     print('队列为空，等待10秒')
                     time.sleep(10)
                     continue
                     
-                info = r.lpop('model:badge_cartoon')
+                info = r.lpop('model:badge_cartoon_test')
                 if info is None:
                     print('队列为空，等待10秒')
                     time.sleep(10)
diff --git a/comfyui/util/utils.py b/comfyui/util/utils.py
index b50b67d..e4a7ed7 100644
--- a/comfyui/util/utils.py
+++ b/comfyui/util/utils.py
@@ -1,4 +1,4 @@
-import toml,time,hmac,hashlib,requests
+import toml,time,hmac,hashlib,requests,json
 import oss2,os
 import redis
 #读取配置文件