整理参数配置

config/photo_review.py

@@ -0,0 +1,8 @@
+# 项目奔溃的重试次数
+RETRY_TIME = 2
+
+# 每次从数据库获取的案子数量
+PHHD_BATCH_SIZE = 10
+
+# 没有查询到案子的等待时间(分钟)
+SLEEP_MINUTES = 5

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("照片审核开始")

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)