dayhoff-tools 1.3.28__tar.gz → 1.3.29__tar.gz

{dayhoff_tools-1.3.28 → dayhoff_tools-1.3.29}/PKG-INFO

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

{dayhoff_tools-1.3.28 → dayhoff_tools-1.3.29}/dayhoff_tools/cli/engine_commands.py

@@ -1314,6 +1314,40 @@ def create_ami(
                 )
                 console.print(f"  Excluding volume at {device_name}")
 
+        # --- Check & detach attached studios --------------------------------------------------
+        # If any user studios are still attached we must detach them before the instance reboots
+        # for snapshot consistency; otherwise Studio-Manager metadata becomes stale.
+
+        attached_resp = make_api_request("GET", f"/engines/{engine['instance_id']}/studios")
+        attached_studios = attached_resp.json().get("studios", []) if attached_resp.status_code == 200 else []
+
+        if attached_studios:
+            console.print(f"[yellow]⚠  {len(attached_studios)} studio(s) attached to this engine.[/yellow]")
+            for s in attached_studios:
+                console.print(f"  • {s['user']} ({s['studio_id']})")
+
+            if not Confirm.ask("Detach all studios before AMI creation?"):
+                console.print("AMI creation cancelled.")
+                return
+
+            console.print("Detaching studios…")
+            for s in attached_studios:
+                resp = make_api_request("POST", f"/studios/{s['studio_id']}/detach")
+                if resp.status_code != 200:
+                    console.print(f"[red]❌ Failed to detach {s['studio_id']} – aborting.[/red]")
+                    return
+
+            # Wait briefly for volumes to become available (max 2 min)
+            import time, boto3
+            ec2_wait = boto3.client("ec2", region_name="us-east-1")
+            vol_ids = [s['studio_id'] for s in attached_studios]
+            console.print("Waiting for volumes to detach…")
+            waiter = ec2_wait.get_waiter("volume_available")
+            try:
+                waiter.wait(VolumeIds=vol_ids, WaiterConfig={"Delay": 5, "MaxAttempts": 24})
+            except Exception:
+                console.print("[yellow]Proceeding even though some volumes may still be detaching.[/yellow]")
+
         # Create the AMI
         with Progress(
             SpinnerColumn(),
@@ -1596,7 +1630,7 @@ def attach_studio(
             console.print("Waiting for engine to be ready...")
             # Wait until the engine is fully running & marked ready (max 5 min)
             console.print("Waiting for engine to be ready (this can take a couple of minutes)…")
-            from rich.progress import Progress, SpinnerColumn, TimeElapsedColumn
+            # Progress classes already imported at module level – do not re-import to avoid UnboundLocalError
 
             with Progress(SpinnerColumn(), TimeElapsedColumn(), TextColumn("[progress.description]{task.description}"), transient=True) as prog:
                 prog.add_task("Awaiting engine readiness…", total=None)

{dayhoff_tools-1.3.28 → dayhoff_tools-1.3.29}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.3.28"
+version = "1.3.29"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}