mirror of
https://github.com/tuna/tunasync-scripts.git
synced 2025-12-25 16:32:47 +00:00
pub-mirror: add timeout to request [ci skip]
Signed-off-by: Shengqi Chen <harry-chen@outlook.com>
This commit is contained in:
parent
15a5d17689
commit
ee17188ef5
|
|
@ -31,6 +31,8 @@ MIRROR_URL = os.getenv(
|
||||||
REPOS = []
|
REPOS = []
|
||||||
UA = "tuna-pub-mirror/0.0 (+https://github.com/tuna/tunasync-scripts)"
|
UA = "tuna-pub-mirror/0.0 (+https://github.com/tuna/tunasync-scripts)"
|
||||||
|
|
||||||
|
# connect and read timeout value
|
||||||
|
TIMEOUT_OPTION = (7, 10)
|
||||||
|
|
||||||
# wrap around requests.get to use token if available
|
# wrap around requests.get to use token if available
|
||||||
def get_with_token(*args, **kwargs):
|
def get_with_token(*args, **kwargs):
|
||||||
|
|
@ -39,6 +41,7 @@ def get_with_token(*args, **kwargs):
|
||||||
headers["Authorization"] = "Bearer {}".format(os.environ["PUB_TOKEN"])
|
headers["Authorization"] = "Bearer {}".format(os.environ["PUB_TOKEN"])
|
||||||
headers["User-Agent"] = UA
|
headers["User-Agent"] = UA
|
||||||
kwargs["headers"] = headers
|
kwargs["headers"] = headers
|
||||||
|
kwargs["timeout"] = TIMEOUT_OPTION
|
||||||
return requests.get(*args, **kwargs)
|
return requests.get(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue