From f41fcc89d7a810df38663632a24418a07a85e5ca Mon Sep 17 00:00:00 2001
From: liuyebo <1515783401@qq.com>
Date: Mon, 9 Sep 2024 10:03:33 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85=E6=97=A0=E9=94=A1=E5=8C=BB?=
 =?UTF-8?q?=E9=99=A2=E5=88=A4=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker-compose.yml                | 12 ++++++++++--
 photo_review/auto_photo_review.py |  5 +++--
 util/html_util.py                 |  6 +++---
 3 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index aa78057..5e9a052 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,6 +1,6 @@
 x-env:
   &template
-  image: fcb_photo_review:1.14.2
+  image: fcb_photo_review:1.14.3
   restart: always
 
 x-review:
@@ -103,4 +103,12 @@ services:
     hostname: photo_mask_2
     depends_on:
       - photo_mask_1
-    command: [ 'photo_mask.py' ]
\ No newline at end of file
+    command: [ 'photo_mask.py' ]
+
+  photo_review_6:
+    <<: *review_template
+    container_name: photo_review_6
+    hostname: photo_review_6
+    depends_on:
+      - photo_mask_2
+    command: [ 'photo_review.py' ]
\ No newline at end of file
diff --git a/photo_review/auto_photo_review.py b/photo_review/auto_photo_review.py
index 4342fa9..5ca5d6f 100644
--- a/photo_review/auto_photo_review.py
+++ b/photo_review/auto_photo_review.py
@@ -121,8 +121,9 @@ def get_better_image_from_qrcode(image, image_id, dpi=150):
 
                 pdf_url = f'{jsczt_base_url}/download?idBase={id_base}'
                 return _parse_pdf_url(pdf_url)
-            elif url.startswith('http://dzfp.wxxsh.net'):  # 无锡市锡山人民医院
-                pdf_url = html_util.get_wxxsh_pdf_url(url)
+            elif url.startswith('http://dzfp.wxxsh.net') or url.startswith('http://dzpj.wuxi5h.com'):
+                # 无锡市锡山人民医院、无锡市第五人民医院
+                pdf_url = html_util.get_wx_pdf_url(url)
                 if not pdf_url:
                     continue
 
diff --git a/util/html_util.py b/util/html_util.py
index 16c1525..fb929af 100644
--- a/util/html_util.py
+++ b/util/html_util.py
@@ -44,11 +44,11 @@ def download_pdf(url, local_filename=None):
 
 
 @retry(stop=stop_after_attempt(3), wait=wait_random(1, 3), reraise=True,
-       after=lambda x: logging.warning('获取无锡锡山人民医院票据失败！'))
-def get_wxxsh_pdf_url(url):
+       after=lambda x: logging.warning('获取无锡医院票据失败！'))
+def get_wx_pdf_url(url):
     response = requests.get(url)
     if response.status_code != 200:
-        raise Exception(f'请求无锡锡山人民医院票据失败！状态码: {response.status_code}')
+        raise Exception(f'请求无锡医院票据失败！状态码: {response.status_code}')
     soup = BeautifulSoup(response.text, 'html.parser')
     pdf_url = soup.find('a', string='点击查看电子票据')
     if pdf_url: