update(k8s): use time_on_duty in export and ui

app/extras/pseudographic.py

@@ -4,6 +4,7 @@ from rich.progress import Progress
 from database import fetch_all
 from kubernetes import client, config
 import requests
+from datetime import datetime, timezone
 
 # Define constants
 THINK_K8S_URL = "http://localhost:8000/think/k8s"
@@ -11,6 +12,21 @@ THINK_K8S_URL = "http://localhost:8000/think/k8s"
 console = Console()
 
 # Helper functions for conversions
+def calculate_time_on_duty(creation_timestamp):
+    """
+    Calculate the time on duty in hours, days, or minutes from the creation timestamp.
+    """
+    now = datetime.now(timezone.utc)
+    delta = now - creation_timestamp
+
+    if delta.days < 1 and delta.seconds < 3600:
+        minutes = delta.seconds // 60
+        return f"{minutes} minutes" if minutes > 1 else "less than a minute"
+    if delta.days < 1:
+        hours = delta.seconds // 3600
+        return f"{hours} hours" if hours > 1 else "1 hour"
+    return f"{delta.days} days" if delta.days > 1 else "1 day"
+
 def convert_cpu_to_cores(cpu):
     if "n" in cpu:
         return round(int(cpu.replace("n", "")) / 1e9, 4)
@@ -39,54 +55,6 @@ def convert_memory_to_gb(memory):
     return float(memory)
 
 # Display tables
-def display_metal_nodes():
-    table = Table(title="🖥️ Metal Nodes", style="bold green")
-    table.add_column("ID", justify="right", style="cyan")
-    table.add_column("Name", style="magenta")
-    table.add_column("Location", style="white")
-    table.add_column("Vendor", style="green")
-    table.add_column("CPU", justify="right", style="yellow")
-    table.add_column("Memory (GB)", justify="right", style="cyan")
-    table.add_column("Storage", style="magenta")
-
-    nodes = fetch_all("metal_nodes")
-    for node in nodes:
-        table.add_row(
-            f"{node[0]}",
-            node[1],
-            node[2],
-            node[3],
-            f"{node[4]}",
-            node[5],
-            node[6]
-        )
-
-    console.print(table)
-
-def display_virtual_machines():
-    table = Table(title="💻 Virtual Machines", style="bold blue")
-    table.add_column("ID", justify="right", style="cyan")
-    table.add_column("Name", style="magenta")
-    table.add_column("Location", style="white")
-    table.add_column("CPU", justify="right", style="yellow")
-    table.add_column("Memory (GB)", justify="right", style="cyan")
-    table.add_column("Storage", style="magenta")
-    table.add_column("Type", style="green")
-
-    vms = fetch_all("virtual_machines")
-    for vm in vms:
-        table.add_row(
-            f"{vm[0]}",
-            vm[1],
-            vm[2],
-            f"{vm[3]}",
-            vm[4],
-            vm[5],
-            vm[6]
-        )
-
-    console.print(table)
-
 def display_kubernetes_nodes():
     config.load_incluster_config()
     v1 = client.CoreV1Api()
@@ -98,11 +66,13 @@ def display_kubernetes_nodes():
     table.add_column("Storage (GB)", justify="right", style="green")
     table.add_column("Instance Type", style="blue")
     table.add_column("Pods Allocatable", justify="right", style="magenta")
+    table.add_column("Time on Duty", justify="right", style="magenta")
 
     nodes = v1.list_node()
     for node in nodes.items:
         ephemeral_storage = node.status.capacity.get("ephemeral-storage", "0")
         instance_type = node.metadata.labels.get("beta.kubernetes.io/instance-type", "N/A")
+        creation_timestamp = node.metadata.creation_timestamp
 
         table.add_row(
             node.metadata.name,
@@ -110,42 +80,8 @@ def display_kubernetes_nodes():
             f"{round(convert_memory_to_mib(node.status.capacity.get('memory')), 2)}",
             f"{round(convert_memory_to_gb(ephemeral_storage), 2)}",
             instance_type,
-            node.status.allocatable.get("pods")
-        )
-
-    console.print(table)
-
-def display_namespace_usage():
-    config.load_incluster_config()
-    metrics_client = client.CustomObjectsApi()
-
-    table = Table(title="📊 Namespace Resource Usage", style="bold magenta")
-    table.add_column("Namespace", style="white")
-    table.add_column("CPU (Cores)", justify="right", style="yellow")
-    table.add_column("Memory (MiB)", justify="right", style="cyan")
-
-    namespace_usage = {}
-    pod_metrics = metrics_client.list_cluster_custom_object(
-        group="metrics.k8s.io", version="v1beta1", plural="pods"
-    )
-
-    for pod in pod_metrics["items"]:
-        namespace = pod["metadata"]["namespace"]
-        if namespace not in namespace_usage:
-            namespace_usage[namespace] = {"cpu": 0, "memory": 0}
-
-        for container in pod["containers"]:
-            cpu_usage = container["usage"]["cpu"]
-            memory_usage = container["usage"]["memory"]
-
-            namespace_usage[namespace]["cpu"] += convert_cpu_to_cores(cpu_usage)
-            namespace_usage[namespace]["memory"] += convert_memory_to_mib(memory_usage)
-
-    for namespace, usage in namespace_usage.items():
-        table.add_row(
-            namespace,
-            f"{round(usage['cpu'], 4)}",
-            f"{round(usage['memory'], 2)}"
+            node.status.allocatable.get("pods"),
+            calculate_time_on_duty(creation_timestamp) if creation_timestamp else "N/A"
         )
 
     console.print(table)