* feat(manage_user_jobs.py): add support for displaying job status in…

… the table

kubejobs/manage_user_jobs.py

@@ -1,11 +1,14 @@
 import json
 import subprocess
 from datetime import datetime, timezone
+from typing import Dict, Optional
 
 import fire
 import rich
+from rich import print
 from rich.console import Console
 from rich.table import Table
+from tqdm import tqdm
 
 console = Console()
 
@@ -41,9 +44,74 @@ def highlight_term(text: str, term: str, style: str = "[bold cyan]") -> str:
     return text.replace(term, f"{style}{term}[/]")
 
 
+def get_job_status(job_name: str, namespace: str, job_dict: Dict) -> str:
+    # "status": {
+    #     "completionTime": "2023-10-23T17:45:25Z",
+    #     "conditions": [
+    #         {
+    #             "lastProbeTime": "2023-10-23T17:45:25Z",
+    #             "lastTransitionTime": "2023-10-23T17:45:25Z",
+    #             "status": "True",
+    #             "type": "Complete"
+    #         }
+    #     ],
+    #     "failed": 1,
+    #     "ready": 0,
+    #     "startTime": "2023-10-23T16:38:56Z",
+    #     "succeeded": 1,
+    #     "uncountedTerminatedPods": {}
+    # }
+
+    # "status": {
+    #     "conditions": [
+    #         {
+    #             "lastProbeTime": "2023-10-23T18:01:12Z",
+    #             "lastTransitionTime": "2023-10-23T18:01:12Z",
+    #             "message": "Job has reached the specified backoff limit",
+    #             "reason": "BackoffLimitExceeded",
+    #             "status": "True",
+    #             "type": "Failed"
+    #         }
+    #     ],
+    #     "failed": 5,
+    #     "ready": 0,
+    #     "startTime": "2023-10-23T17:28:00Z",
+    #     "uncountedTerminatedPods": {}
+    # }
+
+    status_dict = job_dict["status"]
+    failed_count = status_dict.get("failed", 0)
+    succeeded_count = status_dict.get("succeeded", 0)
+    conditions = status_dict.get("conditions", [])
+    ready = status_dict.get("ready", 0)
+    try:
+        output_dict = {
+            "failed": failed_count,
+            "succeeded": succeeded_count,
+            "condition": conditions[-1]["type"] if conditions else "--",
+            "reason": conditions[-1]["reason"]
+            if (conditions and hasattr(conditions[-1], "reason"))
+            else "--",
+            "ready": ready,
+        }
+    except Exception as e:
+        print(e)
+        print(conditions)
+
+        raise e
+
+    output_dict = {k: str(v) for k, v in output_dict.items()}
+
+    return output_dict
+
+
 # 🕵️‍♀️ List or delete Kubernetes jobs by user
 def list_or_delete_jobs_by_user(
-    namespace: str, username: str, term: str, delete: bool = False
+    namespace: str,
+    username: str,
+    term: str,
+    delete: bool = False,
+    show_job_status: bool = False,
 ) -> None:
     # Fetch all jobs from a specific Kubernetes namespace
     get_jobs_cmd = f"kubectl get jobs -n {namespace} -o json"
@@ -54,25 +122,61 @@ def list_or_delete_jobs_by_user(
     current_time = datetime.now(timezone.utc)
 
     # Initialize table for displaying job details
+    # BOXES = [
+    #     "ASCII",
+    #     "ASCII2",
+    #     "ASCII_DOUBLE_HEAD",
+    #     "SQUARE",
+    #     "SQUARE_DOUBLE_HEAD",
+    #     "MINIMAL",
+    #     "MINIMAL_HEAVY_HEAD",
+    #     "MINIMAL_DOUBLE_HEAD",
+    #     "SIMPLE",
+    #     "SIMPLE_HEAD",
+    #     "SIMPLE_HEAVY",
+    #     "HORIZONTALS",
+    #     "ROUNDED",
+    #     "HEAVY",
+    #     "HEAVY_EDGE",
+    #     "HEAVY_HEAD",
+    #     "DOUBLE",
+    #     "DOUBLE_EDGE",
+    #     "MARKDOWN",
+    # ]
     table = Table(
         title="🔍 Kubernetes Jobs",
         show_header=True,
         header_style="bold magenta",
+        box=rich.box.SQUARE,
     )
-    table.add_column("📛 Name", width=50)
-    table.add_column("✅ Completions", justify="right")
+    table.add_column("📛 Name", width=75)
     table.add_column("⏳ Duration", justify="right")
     table.add_column("⌛ Age", justify="right")
+    table.add_column("✅ Completions", justify="right")
+    if show_job_status:
+        table.add_column("💥 Failed", justify="right")
+        table.add_column("🏆 Succeeded", justify="right")
+        table.add_column("👀 Condition", justify="right")
+        table.add_column("🧐 Reason", justify="right")
+        table.add_column("🔖 Ready", justify="right")
 
     # Filter jobs and fill in the table
-    for item in jobs_json["items"]:
+    for item in tqdm(jobs_json["items"]):
         annotations = item["metadata"].get("annotations", {})
         if (
             annotations.get("user", "") == username
             and term in item["metadata"]["name"]
         ):
             filtered_jobs.append(item)
-            add_row_to_table(item, current_time, table, term)
+
+            if show_job_status:
+                job_status = get_job_status(
+                    item["metadata"]["name"], namespace, item
+                )
+                add_row_to_table(item, current_time, table, term, job_status)
+
+            else:
+                add_row_to_table(item, current_time, table, term)
 
     # Display the table
     console.print(table)
@@ -84,29 +188,67 @@ def list_or_delete_jobs_by_user(
 
 # 📝 Add a row to the table with job details
 def add_row_to_table(
-    item: dict, current_time: datetime, table: Table, term: str
+    item: dict,
+    current_time: datetime,
+    table: Table,
+    term: str,
+    job_status: Optional[Dict] = None,
 ) -> None:
     job_name = item["metadata"]["name"]
     job_name = highlight_term(job_name, term)
     completions = (
         f"{item['status'].get('succeeded', 0)}/{item['spec']['completions']}"
     )
-    completion_style = "bold green" if completions == "1/1" else "red"
-    completion_text = (
-        f"[{completion_style}]{completions}✔ Completed[/{completion_style}]"
-        if completions == "1/1"
-        else f"[{completion_style}]{completions}|❌ Incomplete[/{completion_style}]"
-    )
 
+    if job_status:
+        if job_status["condition"] == "Failed":
+            row_style = "red"
+        elif completions == "1/1":
+            row_style = "bold green"
+        else:
+            row_style = "yellow"
+    else:
+        row_style = "yellow" if completions != "1/1" else "bold green"
+
+    # Determine completion text based on the job_status
+    if job_status:
+        completion_text = (
+            f"{completions}✔ Completed"
+            if completions == "1/1"
+            else f"{completions}|❌ Failed"
+            if job_status["condition"] == "Failed"
+            else f"{completions}|⏳ Running"
+        )
+    else:
+        completion_text = (
+            f"{completions}|⏳ Running"
+            if completions != "1/1"
+            else f"{completions}|✔ Completed"
+        )
+
+    # Get other job details
     creation_time = parse_iso_time(item["metadata"]["creationTimestamp"])
     age = time_diff_to_human_readable(creation_time, current_time)
     duration = get_job_duration(item)
-    table.add_row(
-        job_name,
-        completion_text,
-        duration,
-        age,
-    )
+
+    # Add row with determined style
+    if job_status:
+        table.add_row(
+            job_name,
+            duration,
+            age,
+            completion_text,
+            *list(job_status.values()),
+            style=row_style,
+        )
+    else:
+        table.add_row(
+            job_name,
+            duration,
+            age,
+            completion_text,
+            style=row_style,
+        )
 
 
 # ⏱ Calculate job duration if available