From 56790602c66277dae8dd76bb19c82ef11d51ed8a Mon Sep 17 00:00:00 2001
From: Arisu Tachibana <arisu.tachibana@miraclelinux.com>
Date: Thu, 3 Oct 2024 17:45:50 +0900
Subject: [PATCH] kci-dev/results: Get results list

Signed-off-by: Arisu Tachibana <arisu.tachibana@miraclelinux.com>
---
 kci-dev/subcommands/results.py | 33 +++++++++++++++++++++++++++++----
 1 file changed, 29 insertions(+), 4 deletions(-)

diff --git a/kci-dev/subcommands/results.py b/kci-dev/subcommands/results.py
index 960c01c..e5e5e1b 100644
--- a/kci-dev/subcommands/results.py
+++ b/kci-dev/subcommands/results.py
@@ -23,21 +23,46 @@ def get_node(url, nodeid):
     click.secho(url)
     response = requests.get(url, headers=headers)
     click.secho(response.status_code, fg="green")
-    click.secho(pprint.pprint(response.json()), fg="green")
+    try:
+        click.secho(pprint.pprint(response.json()), fg="green")
+    except:
+        click.secho(pprint.pprint(response.text), fg="green")
+
+
+def get_nodes(url):
+    headers = {
+        "Content-Type": "application/json; charset=utf-8",
+    }
+    url = url + "/nodes"
+    click.secho(url)
+    response = requests.get(url, headers=headers)
+    click.secho(response.status_code, fg="green")
+    try:
+        click.secho(pprint.pprint(response.json()), fg="green")
+    except:
+        click.secho(pprint.pprint(response.text), fg="green")
 
 
 @click.command(help="Get results")
 @click.option(
     "--nodeid",
-    required=True,
+    required=False,
     help="select node_id",
 )
+@click.option(
+    "--nodes",
+    required=False,
+    help="Get last nodes results",
+)
 @click.pass_context
-def results(ctx, nodeid):
+def results(ctx, nodeid, nodes):
     config = ctx.obj.get("CFG")
     instance = ctx.obj.get("INSTANCE")
     url = api_connection(config[instance]["host"])
-    get_node(url, nodeid)
+    if nodeid:
+        get_node(url, nodeid)
+    if nodes:
+        get_nodes(url)
 
 
 if __name__ == "__main__":