Simplify and colorize images replicate table (#659)

Author: yec-akamai

linodecli/overrides.py

@@ -11,11 +11,21 @@
 from rich.align import Align
 from rich.console import Console
 from rich.table import Table
+from rich.theme import Theme
 
 from linodecli.output.output_handler import OutputMode
 
 OUTPUT_OVERRIDES = {}
 
+REPLICA_STATUS_THEME = {
+    "available": "bright_green",
+    "creating": "bright_yellow",
+    "pending": "yellow",
+    "pending replication": "yellow",
+    "pending deletion": "red",
+    "replicating": "bright_yellow",
+}
+
 
 def output_override(command: str, action: str, output_mode: OutputMode):
     """
@@ -157,7 +167,12 @@ def build_replicas_output(replicas: List) -> Table:
     for replica in replicas:
         row = []
         for h in replicas_headers:
-            row.append(Align(str(replica[h]), align="left"))
+            if h == "status" and replica[h] in REPLICA_STATUS_THEME:
+                row.append(
+                    Align(str(replica[h]), align="left", style=replica[h])
+                )
+            else:
+                row.append(Align(str(replica[h]), align="left"))
         replicas_output.add_row(*row)
 
     return replicas_output
@@ -167,24 +182,27 @@ def image_replicate_output(json_data) -> bool:
     """
     Parse and format the image replicate output table.
     """
+    console = Console(theme=Theme(REPLICA_STATUS_THEME))
+
     output = Table(
         header_style="bold",
         show_lines=True,
     )
 
     row = []
-    for header in json_data.keys():
-        if header == "regions" and len(json_data[header]) > 0:
-            # leverage `replicas` in output for readability
-            output.add_column("replicas", justify="center")
-            row.append(build_replicas_output(json_data[header]))
-        elif json_data[header] is not None:
-            output.add_column(header, justify="center")
-            row.append(Align(str(json_data[header]), align="left"))
+    headers = ["id", "label", "status", "total_size", "regions"]
+    for header in headers:
+        if header in json_data:
+            if header == "regions" and len(json_data[header]) > 0:
+                # leverage `replicas` in output for readability
+                output.add_column("replicas", justify="center")
+                row.append(build_replicas_output(json_data[header]))
+            elif json_data[header] is not None:
+                output.add_column(header, justify="center")
+                row.append(Align(str(json_data[header]), align="left"))
 
     output.add_row(*row)
 
-    console = Console()
     console.print(output)
 
     return False