mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add timeout to FtRestClient

Browse Source
This commit is contained in:
David Arena
2024-12-20 23:47:18 +01:00
parent 0b0bbd4c42
commit 7d5f021717
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ft_client/freqtrade_client/ft_rest_client.py
+3 -2
View File
@@ -23,10 +23,11 @@ PostDataT = dict[str, Any] | list[dict[str, Any]] | None
class FtRestClient: class FtRestClient:
def __init__( def __init__(
self, serverurl, username=None, password=None, *, pool_connections=10, pool_maxsize=10 self, serverurl, username=None, password=None, *, pool_connections=10, pool_maxsize=10, timeout=10
): ):
self._serverurl = serverurl self._serverurl = serverurl
self._session = requests.Session() self._session = requests.Session()
self._timeout = timeout
# allow configuration of pool # allow configuration of pool
adapter = HTTPAdapter(pool_connections=pool_connections, pool_maxsize=pool_maxsize) adapter = HTTPAdapter(pool_connections=pool_connections, pool_maxsize=pool_maxsize)
@@ -50,7 +51,7 @@ class FtRestClient:
url = urlunparse((schema, netloc, path, par, query, fragment)) url = urlunparse((schema, netloc, path, par, query, fragment))
try: try:
resp = self._session.request(method, url, headers=hd, data=json.dumps(data)) resp = self._session.request(method, url, headers=hd, timeout=self._timeout, data=json.dumps(data))
# return resp.text # return resp.text
return resp.json() return resp.json()
except RequestConnectionError: except RequestConnectionError: