From 66ad9c9e5cfea9f62ef3d798b5cd69840320d9ac Mon Sep 17 00:00:00 2001 From: Jalin Date: Tue, 29 Jan 2019 15:37:04 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E7=BD=91=E7=BB=9C=E9=BB=98?= =?UTF-8?q?=E8=AE=A4=E8=B6=85=E6=97=B6=E6=97=B6=E9=97=B4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- py12306/config.py | 3 +++ py12306/helpers/request.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/py12306/config.py b/py12306/config.py index 0e69ef3..8709dcb 100644 --- a/py12306/config.py +++ b/py12306/config.py @@ -90,6 +90,9 @@ class Config: CDN_ITEM_FILE = PROJECT_DIR + 'data/cdn.txt' CDN_ENABLED_AVAILABLE_ITEM_FILE = QUERY_DATA_DIR + 'available.json' + # Default time out + TIME_OUT_OF_REQUEST = 5 + envs = [] retry_time = 5 last_modify_time = 0 diff --git a/py12306/helpers/request.py b/py12306/helpers/request.py index a24606c..a65e3af 100644 --- a/py12306/helpers/request.py +++ b/py12306/helpers/request.py @@ -53,6 +53,9 @@ class Request(HTMLSession): def request(self, *args, **kwargs): # 拦截所有错误 try: + if not 'timeout' in kwargs: + from py12306.config import Config + kwargs['timeout'] = Config().TIME_OUT_OF_REQUEST response = super().request(*args, **kwargs) return response except RequestException as e: