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

Fix timeout setting in CustomXMLRPCTransport

Browse Source
This commit is contained in:
taoky 2025-01-13 21:12:46 +08:00
parent 5dd445507d
commit bbcab99b4d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
shadowmire.py
View File

@ -24,6 +24,7 @@ import tomllib
from copy import deepcopy from copy import deepcopy
import functools import functools
from http.client import HTTPConnection from http.client import HTTPConnection
import socket
import requests import requests
import click import click
@ -298,8 +299,9 @@ class CustomXMLRPCTransport(xmlrpc.client.Transport):
def make_connection(self, host: tuple[str, dict[str, str]] | str) -> HTTPConnection: def make_connection(self, host: tuple[str, dict[str, str]] | str) -> HTTPConnection:
conn = super().make_connection(host) conn = super().make_connection(host)
if conn.timeout is None: if socket.getdefaulttimeout() is None:
# 2 min timeout # By default conn.timeout is socket._GLOBAL_DEFAULT_TIMEOUT instead of None.
# So here we check if default timeout is set, and if not, add a 2-min timeout
conn.timeout = 120 conn.timeout = 120
return conn return conn