From f04063f2f9c20523a5ab1349e331eb53a51efc87 Mon Sep 17 00:00:00 2001
From: Franck Cuny <fcuny@roblox.com>
Date: Fri, 4 Oct 2024 08:00:34 -0700
Subject: add more tools

---
 src/cli/nomad_allocs.py | 47 -----------------------------------------------
 1 file changed, 47 deletions(-)
 delete mode 100755 src/cli/nomad_allocs.py

(limited to 'src/cli/nomad_allocs.py')

diff --git a/src/cli/nomad_allocs.py b/src/cli/nomad_allocs.py
deleted file mode 100755
index cc7e648..0000000
--- a/src/cli/nomad_allocs.py
+++ /dev/null
@@ -1,47 +0,0 @@
-#!/usr/bin/env python3
-
-import sys
-
-import click
-import requests
-
-
-@click.command()
-@click.argument("job")
-@click.option("--dc", default="bom1", help="Name of the data center")
-@click.option("--token", help="Token for Nomad")
-def cli(job, dc, token):
-    if token is None:
-        print("you need to pass a valid token")
-        sys.exit(1)
-
-    headers = {"Authorization": f"Bearer {token}"}
-    url = f"https://{dc}-nomad.simulprod.com/v1/job/{job}/allocations"
-    try:
-        resp = requests.get(url, headers=headers)
-        resp.raise_for_status()
-    except Exception as e:
-        print("return {}".format(str(e)))
-
-    running_tasks = []
-    terminated_tasks = []
-    for task in resp.json():
-        task_name = list(task["TaskStates"].keys())[0]
-        if task["TaskStates"][task_name]["State"] == "running":
-            running_tasks.append(f"https://{dc}-nomad.simulprod.com/ui/allocations/{task['ID']}/{task_name}/logs")
-        else:
-            terminated_tasks.append(f"https://{dc}-nomad.simulprod.com/ui/allocations/{task['ID']}/{task_name}/logs")
-
-    if len(running_tasks) > 0:
-        print("running tasks")
-        for t in running_tasks:
-            print(f"→ {t}")
-
-    if len(terminated_tasks) > 0:
-        print("terminated tasks")
-        for t in terminated_tasks:
-            print(f"→ {t}")
-
-
-if __name__ == "__main__":
-    cli()
-- 
cgit v1.2.3