ratio1 3.4.63__tar.gz → 3.4.64__tar.gz

{ratio1-3.4.63 → ratio1-3.4.64}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ratio1
-Version: 3.4.63
+Version: 3.4.64
 Summary: `ratio1` or Ration1 SDK is the Python SDK required for client app development for the Ratio1 ecosystem
 Project-URL: Homepage, https://github.com/Ratio1/ratio1_sdk
 Project-URL: Bug Tracker, https://github.com/Ratio1/ratio1_sdk/issues

{ratio1-3.4.63 → ratio1-3.4.64}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "ratio1"
-version = "3.4.63"
+version = "3.4.64"
 authors = [
   { name="Andrei Ionut Damian", email="andrei.damian@ratio1.ai" },
   { name="Cristan Bleotiu", email="cristian.bleotiu@ratio1.ai" },

{ratio1-3.4.63 → ratio1-3.4.64}/ratio1/_ver.py

@@ -1,4 +1,4 @@
-__VER__ = "3.4.63"
+__VER__ = "3.4.64"
 
 if __name__ == "__main__":
   with open("pyproject.toml", "rt") as fd:

{ratio1-3.4.63 → ratio1-3.4.64}/ratio1/cli/cli_commands.py

@@ -143,7 +143,12 @@ CLI_COMMANDS = {
     },
     "oracle-rollout": {
         "func": oracle_rollout,
-        "description": "Rollout update on all nodes in the network. The rollout order is seed nodes -> oracle nodes -> all other edge nodes. This command is needed when defining new environment variables in seed nodes, in order to make it available to all nodes in the network."
+        "description": "Rollout update on all nodes in the network. The rollout order is seed nodes -> oracle nodes -> all other edge nodes. This command is needed when defining new environment variables in seed nodes, in order to make it available to all nodes in the network.",
+        "params": {
+            "--skip-seeds": "Skip the seed nodes in the rollout (flag)",
+            "--skip-oracles": "Skip the oracle nodes in the rollout (flag)",
+            "--no-timeout": "Do not wait between nodes restarts (flag)",
+        }
     },
     "inspect": {
         "func": inspect_node,

{ratio1-3.4.63 → ratio1-3.4.64}/ratio1/cli/oracles.py

@@ -219,6 +219,13 @@ if True:
     This function is needed in order to ensure that all nodes in the network receive the new configuration defined on the seed nodes.
     """
     silent = not args.verbose
+    skip_seeds = args.skip_seeds
+    skip_oracles = args.skip_oracles
+    no_timeout = args.no_timeout
+
+    log_with_color("======================================================", color='b')
+    log_with_color("Starting Oracle Rollout...", color='g')
+    log_with_color("======================================================", color='b')
     session = Session(
       silent=True
     )
@@ -227,6 +234,10 @@ if True:
 
     log_with_color(f"ATTENTION! Current network: {current_network}", color='y')
     log_with_color(f"Are you sure you want to restart ALL nodes on the network {current_network}?", color='b')
+    if skip_seeds:
+      log_with_color("During this run seed nodes will be skipped...", color='b')
+    if skip_oracles:
+      log_with_color("During this run oracle nodes will be skipped...", color='b')
     user_confirmation = input(f"Write down 'RESTART ALL on {current_network}' in order to proceed...\n >")
 
     if user_confirmation != f"RESTART ALL on {current_network}":
@@ -246,15 +257,19 @@ if True:
       if node['address'] not in seed_nodes_addresses
     ]
 
-    # 1. Send restart command to Seed Nodes.
-    log_with_color(f"Sending restart commands to {len(seed_nodes_addresses)} seed nodes: {seed_nodes_addresses}", color='b')
-    _send_restart_command(session=session, nodes=seed_nodes_addresses)
+    if not (skip_seeds or skip_oracles):
+      # 1. Send restart command to Seed Nodes.
+      log_with_color(f"Sending restart commands to {len(seed_nodes_addresses)} seed nodes: {seed_nodes_addresses}",
+                     color='b')
+      _send_restart_command(session=session, nodes=seed_nodes_addresses)
 
-    # Remove seed node addresses from all_nodes_addresses
-    log_with_color(
-      f"Seed nodes restarted. Waiting 30 seconds before sending restart commands to all Oracle nodes, except seed nodes.",
-      color='g')
-    sleep(30)
+      # Remove seed node addresses from all_nodes_addresses
+      log_with_color(
+        f"Seed nodes restarted. Waiting 30 seconds before sending restart commands to all Oracle nodes, except seed nodes.",
+        color='g')
+      sleep(30)
+    else:
+      log_with_color("Skipping Seed Nodes restart as per user request.", color='y')
 
     # 2. Send restart commands to all Oracle nodes, except seed nodes.
     oracle_nodes_addresses = [
@@ -263,9 +278,14 @@ if True:
       if node['oracle'] == True
     ]
 
-    log_with_color(f"Sending restart commands to {len(oracle_nodes_addresses)} Non-Seed Oracle nodes, except seed nodes: {remaining_nodes}", color='b')
+    if not skip_oracles:
+      log_with_color(
+        f"Sending restart commands to {len(oracle_nodes_addresses)} Non-Seed Oracle nodes, except seed nodes: {remaining_nodes}",
+        color='b')
 
-    _send_restart_command(session=session, nodes=oracle_nodes_addresses)
+      _send_restart_command(session=session, nodes=oracle_nodes_addresses)
+    else:
+      log_with_color("Skipping Oracle Nodes restart as per user request.", color='y')
 
     # Remove oracle node addresses from all_nodes_addresses
     remaining_nodes_addresses = [
@@ -282,10 +302,18 @@ if True:
     # 3. Send restart command to all remaining edge nodes.
     log_with_color(f"Sending restart commands to {len(remaining_nodes_addresses)} remaining edge nodes: {remaining_nodes_addresses}", color='b')
 
-    _send_restart_command(session=session, nodes=remaining_nodes_addresses, timeout_min=5, timeout_max=25)
-
-    restarted_seed_nodes_count = len(seed_nodes_addresses)
-    restarted_oracle_nodes_count = len(oracle_nodes_addresses)
+    timeout_min = 0
+    timeout_max = 0
+    if not no_timeout:
+      timeout_min = 5
+      timeout_max = 25
+    _send_restart_command(session=session, nodes=remaining_nodes_addresses, timeout_min=timeout_min, timeout_max=timeout_max)
+    restarted_seed_nodes_count = 0
+    restarted_oracle_nodes_count = 0
+    if not (skip_seeds or skip_oracles):
+        restarted_seed_nodes_count = len(seed_nodes_addresses)
+    if not skip_oracles:
+      restarted_oracle_nodes_count = len(oracle_nodes_addresses)
     restarted_edge_nodes_count = len(remaining_nodes_addresses)
     total_restarted_nodes_count = restarted_seed_nodes_count + restarted_oracle_nodes_count + restarted_edge_nodes_count
     log_with_color(f"All nodes restarted successfully.", color='g')