소스 검색

dlt 数据获取增加超时和捕获错误

jack 1 년 전
부모
커밋
0be6b7dc50
1개의 변경된 파일10개의 추가작업 그리고 6개의 파일을 삭제
  1. 10 6
      message/message_dlt.py

+ 10 - 6
message/message_dlt.py

@@ -32,11 +32,15 @@ class CheckDlt:
         ]
 
     def req(self):
-        resp = httpx.get(self.url, headers=self.headers)
-        if resp.status_code != 200:
-            print('state code: {}'.format(resp.status_code))
-            log_detail = '访问失败, 状态码:{},url:{}'.format(resp.status_code, self.url)
-            print(log_detail)
+        try:
+            resp = httpx.get(self.url, headers=self.headers, timeout=5)
+            if resp.status_code != 200:
+                print('state code: {}'.format(resp.status_code))
+                log_detail = '访问失败, 状态码:{},url:{}'.format(resp.status_code, self.url)
+                print(log_detail)
+                exit(0)
+        except Exception as e:
+            print(str(e))
             exit(0)
 
         resp_json = resp.json()
@@ -141,7 +145,7 @@ class CheckDlt:
         if not text:
             exit(0)
 
-        title = f'大乐透 {subject} 期'
+        title = f'dlt {subject}'
 
         # 推送到 message
         GotifyNotifier(title, text, 'A3bqt9Dlbs.fPUb').send_message()