Przeglądaj źródła

add save src face

ming 3 lat temu
rodzic
commit
ac419239de
1 zmienionych plików z 3 dodań i 2 usunięć
  1. 3 2
      api/main.py

+ 3 - 2
api/main.py

@@ -183,13 +183,14 @@ def gen_avatar(name_hash, imgurl):
     db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
     db['avatar_queue'].insert({'name_hash':name_hash,'imgurl':imgurl})
     while True:
-        if len(works)==0:
-            break
+        
         db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
         statement = 'SELECT * FROM avatar_queue'
         works = []
         for row in db.query(statement):
             works.append({'id':row['id'],'name_hash':row['name_hash'],'imgurl':['imgurl']})
+        if len(works)==0:
+            break
         name_hash = works[0].name_hash
         imgurl = works[0].imgurl
         c = rpyc.connect("localhost", 8868)