From 8e06fdafa078e4ba4917dbb8f9d503ce100f351b Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Thu, 17 Oct 2024 13:38:29 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B8=85=E9=99=A4=E5=9B=A0=E7=A8=8B=E5=BA=8F?=
 =?UTF-8?q?=E9=94=99=E8=AF=AF=E9=81=97=E7=95=99=E7=9A=84=E4=B8=B4=E6=97=B6?=
 =?UTF-8?q?=E5=9B=BE=E7=89=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 photo_mask.py   |  2 +-
 photo_review.py | 12 ++++++++++++
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/photo_mask.py b/photo_mask.py
index 04388f6..23bb989 100644
--- a/photo_mask.py
+++ b/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__':
diff --git a/photo_review.py b/photo_review.py
index 059da5a..8604943 100644
--- a/photo_review.py
+++ b/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'))