oak-cli 0.3.1__tar.gz → 0.3.2__tar.gz

{oak_cli-0.3.1 → oak_cli-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oak_cli
-Version: 0.3.1
+Version: 0.3.2
 Summary: 
 Author: Alexander Malyuk
 Author-email: malyuk.alexander1999@gmail.com

{oak_cli-0.3.1 → oak_cli-0.3.2}/oak_cli/worker/common.py

@@ -18,16 +18,21 @@ def get_process_status(
         shell_cmd=f"ps -aux | grep '{process_cmd}'",
         pure_shell=True,
         text=True,
+        check=False,
     )
-    processes = [line for line in cmd_res.stdout.split("\n") if line != ""]
-    message = ""
-    if process_name:
-        message = f"{process_name}: "
-    # NOTE: The grep cmd and the python subprocess call count as 2 processes in the list.
-    if len(processes) > 2:
-        status = ProcessStatus.RUNNING
-    else:
+    # NOTE: Sometimes flaky errors occur.
+    if cmd_res.returncode != 0:
         status = ProcessStatus.OFFLINE
+    else:
+        processes = [line for line in cmd_res.stdout.split("\n") if line != ""]
+        message = ""
+        if process_name:
+            message = f"{process_name}: "
+        # NOTE: The grep cmd and the python subprocess call count as 2 processes in the list.
+        if len(processes) > 2:
+            status = ProcessStatus.RUNNING
+        else:
+            status = ProcessStatus.OFFLINE
 
     if print_status:
         logger.info(message + status.value)

{oak_cli-0.3.1 → oak_cli-0.3.2}/oak_cli/worker/net_manager.py

@@ -4,6 +4,7 @@ import typer
 from typing_extensions import Annotated
 
 from oak_cli.utils.common import run_in_shell
+from oak_cli.utils.logging import logger
 from oak_cli.utils.typer_augmentations import AliasGroup
 from oak_cli.worker.common import ProcessStatus, get_process_status, stop_process
 
@@ -18,7 +19,8 @@ NET_MANAGER_CMD_PREFIX = f"sudo {NET_MANAGER_NAME}"
 def start_net_manager(
     use_debug_mode: Annotated[Optional[bool], typer.Option("-D")] = False,
 ) -> None:
-    if get_net_manager_status() == ProcessStatus.RUNNING:
+    if get_net_manager_status(print_status=False) == ProcessStatus.RUNNING:
+        logger.info("The NetManager is already running.")
         return
 
     cmd = f"{NET_MANAGER_CMD_PREFIX} -p 6000"
@@ -28,11 +30,11 @@ def start_net_manager(
 
 
 @app.command("status", help=f"Show the status of the {NET_MANAGER_NAME}.")
-def get_net_manager_status() -> ProcessStatus:
+def get_net_manager_status(print_status: bool = True) -> ProcessStatus:
     return get_process_status(
         process_cmd=NET_MANAGER_CMD_PREFIX,
         process_name=NET_MANAGER_NAME,
-        print_status=True,
+        print_status=print_status,
     )
 
 

{oak_cli-0.3.1 → oak_cli-0.3.2}/oak_cli/worker/node_engine.py

@@ -4,6 +4,7 @@ import typer
 from typing_extensions import Annotated
 
 from oak_cli.utils.common import get_env_var, run_in_shell
+from oak_cli.utils.logging import logger
 from oak_cli.utils.typer_augmentations import AliasGroup
 from oak_cli.worker.common import ProcessStatus, get_process_status, stop_process
 
@@ -20,7 +21,8 @@ def start_node_engine(
         Optional[bool], typer.Option("--ml_data_server_for_flops_addon_learner")
     ] = False,
 ) -> None:
-    if get_node_engine_status() == ProcessStatus.RUNNING:
+    if get_node_engine_status(print_status=False) == ProcessStatus.RUNNING:
+        logger.info("The NodeEngine is already running.")
         return
 
     cmd = f"{NODE_ENGINE_CMD_PREFIX} -p 6000 -p 10100 -a {get_env_var(name='SYSTEM_MANAGER_URL')}"
@@ -30,11 +32,11 @@ def start_node_engine(
 
 
 @app.command("status", help=f"Show the status of the {NODE_ENGINE_NAME}.")
-def get_node_engine_status() -> ProcessStatus:
+def get_node_engine_status(print_status: bool = True) -> ProcessStatus:
     return get_process_status(
         process_cmd=NODE_ENGINE_CMD_PREFIX,
         process_name=NODE_ENGINE_NAME,
-        print_status=True,
+        print_status=print_status,
     )
 
 

{oak_cli-0.3.1 → oak_cli-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "oak_cli"
-version = "0.3.1"
+version = "0.3.2"
 description = ""
 authors = ["Alexander Malyuk <malyuk.alexander1999@gmail.com>"]
 readme = "README.md"