diff options
Diffstat (limited to 'arthur/exts/kubernetes/pods.py')
| -rw-r--r-- | arthur/exts/kubernetes/pods.py | 27 |
1 files changed, 17 insertions, 10 deletions
diff --git a/arthur/exts/kubernetes/pods.py b/arthur/exts/kubernetes/pods.py index 8ecaf50..f69e77e 100644 --- a/arthur/exts/kubernetes/pods.py +++ b/arthur/exts/kubernetes/pods.py @@ -1,7 +1,9 @@ """The Pods cog helps with managing Kubernetes pods.""" -from textwrap import dedent +import zoneinfo +from datetime import datetime +import humanize from discord.ext import commands from tabulate import tabulate @@ -9,23 +11,19 @@ from arthur.apis.kubernetes import pods from arthur.bot import KingArthur from arthur.utils import generate_error_message -MAX_MESSAGE_LENGTH = 4000 +MAX_MESSAGE_LENGTH = 2000 def tabulate_pod_data(data: list[list[str]]) -> str: """Tabulate the pod data to be sent to Discord.""" table = tabulate( data, - headers=["Status", "Pod", "Phase", "IP", "Node"], + headers=["Status", "Pod", "Phase", "IP", "Node", "Age", "Restarts"], tablefmt="psql", - colalign=("center", "left", "center", "center", "center"), + colalign=("center", "left", "left", "center", "center", "left", "center"), ) - return dedent(f""" - ``` - {table} - ``` - """) + return f"```\n{table}```" class Pods(commands.Cog): @@ -66,12 +64,21 @@ class Pods(commands.Cog): case _: emote = "\N{BLACK QUESTION MARK ORNAMENT}" + time_human = humanize.naturaldelta( + datetime.now(tz=zoneinfo.ZoneInfo("UTC")) - pod.metadata.creation_timestamp + ) + + # we know that Linode formats names like "lke<cluster>-<pool>-<node>" + node_name = pod.spec.node_name.split("-")[2] + table_data = [ emote, pod.metadata.name, pod.status.phase, pod.status.pod_ip, - pod.spec.node_name, + node_name, + time_human, + pod.status.container_statuses[0].restart_count, ] if len(tabulate_pod_data(tables[-1] + [table_data])) > MAX_MESSAGE_LENGTH: |