Session get now supports multiple ids

aneoconsulting · Jan 23, 2025 · 5967b6a · 5967b6a
1 parent ba4c347
commit 5967b6a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/src/armonik_cli/commands/sessions.py b/src/armonik_cli/commands/sessions.py
@@ -89,15 +89,18 @@ def list(
 
 
 @sessions.command(name="get")
-@session_argument
+@click.argument("session-ids", required=True, type=str, nargs=-1)
 @base_command
-def session_get(endpoint: str, output: str, session_id: str, debug: bool) -> None:
+def session_get(endpoint: str, output: str, session_ids: List[str], debug: bool) -> None:
     """Get details of a given session."""
     with grpc.insecure_channel(endpoint) as channel:
         sessions_client = ArmoniKSessions(channel)
-        session = sessions_client.get_session(session_id=session_id)
-        session = _clean_up_status(session)
-        console.formatted_print(session, format=output, table_cols=SESSION_TABLE_COLS)
+        sessions = []
+        for session_id in session_ids:
+            session = sessions_client.get_session(session_id=session_id)
+            session = _clean_up_status(session)
+            sessions.append(session)
+        console.formatted_print(sessions, format=output, table_cols=SESSION_TABLE_COLS)
 
 
 @sessions.command(name="create")

diff --git a/tests/commands/test_sessions.py b/tests/commands/test_sessions.py
@@ -87,7 +87,7 @@ def test_session_list(mocker, cmd):
 def test_session_get(mocker, cmd):
     mocker.patch.object(ArmoniKSessions, "get_session", return_value=deepcopy(raw_session))
     result = run_cmd_and_assert_exit_code(cmd)
-    assert reformat_cmd_output(result.output, deserialize=True) == serialized_session
+    assert reformat_cmd_output(result.output, deserialize=True) == [serialized_session]
 
 
 @pytest.mark.parametrize(