liqian 1 year ago
parent
commit
b4318441fc
2 changed files with 7 additions and 4 deletions
  1. 1 1
      ai_tag_task.py
  2. 6 3
      asr_task.py

+ 1 - 1
ai_tag_task.py

@@ -182,7 +182,7 @@ def ai_tags_new(project, table, dt):
             retry += 1
             time.sleep(60)
             continue
-
+        retry = 0
         for asr_filename in asr_file_list:
             video_id = asr_filename[:-4]
             if video_id not in video_id_list:

+ 6 - 3
asr_task.py

@@ -45,7 +45,10 @@ def asr_process(project, table, dt):
             if video_id not in video_id_list:
                 continue
             if video_info.get(video_id, None) is None:
-                shutil.rmtree(os.path.join(download_folder, video_id))
+                try:
+                    shutil.rmtree(os.path.join(download_folder, video_id))
+                except:
+                    continue
             else:
                 video_folder = os.path.join(download_folder, video_id)
                 for filename in os.listdir(video_folder):
@@ -61,8 +64,8 @@ def asr_process(project, table, dt):
                             wf.write(asr_res_initial)
                         # 将处理过的视频进行删除
                         shutil.rmtree(os.path.join(download_folder, video_id))
-                    else:
-                        shutil.rmtree(os.path.join(download_folder, video_id))
+                        break
+
 
 def timer_check():
     try: