yacloud/shadowmire
1
0
Fork 0
mirror of https://github.com/taoky/shadowmire.git synced 2025-07-08 09:12:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add retry for requests.Session

Browse Source
This commit is contained in:
taoky 2024-08-03 02:35:42 +08:00
parent 9ecdfee147
commit ada070d10e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
shadowmire.py
View File

@ -17,6 +17,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
import signal import signal
import requests import requests
from tqdm import tqdm from tqdm import tqdm
from requests.adapters import HTTPAdapter, Retry
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -192,6 +193,9 @@ class CustomXMLRPCTransport(xmlrpc.client.Transport):
def create_requests_session() -> requests.Session: def create_requests_session() -> requests.Session:
s = 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}) s.headers.update({"User-Agent": USER_AGENT})
return s return s