liqian преди 1 година
родител
ревизия
d48f88ecea
променени са 1 файла, в които са добавени 4 реда и са изтрити 2 реда
  1. 4 2
      ai_tag_task.py

+ 4 - 2
ai_tag_task.py

@@ -112,7 +112,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.rmdir(os.path.join(download_folder, video_id))
+                os.removedirs(os.path.join(download_folder, video_id))
             else:
                 video_folder = os.path.join(download_folder, video_id)
                 for filename in os.listdir(video_folder):
@@ -121,7 +121,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.rmdir(os.path.join(download_folder, video_id))
+                        os.removedirs(os.path.join(download_folder, video_id))
+                    else:
+                        os.removedirs(os.path.join(download_folder, video_id))
 
 
 def timer_check():