redisbench-admin 0.11.14__tar.gz → 0.11.15__tar.gz

{redisbench_admin-0.11.14 → redisbench_admin-0.11.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: redisbench-admin
-Version: 0.11.14
+Version: 0.11.15
 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.14 → redisbench_admin-0.11.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "redisbench-admin"
-version = "0.11.14"
+version = "0.11.15"
 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.14 → redisbench_admin-0.11.15}/redisbench_admin/run/ssh.py

@@ -70,6 +70,19 @@ def check_connection(ssh_conn):
         return False
 
 
+def ensure_400_permissions(file_path):
+    import stat
+
+    file_stat = os.stat(file_path)
+    current_permissions = stat.S_IMODE(file_stat.st_mode)
+
+    if current_permissions != 0o400:
+        logging.info(f"Changing permissions of {file_path} to 400")
+        os.chmod(file_path, 0o400)
+    else:
+        logging.info(f"{file_path} already has 400 permissions.")
+
+
 def ssh_pem_check(EC2_PRIVATE_PEM, private_key):
     if os.path.exists(private_key) is False:
         if EC2_PRIVATE_PEM is not None and EC2_PRIVATE_PEM != "":
@@ -90,3 +103,4 @@ def ssh_pem_check(EC2_PRIVATE_PEM, private_key):
         logging.info(
             "Confirmed that private key path artifact: '{}' exists!".format(private_key)
         )
+    ensure_400_permissions(private_key)