yellowdog-python-examples 8.1.6__tar.gz → 8.1.7__tar.gz

{yellowdog_python_examples-8.1.6/yellowdog_python_examples.egg-info → yellowdog_python_examples-8.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yellowdog-python-examples
-Version: 8.1.6
+Version: 8.1.7
 Summary: Python CLI commands using the YellowDog Python SDK
 Author-email: YellowDog Limited <support@yellowdog.co>
 License-Expression: Apache-2.0

yellowdog_python_examples-8.1.7/yellowdog_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "8.1.7"

{yellowdog_python_examples-8.1.6 → yellowdog_python_examples-8.1.7}/yellowdog_cli/shutdown.py

@@ -16,11 +16,12 @@ from yellowdog_client.model import (
 from yellowdog_cli.utils.entity_utils import (
     get_worker_pool_by_id,
     get_worker_pool_id_by_name,
+    get_worker_pools,
 )
 from yellowdog_cli.utils.follow_utils import follow_ids
 from yellowdog_cli.utils.interactive import confirmed, select
 from yellowdog_cli.utils.misc_utils import link_entity
-from yellowdog_cli.utils.printing import print_error, print_log
+from yellowdog_cli.utils.printing import print_error, print_log, print_warning
 from yellowdog_cli.utils.wrapper import ARGS_PARSER, CLIENT, CONFIG_COMMON, main_wrapper
 from yellowdog_cli.utils.ydid_utils import YDIDType, get_ydid_type
 
@@ -37,9 +38,10 @@ def main():
         f"names including '{CONFIG_COMMON.name_tag}'"
     )
 
-    worker_pool_summaries: List[WorkerPoolSummary] = (
-        CLIENT.worker_pool_client.find_all_worker_pools()
+    worker_pool_summaries: List[WorkerPoolSummary] = get_worker_pools(
+        CLIENT, CONFIG_COMMON.namespace, CONFIG_COMMON.name_tag
     )
+
     shutdown_count = 0
 
     selected_worker_pool_summaries: List[WorkerPoolSummary] = []
@@ -73,7 +75,7 @@ def main():
                 print_log(f"Shut down {link_entity(CONFIG_COMMON.url, worker_pool)}")
                 optionally_terminate_compute_requirement(worker_pool_summary.id)
             except Exception as e:
-                print_error(f"Unable to shut down '{worker_pool_summary.name}': {e}")
+                print_error(f"Failed to shut down '{worker_pool_summary.name}': {e}")
 
     if shutdown_count > 0:
         print_log(f"Shut down {shutdown_count} Worker Pool(s)")
@@ -101,7 +103,7 @@ def shutdown_by_names_or_ids(names_or_ids: List[str]):
         else:
             worker_pool_id = get_worker_pool_id_by_name(CLIENT, name_or_id)
             if worker_pool_id is None:
-                print_error(f"Worker Pool '{name_or_id}' not found")
+                print_warning(f"Worker Pool '{name_or_id}' not found")
                 continue
         worker_pool_ids.append(worker_pool_id)
 
@@ -120,14 +122,14 @@ def shutdown_by_names_or_ids(names_or_ids: List[str]):
             print_log(f"Shut down Worker Pool '{worker_pool_id}'")
             optionally_terminate_compute_requirement(worker_pool_id)
         except Exception as e:
-            print_error(f"Unable to shut down Worker Pool '{worker_pool_id}': ({e})")
+            print_error(f"Failed to shut down Worker Pool '{worker_pool_id}': ({e})")
 
     for node_id in node_ids:
         try:
             CLIENT.worker_pool_client.shutdown_node_by_id(node_id)
             print_log(f"Shut down Node '{node_id}'")
         except Exception as e:
-            print_error(f"Unable to shut down Node '{node_id}': ({e})")
+            print_error(f"Failed to to shut down Node '{node_id}': ({e})")
 
     if ARGS_PARSER.follow:
         follow_ids(worker_pool_ids, auto_cr=ARGS_PARSER.auto_cr)
@@ -151,7 +153,7 @@ def optionally_terminate_compute_requirement(worker_pool_id: str):
             f"Terminated associated Compute Requirement '{worker_pool.computeRequirementId}'"
         )
     except Exception as e:
-        print_error(f"Error terminating Compute Requirement: ({e})")
+        print_error(f"Failed to terminate Compute Requirement: ({e})")
 
 
 # Entry point

{yellowdog_python_examples-8.1.6 → yellowdog_python_examples-8.1.7}/yellowdog_cli/terminate.py

@@ -77,7 +77,7 @@ def main():
                 )
             except Exception as e:
                 print_error(
-                    f"Unable to terminate '{compute_requirement_summary.name}': {e}"
+                    f"Failed to terminate '{compute_requirement_summary.name}': {e}"
                 )
 
     if terminated_count > 0:
@@ -125,7 +125,7 @@ def terminate_cr_by_name_or_id(names_or_ids: List[str]):
             )
             print_log(f"Terminated '{compute_requirement_id}'")
         except Exception as e:
-            print_error(f"Unable to terminate '{compute_requirement_id}': ({e})")
+            print_error(f"Failed to terminate '{compute_requirement_id}': ({e})")
 
     if ARGS_PARSER.follow:
         follow_ids(compute_requirement_ids)

{yellowdog_python_examples-8.1.6 → yellowdog_python_examples-8.1.7/yellowdog_python_examples.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: yellowdog-python-examples
-Version: 8.1.6
+Version: 8.1.7
 Summary: Python CLI commands using the YellowDog Python SDK
 Author-email: YellowDog Limited <support@yellowdog.co>
 License-Expression: Apache-2.0

yellowdog_python_examples-8.1.6/yellowdog_cli/__init__.py

@@ -1 +0,0 @@
-__version__ = "8.1.6"