naeural-client 2.3.1__tar.gz → 2.3.2__tar.gz

{naeural_client-2.3.1 → naeural_client-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: naeural_client
-Version: 2.3.1
+Version: 2.3.2
 Summary: `naeural_client` is the Python SDK required for client app development for the Naeural Edge Protocol Edge Protocol framework
 Project-URL: Homepage, https://github.com/NaeuralEdgeProtocol/naeural_client
 Project-URL: Bug Tracker, https://github.com/NaeuralEdgeProtocol/naeural_client/issues

{naeural_client-2.3.1 → naeural_client-2.3.2}/naeural_client/_ver.py

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

naeural_client-2.3.2/naeural_client/cli/cli.py

@@ -0,0 +1,90 @@
+import argparse
+
+from naeural_client.utils.config import maybe_init_config, log_with_color
+from naeural_client.cli.cli_commands import CLI_COMMANDS
+
+from naeural_client import version
+
+def build_parser():
+  """
+  Dynamically builds the argument parser based on CLI_COMMANDS.
+
+  Returns
+  -------
+  argparse.ArgumentParser
+      Configured argument parser.
+  """
+  descr = f"nepctl v{version} - CLI for Naeural Edge Protocol SDK package"
+  parser = argparse.ArgumentParser(description=descr)
+  subparsers = parser.add_subparsers(dest="command", help="Available commands")
+
+  for command, subcommands in CLI_COMMANDS.items():
+    command_parser = subparsers.add_parser(command, help=f"{command} commands")
+
+    if isinstance(subcommands, dict) and "func" not in subcommands:
+      # Nested subcommands
+      command_subparsers = command_parser.add_subparsers(dest="subcommand")
+      for subcommand, subcmd_info in subcommands.items():
+        subcommand_parser = command_subparsers.add_parser(
+          subcommand, help=f"{subcommand} command"
+        )
+        if isinstance(subcmd_info, dict) and "params" in subcmd_info:
+          for param, description in subcmd_info["params"].items():
+            if param.startswith("--"):
+              subcommand_parser.add_argument(
+                param, action="store_true", help=description
+              )
+            else:
+              subcommand_parser.add_argument(
+                param, help=description
+              )
+            #end if
+          #end for
+        #end if
+        subcommand_parser.set_defaults(func=subcmd_info["func"])
+    else:
+      # Single-level commands with parameters
+      if "params" in subcommands:
+        for param, description in subcommands["params"].items():
+          if param.startswith("--"):
+            command_parser.add_argument(
+              param, action="store_true", help=description
+            )
+          else:
+            command_parser.add_argument(
+              param, help=description
+            )
+          #end if
+      command_parser.set_defaults(func=subcommands["func"])
+
+  return parser
+
+
+
+def main():
+  """
+  Main entry point for the CLI.
+  Ensures the configuration is initialized, builds the parser, 
+  and executes the appropriate command function.
+  """
+  try:
+    # Initialize configuration if necessary
+    maybe_init_config()
+
+    # Build the CLI parser
+    parser = build_parser()
+    args = parser.parse_args()
+
+    # Check if a command function is provided
+    if hasattr(args, "func"):
+      args.func(args)  # Pass parsed arguments to the command function
+    else:
+      parser.print_help()
+
+  except Exception as e:
+    # Handle unexpected errors gracefully
+    print(f"Error: {e}")
+
+
+if __name__ == "__main__":
+  main()

naeural_client-2.3.2/naeural_client/cli/cli_commands.py

@@ -0,0 +1,42 @@
+from naeural_client.cli.nodes import (
+  get_nodes, get_supervisors, 
+  restart_node, shutdown_node
+)
+from naeural_client.utils.config import show_config, reset_config
+
+
+# Define the available commands
+CLI_COMMANDS = {
+    "get": {
+        "nodes": {
+            "func": get_nodes,
+            "params": {
+                "--all": "Get all nodes", 
+                "--peered": "Get only peered nodes"
+            }
+        },
+        "supervisors": {
+            "func": get_supervisors,
+        },
+    },
+    "config": {
+        "show": {
+            "func": show_config,
+        },
+        "reset": {
+            "func": reset_config,
+        },
+    },
+    "restart": {
+        "func": restart_node,
+        "params": {
+            "node": "The node to restart"
+        }
+    },
+    "shutdown": {
+        "func": shutdown_node,
+        "params": {
+            "node": "The node to shutdown"
+        }
+    }
+}

naeural_client-2.3.2/naeural_client/cli/nodes.py

@@ -0,0 +1,53 @@
+from naeural_client.utils.config import log_with_color
+
+
+def get_nodes(args):
+  """
+  This function is used to get the information about the nodes and it will perform the following:
+  
+  1. Create a Session object.
+  2. Wait for the first net mon message via Session and show progress. 
+  3. Wait for the second net mon message via Session and show progress.  
+  4. Get the active nodes union via Session and display the nodes marking those peered vs non-peered.
+  """
+  if args.all:
+    log_with_color("Getting all nodes information", color="b")
+  elif args.peered:
+    log_with_color("Getting peered nodes information", color="b")
+  else:
+    log_with_color("Getting default nodes information", color="b")
+  return
+  
+  
+def get_supervisors(args):
+  """
+  This function is used to get the information about the supervisors.
+  """
+  log_with_color("Getting supervisors information", color='b')
+  return
+
+
+def restart_node(args):
+  """
+  This function is used to restart the node.
+  
+  Parameters
+  ----------
+  args : argparse.Namespace
+      Arguments passed to the function.
+  """
+  log_with_color(f"Restarting node {args.node}", color='b')
+  return
+
+
+def shutdown_node(args):
+  """
+  This function is used to shutdown the node.
+  
+  Parameters
+  ----------
+  args : argparse.Namespace
+      Arguments passed to the function.
+  """
+  log_with_color(f"Shutting down node {args.node}", color='b')
+  return

{naeural_client-2.3.1 → naeural_client-2.3.2}/naeural_client/const/misc.py

@@ -4,7 +4,7 @@ class COLORS:
   TIMERS = 'd'
   STATUS = 'n'
   SERVING = 'm'
-  BIZ = 'b'
+  BIZ = 'c'
   COMM = 'y'
   TIMERS = 'd'
 

{naeural_client-2.3.1 → naeural_client-2.3.2}/naeural_client/logging/base_logger.py

@@ -35,6 +35,8 @@ COLORS = {
   'y': "\x1b[1;33m",
   'b': "\x1b[1;34m",
   'm': "\x1b[1;35m",
+  'c': "\x1b[1;36m",  # bright cyan
+  'v': "\x1b[0;36m",  # dim cyan  
   'a': "\x1b[41m",
   'e': "\x1b[41m",
   'w': "\x1b[1;31m", # warning == red

{naeural_client-2.3.1 → naeural_client-2.3.2}/naeural_client/utils/config.py

@@ -31,8 +31,9 @@ def log_with_color(s, color='n'):
       'r': '\033[31m',  # Red
       'g': '\033[32m',  # Green
       'y': '\033[33m',  # Yellow
-      'b': '\033[34m',  # Blue
+      'b': "\x1b[1;34m",  # bright blue
       'w': '\033[97m',  # Light white
+      'c': "\x1b[1;36m",  # bright cyan
       'n': '\033[37m',  # Dark white (default)
   }
 
@@ -53,7 +54,7 @@ def get_user_config_file():
   """
   return get_user_folder() / "config"
 
-def reset_config():
+def reset_config(args):
   """
   Resets the configuration by creating a ~/.naeural folder and populating
   ~/.naeural/config with values from a local .env file, if it exists.
@@ -78,7 +79,7 @@ def reset_config():
     log_with_color(f"Configuration has been reset to default in {config_file}:\n{ENV_TEMPLATE}", color='y')
 
 
-def show_config():
+def show_config(args):
   """
   Displays the current configuration from ~/.naeural/config.
   """

{naeural_client-2.3.1 → naeural_client-2.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "naeural_client"
-version = "2.3.1"
+version = "2.3.2"
 authors = [
   { name="Andrei Ionut Damian", email="andrei.damian@me.com" },
   { name="Cristan Bleotiu", email="cristibleotiu@gmail.com" },

naeural_client-2.3.1/naeural_client/cli/cli.py

@@ -1,48 +0,0 @@
-import argparse
-
-from naeural_client.utils.config import maybe_init_config, log_with_color
-from naeural_client.cli.cli_commands import CLI_COMMANDS
-
-from naeural_client import version
-
-def build_parser():
-  """
-  Dynamically builds the argument parser based on CLI_COMMANDS.
-
-  Returns
-  -------
-  argparse.ArgumentParser
-      Configured argument parser.
-  """
-  descr = f"nepctl v{version} - CLI for Naeural Edge Protocol SDK package"
-  
- 
-  parser = argparse.ArgumentParser(description=descr)
-  subparsers = parser.add_subparsers(dest="command", help="Available commands")
-
-  # Iterate over top-level commands
-  for command, subcommands in CLI_COMMANDS.items():
-    command_parser = subparsers.add_parser(command, help=f"{command} commands")
-    if isinstance(subcommands, dict):  # Nested subcommands
-      command_subparsers = command_parser.add_subparsers(dest="subcommand")
-      for subcommand, func in subcommands.items():
-        subcommand_parser = command_subparsers.add_parser(
-            subcommand, help=f"{subcommand} command"
-        )
-        subcommand_parser.set_defaults(func=func)
-    else:
-      command_parser.set_defaults(func=subcommands)
-
-  return parser
-
-def main():
-  maybe_init_config()
-  parser = build_parser()
-  args = parser.parse_args()
-  if hasattr(args, "func"):
-    args.func()  # Call the dynamically loaded function
-  else:
-    parser.print_help()
-
-if __name__ == "__main__":
-  main()

naeural_client-2.3.1/naeural_client/cli/cli_commands.py

@@ -1,15 +0,0 @@
-from naeural_client.cli.nodes import get_nodes, get_supervisors
-from naeural_client.utils.config import show_config, reset_config
-
-
-# Define the available commands
-CLI_COMMANDS = {
-    "get": {
-        "nodes": get_nodes,
-        "supervisors": get_supervisors,
-    },
-    "config": {
-        "show": show_config,
-        "reset": reset_config,
-    },
-}

naeural_client-2.3.1/naeural_client/cli/nodes.py

@@ -1,22 +0,0 @@
-from naeural_client.utils.config import log_with_color
-
-
-def get_nodes():
-  """
-  This function is used to get the information about the nodes and it will perform the following:
-  
-  1. Create a Session object.
-  2. Wait for the first net mon message via Session and show progress. 
-  3. Wait for the second net mon message via Session and show progress.  
-  4. Get the active nodes union via Session and display the nodes marking those peered vs non-peered.
-  """
-  log_with_color("Getting nodes information", color="b")
-  return
-  
-  
-def get_supervisors():
-  """
-  This function is used to get the information about the supervisors.
-  """
-  log_with_color("Getting supervisors information", color='b')
-  return