diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index ce8dd70..f60eb0d 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -430,11 +430,11 @@ def main(): while 1: session = MysqlSession() # 查询需要识别的案子 - phhds = session.query(ZxPhhd.pk_phhd, ZxPhhd.exsuccess_flag).filter(ZxPhhd.exsuccess_flag == '1').limit( - PHHD_BATCH_SIZE).all() + phhds = session.query(ZxPhhd.pk_phhd).filter(ZxPhhd.exsuccess_flag == '1').limit(PHHD_BATCH_SIZE).all() # 将状态改为正在识别中 - for phhd in phhds: - phhd.exsuccess_flag = '2' + pk_phhd_values = [phhd.pk_phhd for phhd in phhds] + update_flag = (update(ZxPhhd).where(ZxPhhd.pk_phhd.in_(pk_phhd_values)).values(exsuccess_flag=2)) + session.execute(update_flag) session.commit() session.close() if phhds: