dayhoff-tools 1.14.11__py3-none-any.whl → 1.14.12__py3-none-any.whl

dayhoff_tools/cli/batch/commands/status.py

@@ -174,15 +174,41 @@ def _show_job_list(user, status_filter, pipeline, base_path):
     click.echo("Use 'dh batch status <job-id>' for details.")
 
 
+def _parse_retry_job_id(job_id: str) -> tuple[str, str | None]:
+    """Parse a job ID to extract parent job ID and retry suffix.
+
+    Args:
+        job_id: Job ID like 'dma-embed-20260120-63ec' or 'dma-embed-20260120-63ec-r1'
+
+    Returns:
+        Tuple of (parent_job_id, retry_id or None)
+    """
+    import re
+
+    # Check for retry suffix like -r1, -r2, etc.
+    match = re.match(r"^(.+)(-r\d+)$", job_id)
+    if match:
+        return match.group(1), job_id
+    return job_id, None
+
+
 def _show_job_details(job_id: str, base_path: str):
     """Show detailed status for a specific job."""
+    # Check if this is a retry job ID
+    parent_job_id, retry_id = _parse_retry_job_id(job_id)
+
     try:
-        manifest = load_manifest(job_id, base_path)
+        manifest = load_manifest(parent_job_id, base_path)
     except FileNotFoundError:
         click.echo(f"Job not found: {job_id}", err=True)
-        click.echo(f"Looking in: {base_path}/{job_id}/manifest.json", err=True)
+        click.echo(f"Looking in: {base_path}/{parent_job_id}/manifest.json", err=True)
         raise SystemExit(1)
 
+    # If showing a retry job, show retry-specific details
+    if retry_id:
+        _show_retry_details(manifest, retry_id)
+        return
+
     click.echo()
     click.echo(f"Job ID:    {manifest.job_id}")
     click.echo(f"Status:    {format_status(manifest.status)}")
@@ -235,7 +261,31 @@ def _show_job_details(job_id: str, base_path: str):
         click.echo()
         click.echo("Retries:")
         for retry in manifest.retries:
-            click.echo(f"  - {retry.retry_id}: indices {retry.indices}")
+            reslice_info = ""
+            if retry.reslice_prefix:
+                reslice_info = f" (resliced to {retry.reslice_count} chunks)"
+            click.echo(f"  - {retry.retry_id}: {len(retry.indices)} indices{reslice_info}")
+            click.echo(f"    Indices: {retry.indices}")
+            if retry.batch_job_id:
+                # Show brief status for retry job
+                try:
+                    client = BatchClient()
+                    array_status = client.get_array_job_status(retry.batch_job_id)
+                    if array_status.is_complete:
+                        pct = array_status.success_rate * 100
+                        color = "green" if pct == 100 else "yellow" if pct > 90 else "red"
+                        click.echo(
+                            f"    Status: Complete - {click.style(f'{pct:.0f}%', fg=color)} "
+                            f"({array_status.succeeded}/{array_status.total} succeeded)"
+                        )
+                    else:
+                        click.echo(
+                            f"    Status: Running - {array_status.succeeded}/{array_status.total} done, "
+                            f"{array_status.running} running"
+                        )
+                except BatchError:
+                    click.echo(f"    Status: (could not fetch)")
+            click.echo(f"    Details: dh batch status {retry.retry_id}")
 
     # Suggest next steps
     click.echo()
@@ -285,3 +335,48 @@ def _show_array_status(batch_job_id: str):
 
     except BatchError as e:
         click.echo(f"    (Could not fetch live status: {e})")
+
+
+def _show_retry_details(manifest, retry_id: str):
+    """Show detailed status for a retry job."""
+    # Find the retry info
+    retry_info = None
+    for retry in manifest.retries:
+        if retry.retry_id == retry_id:
+            retry_info = retry
+            break
+
+    if not retry_info:
+        click.echo(f"Retry job not found: {retry_id}", err=True)
+        click.echo(f"Known retries: {[r.retry_id for r in manifest.retries]}", err=True)
+        raise SystemExit(1)
+
+    click.echo()
+    click.echo(f"Retry Job:   {retry_id}")
+    click.echo(f"Parent Job:  {manifest.job_id}")
+    click.echo(f"Pipeline:    {manifest.pipeline}")
+    click.echo(f"User:        {manifest.user}")
+    click.echo(
+        f"Created:     {retry_info.created.isoformat()} ({format_time_ago(retry_info.created)})"
+    )
+
+    click.echo()
+    click.echo("Retry Config:")
+    click.echo(f"  Indices:   {retry_info.indices}")
+    if retry_info.reslice_prefix:
+        click.echo(f"  Reslice:   {retry_info.reslice_prefix} ({retry_info.reslice_count} chunks)")
+    else:
+        click.echo(f"  Reslice:   No (retrying original chunks)")
+
+    if retry_info.batch_job_id:
+        click.echo()
+        click.echo("Batch:")
+        click.echo(f"  AWS Job ID: {retry_info.batch_job_id}")
+
+        # Get live status from AWS Batch
+        _show_array_status(retry_info.batch_job_id)
+
+    click.echo()
+    click.echo("Next steps:")
+    click.echo(f"  View logs:      dh batch logs {manifest.job_id}")
+    click.echo(f"  Parent status:  dh batch status {manifest.job_id}")

{dayhoff_tools-1.14.11.dist-info → dayhoff_tools-1.14.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dayhoff-tools
-Version: 1.14.11
+Version: 1.14.12
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.14.11.dist-info → dayhoff_tools-1.14.12.dist-info}/RECORD

@@ -19,7 +19,7 @@ dayhoff_tools/cli/batch/commands/list_jobs.py,sha256=COfxZddDVUAHeTayNAB3ruYNhgr
 dayhoff_tools/cli/batch/commands/local.py,sha256=dZeKhNakaM1jS-EoByAwg1nWspRRoOmYzcwzjEKBaIA,3226
 dayhoff_tools/cli/batch/commands/logs.py,sha256=ctgJksdzFmqBdD18ePPsZe2BpuJYtHz2xAaMPnUplmQ,5293
 dayhoff_tools/cli/batch/commands/retry.py,sha256=JgOMiwuESNG4Wp_fQ_vxy1RwyOZPt_kmVbLYdxYTVBY,9897
-dayhoff_tools/cli/batch/commands/status.py,sha256=k8ltke62vKUX_4MmnFDEd6H2s8KmVFA71UUXLbleA7o,9790
+dayhoff_tools/cli/batch/commands/status.py,sha256=4kfii7mgtJKXWwZDsR4mtb_dQgCdryGkGEjNOpDxTT0,13404
 dayhoff_tools/cli/batch/commands/submit.py,sha256=AXbvSReN8fLlzR5swE81pH7yvbCC1GUMCbsDrfoHAws,6786
 dayhoff_tools/cli/batch/job_id.py,sha256=mqr8SwcPlUWIYLaR_C4kACmL2ZFK8jaddd7B-45-XaQ,4246
 dayhoff_tools/cli/batch/manifest.py,sha256=eLyiOFXonAUh5rfHcXyR2CzmRKXQz9-tTbJcWnVbbmE,8857
@@ -71,7 +71,7 @@ dayhoff_tools/intake/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJq
 dayhoff_tools/logs.py,sha256=DKdeP0k0kliRcilwvX0mUB2eipO5BdWUeHwh-VnsICs,838
 dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
 dayhoff_tools/warehouse.py,sha256=UETBtZD3r7WgvURqfGbyHlT7cxoiVq8isjzMuerKw8I,24475
-dayhoff_tools-1.14.11.dist-info/METADATA,sha256=-3MjX0CiSeDoQ0MiF0mQlw700sWzZgeyVowtbljZ5qc,3185
-dayhoff_tools-1.14.11.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-dayhoff_tools-1.14.11.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.14.11.dist-info/RECORD,,
+dayhoff_tools-1.14.12.dist-info/METADATA,sha256=Sfuw9gESP4Yfum_ySCZoxyuIFUybEcfGvywZaBSBy7c,3185
+dayhoff_tools-1.14.12.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+dayhoff_tools-1.14.12.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.14.12.dist-info/RECORD,,