为照片涂抹编写docker

photo_mask.py

@@ -1,13 +1,30 @@
+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_mask import photo_mask, SEND_ERROR_EMAIL
 
 if __name__ == '__main__':
     program_name = "照片审核自动涂抹脚本"
     logging.config.dictConfig(LOGGING_CONFIG)
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--clean", default=False, type=bool, help="是否将涂抹中的案子改为待涂抹状态")
+    args = parser.parse_args()
+    if args.clean:
+        # 主要用于启动时，清除仍在涂抹中的案子
+        session = MysqlSession()
+        update_flag = (update(ZxPhhd).where(ZxPhhd.paint_flag == "2").values(paint_flag="1"))
+        session.execute(update_flag)
+        session.commit()
+        session.close()
+
     try:
         logging.info(f"【{program_name}】开始运行")
         photo_mask.main()