redisbench-admin 0.11.17__tar.gz → 0.11.18__tar.gz

{redisbench_admin-0.11.17 → redisbench_admin-0.11.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: redisbench-admin
-Version: 0.11.17
+Version: 0.11.18
 Summary: Redis benchmark run helper. A wrapper around Redis and Redis Modules benchmark tools ( ftsb_redisearch, memtier_benchmark, redis-benchmark, aibench, etc... ).
 Author: filipecosta90
 Author-email: filipecosta.90@gmail.com

{redisbench_admin-0.11.17 → redisbench_admin-0.11.18}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "redisbench-admin"
-version = "0.11.17"
+version = "0.11.18"
 description = "Redis benchmark run helper. A wrapper around Redis and Redis Modules benchmark tools ( ftsb_redisearch, memtier_benchmark, redis-benchmark, aibench, etc... )."
 authors = ["filipecosta90 <filipecosta.90@gmail.com>","Redis Performance Group <performance@redis.com>"]
 readme = "README.md"

{redisbench_admin-0.11.17 → redisbench_admin-0.11.18}/redisbench_admin/utils/remote.py

@@ -159,6 +159,7 @@ def execute_remote_commands(
 
 def connect_remote_ssh(port, private_key, server_public_ip, username):
     import time
+    import socket
 
     k = paramiko.RSAKey.from_private_key_file(private_key)
     c = paramiko.SSHClient()
@@ -182,7 +183,11 @@ def connect_remote_ssh(port, private_key, server_public_ip, username):
             )
             logging.info(f"Connected to remote server {server_public_ip}")
             return c
-        except paramiko.ssh_exception.NoValidConnectionsError as e:
+        except (
+            paramiko.ssh_exception.NoValidConnectionsError,
+            socket.timeout,
+            paramiko.ssh_exception.SSHException,
+        ) as e:
             wait_time = min(base_delay * (2**attempt), max_delay)
             logging.warning(
                 f"Connection attempt {attempt + 1} failed: {e}. Retrying in {wait_time} seconds..."
@@ -190,7 +195,7 @@ def connect_remote_ssh(port, private_key, server_public_ip, username):
             time.sleep(wait_time)
 
     logging.error("Failed to connect after multiple attempts.")
-    raise paramiko.ssh_exception.NoValidConnectionsError(
+    raise Exception(
         f"Could not connect to {server_public_ip} after {max_retries} attempts."
     )