From ada070d10e062ecfb4b26bcac615705f1642d645 Mon Sep 17 00:00:00 2001 From: taoky Date: Sat, 3 Aug 2024 02:35:42 +0800 Subject: [PATCH] Add retry for requests.Session --- shadowmire.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/shadowmire.py b/shadowmire.py index 1b0acb8..f04ac8c 100755 --- a/shadowmire.py +++ b/shadowmire.py @@ -17,6 +17,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed import signal import requests from tqdm import tqdm +from requests.adapters import HTTPAdapter, Retry logger = logging.getLogger(__name__) @@ -192,6 +193,9 @@ class CustomXMLRPCTransport(xmlrpc.client.Transport): def create_requests_session() -> requests.Session: s = requests.Session() + retries = Retry(total=3, backoff_factor=0.1) + s.mount('http://', HTTPAdapter(max_retries=retries)) + s.mount('https://', HTTPAdapter(max_retries=retries)) s.headers.update({"User-Agent": USER_AGENT}) return s