From aa6296e3505bbc25efb0a39e1d381828bd36ba8c Mon Sep 17 00:00:00 2001 From: AKHIL JHA Date: Sun, 30 Apr 2023 22:59:43 +0530 Subject: [PATCH] Custom socket connect timeout in Session() --- broker/session.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/broker/session.py b/broker/session.py index e3552dc4..3f4cee3a 100644 --- a/broker/session.py +++ b/broker/session.py @@ -31,7 +31,8 @@ def __init__(self, **kwargs): sock.settimeout(kwargs.get("timeout")) port = kwargs.get("port", 22) key_filename = kwargs.get("key_filename") - helpers.simple_retry(sock.connect, [(host, port)]) + socket_connect_timeout = kwargs.get("socket_connect_timeout", 60) + helpers.simple_retry(sock.connect, [(host, port)], max_timeout=socket_connect_timeout) self.session = ssh2_Session() self.session.handshake(sock) if key_filename: