add save src face

api/main.py

@@ -183,14 +183,24 @@ 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:
-        
         db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/AI_anchor?charset=utf8mb4')
+
+        statement = 'SELECT * FROM avatar_service_status'
+        status = -1
+        for row in db.query(statement):
+            status = row['status']
+        if status == 1:
+            break
+
         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
+        
+        statement = 'UPDATE table_name SET status = 1 WHERE id=1;'
+        db.query(statement):
         name_hash = works[0]['name_hash']
         imgurl = works[0].['imgurl']
         c = rpyc.connect("localhost", 8868)
@@ -201,6 +211,8 @@ def gen_avatar(name_hash, imgurl):
         shutil.copy(tmp_avatar_dir+name_hash+'.mp4',avatar_dest+name_hash+'.mp4')
         os.remove(tmp_avatar_dir+name_hash+'.mp4')
         db['avatar_queue'].delete(id=works[0]['id'])
+        statement = 'UPDATE table_name SET status = 0 WHERE id=1;'
+        db.query(statement):