为照片识别编写docker

photo_review.py

@@ -1,7 +1,12 @@
+import argparse
 import logging.config
 import traceback
 
+from sqlalchemy import update
+
 from auto_email.error_email import send_error_email
+from db import MysqlSession
+from db.mysql import ZxPhhd
 from log import LOGGING_CONFIG
 from photo_review import photo_review, SEND_ERROR_EMAIL, RETRY_TIME
 
@@ -10,8 +15,19 @@ if __name__ == '__main__':
     program_name = '照片审核自动识别脚本'
     logging.config.dictConfig(LOGGING_CONFIG)
 
-    # 崩溃后的重试次数
-    for _ in range(RETRY_TIME + 1):
+    for i in range(RETRY_TIME + 1):
+        parser = argparse.ArgumentParser()
+        parser.add_argument("--clean", default=False, type=bool, help="是否将识别中的案子改为待识别状态")
+        args = parser.parse_args()
+        if args.clean or i > 0:
+            # 主要用于启动时，清除仍在涂抹中的案子
+            session = MysqlSession()
+            update_flag = (update(ZxPhhd).where(ZxPhhd.exsuccess_flag == "2").values(exsuccess_flag="1"))
+            session.execute(update_flag)
+            session.commit()
+            session.close()
+            logging.info("已释放残余的识别案子！")
+
         try:
             logging.info(f"【{program_name}】开始运行")
             photo_review.main()