liqian hace 1 año
padre
commit
b54c92704d
Se han modificado 1 ficheros con 4 adiciones y 3 borrados
  1. 4 3
      ai_tag_task.py

+ 4 - 3
ai_tag_task.py

@@ -1,4 +1,5 @@
 import os
+import shutil
 import json
 import datetime
 import time
@@ -114,7 +115,7 @@ def ai_tags(project, table, dt):
             if video_id not in video_id_list:
                 continue
             if video_info.get(video_id, None) is None:
-                os.removedirs(os.path.join(download_folder, video_id))
+                shutil.rmtree(os.path.join(download_folder, video_id))
             else:
                 video_folder = os.path.join(download_folder, video_id)
                 for filename in os.listdir(video_folder):
@@ -123,9 +124,9 @@ def ai_tags(project, table, dt):
                         video_file = os.path.join(video_folder, filename)
                         get_video_ai_tags(video_id=video_id, video_file=video_file, video_info=video_info.get(video_id))
                         # 将处理过的视频进行删除
-                        os.removedirs(os.path.join(download_folder, video_id))
+                        shutil.rmtree(os.path.join(download_folder, video_id))
                     else:
-                        os.removedirs(os.path.join(download_folder, video_id))
+                        shutil.rmtree(os.path.join(download_folder, video_id))
 
 
 def timer_check():