清除因程序错误遗留的临时图片

photo_mask.py

@@ -5,10 +5,10 @@ from time import sleep
 
 from sqlalchemy import update
 
-from my_email.error_email import send_error_email
 from db import MysqlSession
 from db.mysql import ZxPhhd
 from log import LOGGING_CONFIG
+from my_email.error_email import send_error_email
 from photo_mask import auto_photo_mask, SEND_ERROR_EMAIL
 
 if __name__ == '__main__':

photo_review.py

@@ -1,5 +1,7 @@
 import argparse
 import logging.config
+import os
+import shutil
 import traceback
 from time import sleep
 
@@ -10,6 +12,7 @@ from db.mysql import ZxPhhd
 from log import LOGGING_CONFIG
 from my_email.error_email import send_error_email
 from photo_review import auto_photo_review, SEND_ERROR_EMAIL
+from util import common_util
 
 # 照片审核自动识别脚本入口
 if __name__ == '__main__':
@@ -23,6 +26,15 @@ if __name__ == '__main__':
     parser.add_argument('--clean', default=False, type=bool, help='是否将识别中的案子改为待识别状态')
     args = parser.parse_args()
     if args.clean:
+        # 清除临时图片
+        for root, dirs, files in os.walk(common_util.get_tmp_img_path(''), topdown=False):
+            for name in dirs:
+                folder_path = os.path.join(root, name)
+                try:
+                    shutil.rmtree(folder_path)
+                    print(f"已删除文件夹: {folder_path}")
+                except Exception as e:
+                    print(f"删除文件夹 {folder_path} 时出错: {e}")
         # 启动时清除仍在识别中的案子
         session = MysqlSession()
         update_flag = (update(ZxPhhd).where(ZxPhhd.exsuccess_flag == '2').values(exsuccess_flag='1'))