照片识别的重试机制依靠docker即可实现,删除多余代码

This commit is contained in:
2024-07-26 10:51:52 +08:00
parent c304d97609
commit 22fa38525f
3 changed files with 23 additions and 26 deletions

View File

@@ -36,4 +36,4 @@ if __name__ == '__main__':
error_logger = logging.getLogger("error") error_logger = logging.getLogger("error")
error_logger.error(traceback.format_exc()) error_logger.error(traceback.format_exc())
if SEND_ERROR_EMAIL: if SEND_ERROR_EMAIL:
send_error_email(program_name=program_name, error_name=repr(e), error_detail=traceback.format_exc()) send_error_email(program_name, repr(e), traceback.format_exc())

View File

@@ -9,33 +9,32 @@ from auto_email.error_email import send_error_email
from db import MysqlSession from db import MysqlSession
from db.mysql import ZxPhhd from db.mysql import ZxPhhd
from log import LOGGING_CONFIG from log import LOGGING_CONFIG
from photo_review import photo_review, SEND_ERROR_EMAIL, RETRY_TIME from photo_review import photo_review, SEND_ERROR_EMAIL
# 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 # 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生
if __name__ == '__main__': if __name__ == '__main__':
program_name = '照片审核自动识别脚本' program_name = '照片审核自动识别脚本'
logging.config.dictConfig(LOGGING_CONFIG) logging.config.dictConfig(LOGGING_CONFIG)
for i in range(RETRY_TIME + 1): parser = argparse.ArgumentParser()
parser = argparse.ArgumentParser() parser.add_argument("--clean", default=False, type=bool, help="是否将识别中的案子改为待识别状态")
parser.add_argument("--clean", default=False, type=bool, help="是否将识别中的案子改为待识别状态") args = parser.parse_args()
args = parser.parse_args() if args.clean:
if args.clean or i > 0: # 主要用于启动时,清除仍在涂抹中的案子
# 主要用于启动时,清除仍在涂抹中的案子 session = MysqlSession()
session = MysqlSession() update_flag = (update(ZxPhhd).where(ZxPhhd.exsuccess_flag == "2").values(exsuccess_flag="1"))
update_flag = (update(ZxPhhd).where(ZxPhhd.exsuccess_flag == "2").values(exsuccess_flag="1")) session.execute(update_flag)
session.execute(update_flag) session.commit()
session.commit() session.close()
session.close() logging.info("已释放残余的识别案子!")
logging.info("已释放残余的识别案子!") else:
else: sleep(5)
sleep(5)
try: try:
logging.info(f"{program_name}】开始运行") logging.info(f"{program_name}】开始运行")
photo_review.main() photo_review.main()
except Exception as e: except Exception as e:
error_logger = logging.getLogger('error') error_logger = logging.getLogger('error')
error_logger.error(traceback.format_exc()) error_logger.error(traceback.format_exc())
if SEND_ERROR_EMAIL: if SEND_ERROR_EMAIL:
send_error_email(program_name, repr(e), traceback.format_exc()) send_error_email(program_name, repr(e), traceback.format_exc())

View File

@@ -9,8 +9,6 @@ HOSTNAME = socket.gethostname()
""" """
项目配置 项目配置
""" """
# 项目奔溃的重试次数
RETRY_TIME = 1
# 每次从数据库获取的案子数量 # 每次从数据库获取的案子数量
PHHD_BATCH_SIZE = 10 PHHD_BATCH_SIZE = 10
# 没有查询到案子的等待时间(分钟) # 没有查询到案子的等待时间(分钟)