diff --git a/config/photo_review.py b/config/photo_review.py new file mode 100644 index 0000000..89a307a --- /dev/null +++ b/config/photo_review.py @@ -0,0 +1,8 @@ +# 项目奔溃的重试次数 +RETRY_TIME = 2 + +# 每次从数据库获取的案子数量 +PHHD_BATCH_SIZE = 10 + +# 没有查询到案子的等待时间(分钟) +SLEEP_MINUTES = 5 diff --git a/main.py b/main.py index 770cd5b..a947d9c 100644 --- a/main.py +++ b/main.py @@ -3,6 +3,7 @@ import traceback from auto_email.error_email import send_an_error_email from config.log import LOGGING_CONFIG +from config.photo_review import RETRY_TIME from photo_review.photo_review import main # 项目必须从此处启动,否则代码中的相对路径可能导致错误的发生 @@ -11,7 +12,7 @@ if __name__ == '__main__': log = logging.getLogger() # 崩溃后的重试次数 - retry_time = 2 + retry_time = RETRY_TIME for _ in range(++retry_time): try: log.info("照片审核开始") diff --git a/photo_review/photo_review.py b/photo_review/photo_review.py index bc7bdc7..8bb3a84 100644 --- a/photo_review/photo_review.py +++ b/photo_review/photo_review.py @@ -7,6 +7,7 @@ from paddlenlp import Taskflow from config.keys import PATIENT_NAME, ADMISSION_DATE, DISCHARGE_DATE, MEDICAL_EXPENSES, PERSONAL_CASH_PAYMENT, \ PERSONAL_ACCOUNT_PAYMENT, PERSONAL_FUNDED_AMOUNT, MEDICAL_INSURANCE_TYPE, HOSPITAL, DEPARTMENT, DOCTOR from config.mysql import MysqlSession +from config.photo_review import PHHD_BATCH_SIZE, SLEEP_MINUTES from photo_review.entity.bd_yljg import BdYljg from photo_review.entity.bd_ylks import BdYlks from photo_review.entity.zx_ie_cost import ZxIeCost @@ -204,7 +205,7 @@ def main(): phhds = session.query(ZxPhhd.pk_phhd) \ .filter(ZxPhhd.pk_phhd > last_pk_phhd) \ .filter(ZxPhhd.cStatus == '2') \ - .limit(1) \ + .limit(PHHD_BATCH_SIZE) \ .all() session.close() if phhds: @@ -213,8 +214,8 @@ def main(): photo_review(pk_phhd) last_pk_phhd = pk_phhd else: - # 没有查询到新案子,等待 5 分钟后再查 - sleep_minutes = 5 + # 没有查询到新案子,等待一段时间后再查 + sleep_minutes = SLEEP_MINUTES log = logging.getLogger() log.info(f"暂未查询到新案子,等待{sleep_minutes}分钟...") sleep(sleep_minutes * 60)