1111

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)

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
 #读取配置文件