Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 20 additions & 1 deletion src/endpoints_submission_cli/commands/submissions/get.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from __future__ import annotations

import sys
from pathlib import Path

import click

Expand All @@ -18,14 +19,22 @@

@click.command("get")
@click.option("--submission-id", required=True, help="Submission UUID.")
@click.option(
"--download-to",
default=None,
type=click.Path(file_okay=False),
help="Download the submission archive (.tar.gz) to this directory.",
)
@click.option(
"--token",
envvar="PRISM_USER_API_TOKEN",
default=None,
help="PRISM API key (mlc_...).",
)
@click.option("-j", "--json", "as_json", is_flag=True, default=False, help="Output raw JSON.")
def submissions_get(submission_id: str, token: str | None, as_json: bool) -> None:
def submissions_get(
submission_id: str, download_to: str | None, token: str | None, as_json: bool
) -> None:
"""Get full submission details including embedded runs."""
resolved_token = _get_token(token)
try:
Expand All @@ -38,3 +47,13 @@ def submissions_get(submission_id: str, token: str | None, as_json: bool) -> Non
output_json(sub)
else:
print_submission_detail(sub)

if download_to is not None:
try:
dest = subs_api.download_submission_archive(
resolved_token, submission_id, Path(download_to)
)
except APIError as exc:
_console.print(f"[bold red]Error:[/bold red] {exc}")
sys.exit(1)
_console.print(f"Archive saved to [bold]{dest}[/bold]")
Original file line number Diff line number Diff line change
Expand Up @@ -101,6 +101,55 @@ def test_get_api_error_exits_1(self) -> None:
)
assert result.exit_code == 1

def test_get_download_to_saves_archive(self, tmp_path: Path) -> None:
archive = tmp_path / f"{SUBMISSION_ID}.tar.gz"
with patch(
"endpoints_submission_cli.submissions.api.get_submission", return_value=SUBMISSION_OUT
):
with patch(
"endpoints_submission_cli.submissions.api.download_submission_archive",
return_value=archive,
) as mock_dl:
with patch("endpoints_submission_cli._http.get_token", return_value=TOKEN):
result = _runner.invoke(
app,
[
"submissions",
"get",
"--submission-id",
SUBMISSION_ID,
"--download-to",
str(tmp_path),
*_TOKEN_ARGS,
],
)
assert result.exit_code == 0
mock_dl.assert_called_once_with(TOKEN, SUBMISSION_ID, tmp_path)
assert "Archive saved to" in result.output

def test_get_download_api_error_exits_1(self, tmp_path: Path) -> None:
with patch(
"endpoints_submission_cli.submissions.api.get_submission", return_value=SUBMISSION_OUT
):
with patch(
"endpoints_submission_cli.submissions.api.download_submission_archive",
side_effect=APIError("download failed"),
):
with patch("endpoints_submission_cli._http.get_token", return_value=TOKEN):
result = _runner.invoke(
app,
[
"submissions",
"get",
"--submission-id",
SUBMISSION_ID,
"--download-to",
str(tmp_path),
*_TOKEN_ARGS,
],
)
assert result.exit_code == 1


@pytest.mark.unit
class TestSubmissionsUpdate:
Expand Down
Loading