dayhoff-tools 1.3.28__py3-none-any.whl → 1.4.0__py3-none-any.whl

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.dist-info → dayhoff_tools-1.4.0.dist-info}/METADATA

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

{dayhoff_tools-1.3.28.dist-info → dayhoff_tools-1.4.0.dist-info}/RECORD

@@ -3,7 +3,7 @@ dayhoff_tools/chemistry/standardizer.py,sha256=uMn7VwHnx02nc404eO6fRuS4rsl4dvSPf
 dayhoff_tools/chemistry/utils.py,sha256=jt-7JgF-GeeVC421acX-bobKbLU_X94KNOW24p_P-_M,2257
 dayhoff_tools/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dayhoff_tools/cli/cloud_commands.py,sha256=33qcWLmq-FwEXMdL3F0OHm-5Stlh2r65CldyEZgQ1no,40904
-dayhoff_tools/cli/engine_commands.py,sha256=qnueTQBXrnNkEaPjICvNdFytDI-mO-RIc4REsGrFiiQ,81715
+dayhoff_tools/cli/engine_commands.py,sha256=xzhaWokapifSY74SpQ9W9Ens9isaCbTo_HDsgUKzmzo,83594
 dayhoff_tools/cli/main.py,sha256=rgeEHD9lJ8SBCR34BTLb7gVInHUUdmEBNXAJnq5yEU4,4795
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=qs8vH9TBFHsOPC3X8cU3qZigM3dDn-2Ytq4o_F2WubU,27874
@@ -27,7 +27,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=8YbnQ--usrEgDQGfvpV4MrMji55A0rq2hZaOgFGh6ag,15896
-dayhoff_tools-1.3.28.dist-info/METADATA,sha256=PKfyhnreJSp0BFCQDAQ_6_7GWk44e_xs0FszGQOwJv0,2825
-dayhoff_tools-1.3.28.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.3.28.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.3.28.dist-info/RECORD,,
+dayhoff_tools-1.4.0.dist-info/METADATA,sha256=BHiY-xDr4AnxgU9xob8jdZcqZgEAPjEMFj8gMdcFGXY,2824
+dayhoff_tools-1.4.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.4.0.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.4.0.dist-info/RECORD,,