Refactor status command to use a table argument for customizable output

Author: hvdbm

cluv/__main__.py

@@ -139,20 +139,15 @@ def add_submit_args(
 def add_status_args(subparsers: Subparsers) -> argparse.ArgumentParser:
     status_parser = subparsers.add_parser(
         "status",
-        help="Get the status of available clusters.",
+        help="Get the status of clusters and jobs.",
         formatter_class=rich_argparse.RichHelpFormatter,
     )
     status_parser.add_argument(
-        "clusters",
-        nargs="*",
-        default=None,
-        metavar="<cluster>",
-        help=("Cluster(s) to query. Leave empty to query all clusters with an active connection."),
-    )
-    status_parser.add_argument(
-        "--show",
+        "table",
+        nargs="?",
         choices=["clusters", "jobs", "all"],
         default="all",
+        metavar="<table>",
         help="Which table to display: cluster overview, jobs overview, or both (default: all).",
     )
     status_parser.set_defaults(func=status)

cluv/cli/status.py

@@ -522,12 +522,13 @@ def _build_legend() -> Panel:
     return Panel(legend, title="Legend", border_style="dim", padding=(0, 1))
 
 
-async def status(clusters: list[str] | None = None, show: str = "all"):
+async def status(table: str) -> None:
     """Gets the status of available clusters.
     - Gives you an overview of the state of each cluster, and displays an overview of the state of your jobs across the clusters.
     - Displays the number of idle nodes, or the number of idle GPUs, or something similar, for each cluster
     """
     console = Console()
+    clusters = get_config().clusters_names
     clusters = list(clusters or [])
 
     if clusters:
@@ -559,10 +560,10 @@ async def status(clusters: list[str] | None = None, show: str = "all"):
     console.rule(f"[bold cyan]cluv status[/bold cyan]  {label}")
     console.print()
 
-    if show in ("clusters", "all"):
+    if table in ("clusters", "all"):
         console.print(_build_cluster_table(data))
         console.print(_build_legend())
         console.print()
-    if show in ("jobs", "all"):
+    if table in ("jobs", "all"):
         console.print(_build_my_jobs_table(data))
         console.print()