app wrappers statys and list (#15)

Co-authored-by: Atin Sood <[email protected]>

Author: atinsood

src/codeflare_sdk/cluster/cluster.py

@@ -92,12 +92,15 @@ def list_all_clusters(print_to_console=True):
     return clusters
 
 
-# private methods
-
-def _get_appwrappers(namespace='default'):
-    with oc.project(namespace), oc.timeout(10*60):
-        app_wrappers = oc.selector('appwrappers').qnames()
+def list_all_queued(print_to_console=True):
+    app_wrappers = _get_app_wrappers(filter=[AppWrapperStatus.RUNNING, AppWrapperStatus.PENDING])
+    if print_to_console:
+        pretty_print.print_appwrappers_status(app_wrappers)
     return app_wrappers
+   
+
+
+# private methods
 
 
 def _app_wrapper_status(name, namespace='default') -> Optional[AppWrapper]:
@@ -127,6 +130,22 @@ def _get_ray_clusters(namespace='default') -> List[RayCluster]:
     return list_of_clusters
 
 
+
+def _get_app_wrappers(filter:List[AppWrapperStatus], namespace='default') -> List[AppWrapper]:
+    list_of_app_wrappers = []
+
+    with oc.project(namespace), oc.timeout(10*60):
+        app_wrappers = oc.selector('appwrappers').objects()
+
+    for item in app_wrappers:
+        app_wrapper = _map_to_app_wrapper(item)
+        if filter and app_wrapper.status in filter:
+            list_of_app_wrappers.append(app_wrapper)
+        else:
+            list_of_app_wrappers.append(app_wrapper)
+    return list_of_app_wrappers
+
+
 def _map_to_ray_cluster(cluster) -> RayCluster:
     cluster_model = cluster.model
     return RayCluster(

src/codeflare_sdk/utils/pretty_print.py

@@ -12,8 +12,24 @@ def print_no_resources_found():
     console.print(Panel("[red]No resources found"))
 
 
-def print_appwrappers_status():
-    pass
+def print_appwrappers_status(app_wrappers:List[AppWrapper]):
+    if not app_wrappers == 0:
+        print_no_resources_found()
+        return #shortcircuit
+    
+    console = Console()
+    for app_wrapper in app_wrappers:
+        name = app_wrapper.name
+        status = app_wrapper.status.value
+
+        table = Table(box=None, title="[bold] :rocket: List of CodeFlare clusters in queue:rocket:")
+        table.add_column("Name", style="cyan", no_wrap=True)
+        table.add_column("Status", style="magenta")
+        table.add_row("[bold underline]"+name,status)        
+        table.add_row("") #empty row for spacing
+        console.print(Panel.fit(table))
+
+
 
 def print_clusters(clusters:List[RayCluster], verbose=True):
     if not clusters == 0:
@@ -22,10 +38,6 @@ def print_clusters(clusters:List[RayCluster], verbose=True):
 
     console = Console()
     title_printed = False
-    #FIXME handle case where no clusters are found
-    if len(clusters) == 0:
-        print_no_resources_found()
-        return #exit early
 
     for cluster in clusters:
         status = "Active :white_heavy_check_mark:" if cluster.status == RayClusterStatus.READY else "InActive :x:"

tests/test_clusters.py

@@ -1,4 +1,4 @@
-from codeflare_sdk.cluster.cluster import list_all_clusters, _app_wrapper_status
+from codeflare_sdk.cluster.cluster import list_all_clusters, list_all_queued, _app_wrapper_status
 from codeflare_sdk.cluster.cluster import Cluster, ClusterConfiguration
 
 import time
@@ -26,3 +26,7 @@ def test_cluster_down():
 def test_no_resources_found():
     from codeflare_sdk.utils import pretty_print
     pretty_print.print_no_resources_found()
+
+def test_list_app_wrappers():
+    app_wrappers = list_all_queued()
+