Browse Source

add save src face

ming 3 years ago
parent
commit
48b3ba784d
1 changed files with 3 additions and 3 deletions
  1. 3 3
      api/main.py

+ 3 - 3
api/main.py

@@ -191,8 +191,8 @@ def gen_avatar(name_hash, imgurl):
             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
+        name_hash = works[0]['name_hash']
+        imgurl = works[0].['imgurl']
         c = rpyc.connect("localhost", 8868)
         c._config['sync_request_timeout'] = None
         remote_svc = c.root
@@ -200,7 +200,7 @@ 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)
+        db['avatar_queue'].delete(id=works[0]['id'])