ratio1 3.4.39__tar.gz → 3.4.40__tar.gz

{ratio1-3.4.39 → ratio1-3.4.40}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ratio1
-Version: 3.4.39
+Version: 3.4.40
 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.39 → ratio1-3.4.40}/pyproject.toml

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

{ratio1-3.4.39 → ratio1-3.4.40}/ratio1/_ver.py

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

{ratio1-3.4.39 → ratio1-3.4.40}/ratio1/base/generic_session.py

@@ -2696,7 +2696,7 @@ class GenericSession(BaseDecentrAIObject):
       node = self.__get_node_address(node)
       return node in self._dct_online_nodes_pipelines
 
-    def is_peered(self, node):
+    def is_peered(self, node, return_full_address=False):
       """
       Public method for checking if a node is peered with the current session.
       Parameters
@@ -2709,8 +2709,10 @@ class GenericSession(BaseDecentrAIObject):
       bool
           True if the node is peered, False otherwise.
       """
-      node = self.__get_node_address(node)
-      return self._dct_can_send_to_node.get(node, False)
+      node_addr = self.__get_node_address(node)
+      if return_full_address:
+        return self._dct_can_send_to_node.get(node_addr, False), node_addr
+      return self._dct_can_send_to_node.get(node_addr, False)
 
     def get_last_seen_time(self, node, *, default_value=0):
       """

{ratio1-3.4.39 → ratio1-3.4.40}/ratio1/cli/nodes.py

@@ -209,52 +209,39 @@ def get_apps(args):
   return
 
 def _send_command_to_node(args, command, ignore_not_found=False):
+  from ratio1 import Session
+
   node = args.node
   silent = not args.verbose   
   ignore_peering = args.ignore_peering
 
   t1 = time()
-  df, _, _, _, _, sess = _get_netstats(
-    silent=silent, online_only=True, return_session=True, all_info=True,
-    wait_for_node=node
-  )
-  
-  peered = None
-  selection = (df.Alias == node) | (df.Address == node)
-  if 'ETH Address' in df.columns:
-    selection = selection | (df['ETH Address'] == node)
-  found = selection.any()
-  node_addr = None
-  alias = None
-  df_found =  df[selection]
-  if found:
-    alias = df_found.Alias.values[0]
-    peered = df_found.Peered.values[0]
-    node_addr = df_found.Address.values[0]   
-    log_with_color(f"{df_found}")
-  else:
-    log_with_color("Node '{}' <{}> not found in network (total {} nodes, {} peered).".format(
-      node, node_addr, df.shape[0], df.Peered.sum()), color='r'
-    )
-    node_addr = node
+  sess = Session(silent=silent)
+
+  peered, node_addr = sess.is_peered(node, return_full_address=True)
+  found = False
 
-  if not peered and not ignore_peering:
-    log_with_color(f"Node '{node}' <{node_addr}> not peered, exiting...", color='r')
-    return
+  if not ignore_peering:
+    found = sess.wait_for_node(node_addr, timeout=30)
+    sess.P(f"Node {node_addr} is online.")
 
+  # Display peering status.
   if not peered:
+    if not ignore_peering:
+      log_with_color(f"Node <{node_addr}> is not peered. Exiting...", color='r')
+      return
+
     if found:
       log_with_color(f"Node '{node}' <{node_addr}> is not peered.", color='r')
     else:
       log_with_color(f"Node '{node}' <{node_addr}> may not accept this command.", color='r')
 
-  # TODO: currently this is based on node alias, but we should be based on node address
-  #       and maybe even node alias
-  if (found and peered) or ignore_not_found:
-    if found and peered:
-      log_with_color(f"Sending '{command}' to node '{alias}' <{node_addr}>", color='b')
-    else:
-      log_with_color(f"Sending blind '{command}' to node '{alias}' <{node_addr}>", color='b')      
+  if found and peered:
+    log_with_color(f"Sending '{command}' to node <{node_addr}>", color='b')
+  else:
+    log_with_color(f"Sending blind '{command}' to node <{node_addr}>", color='b')
+
+  if (found and peered) or ignore_not_found or ignore_peering:
     if command == COMMANDS.RESTART:
       sess._send_command_restart_node(node_addr)
     elif command == COMMANDS.STOP:
@@ -262,7 +249,12 @@ def _send_command_to_node(args, command, ignore_not_found=False):
     else:
       log_with_color(f"Command '{command}' not supported.", color='r')
       return
-  elapsed = time() - t1  
+
+  sess.close()
+  log_with_color(f"Command successfully sent.")
+  elapsed = time() - t1
+  if not silent:
+    log_with_color(f"Command '{command}' seinging took {elapsed}s.", color='b')
   return  
 
 def restart_node(args):