dayhoff-tools 1.6.5__py3-none-any.whl → 1.6.7__py3-none-any.whl

dayhoff_tools/cli/engine_commands.py

@@ -1710,6 +1710,23 @@ def attach_studio(
 
     console.print(f"Attaching studio to engine [cyan]{engine['name']}[/cyan]...")
 
+    # Helper --------------------------------------------------------------
+    def _is_studio_attached(target_studio_id: str, target_vm_id: str) -> bool:
+        """Return True when the given studio already shows as attached to the VM.
+
+        Using this extra check lets us stop the outer retry loop as soon as the
+        asynchronous attach operation actually finishes, even in the unlikely
+        event that the operation-tracking DynamoDB record is not yet updated.
+        """
+        resp = make_api_request("GET", f"/studios/{target_studio_id}")
+        if resp.status_code != 200:
+            return False
+        data = resp.json()
+        return (
+            data.get("status") == "in-use"
+            and data.get("attached_vm_id") == target_vm_id
+        )
+
     # Determine retry strategy based on whether we just started the engine
     if engine_started_now:
         max_attempts = 40  # About 7 minutes total with exponential backoff
@@ -1734,6 +1751,12 @@ def attach_studio(
         last_error = None
 
         for attempt in range(max_attempts):
+            # If the attach already completed in the previous iteration (but we
+            # didn't notice because the operation table wasn’t updated yet),
+            # bail out early.
+            if _is_studio_attached(studio["studio_id"], engine["instance_id"]):
+                success = True
+                break
             success, error_msg = _attempt_studio_attach(
                 studio, engine, target_user, public_key
             )
@@ -1860,27 +1883,40 @@ def _attempt_studio_attach(studio, engine, target_user, public_key):
 
 
 def _poll_operation(operation_id: str, timeout: int = 600) -> Tuple[bool, str]:
-    """Poll /operations/<id> until COMPLETED or FAILED.
+    """Poll `/operations/<id>` until it reports COMPLETED or FAILED.
 
-    Returns (True, "") on success or (False, error_message) on failure / timeout.
-    Uses a simple status spinner to avoid nested Live displays.
+    Returns (True, "") on success or (False, error_message) on failure or
+    timeout.
+
+    IMPORTANT: This helper deliberately *avoids* creating its own Rich Live/
+    spinner so that it can be called from inside another Progress/Live context
+    (the outer "Attaching studio…" progress bar).  Nested Live renderables lead
+    to constant hide/show repaint cycles, which is what caused the flickering
+    the user observed.
     """
     start_ts = time.time()
-    with console.status("[cyan]Waiting for studio attach to complete…", spinner="dots") as status:
-        while True:
-            if time.time() - start_ts > timeout:
-                return False, "Timed out waiting for attach operation to complete"
-            op_resp = make_api_request("GET", f"/operations/{operation_id}")
-            if op_resp.status_code != 200:
-                return False, f"Failed to fetch operation status (HTTP {op_resp.status_code})"
-            data = op_resp.json()
-            op_status = data.get("status")
-            if op_status == "COMPLETED":
-                return True, ""
-            if op_status == "FAILED":
-                return False, data.get("error", "Operation failed")
-            # keep spinner steady – no constant message updates
-            time.sleep(3)
+    while True:
+        # ---- timeout guard -------------------------------------------------
+        if time.time() - start_ts > timeout:
+            return False, "Timed out waiting for attach operation to complete"
+
+        # ---- fetch status ---------------------------------------------------
+        op_resp = make_api_request("GET", f"/operations/{operation_id}")
+        if op_resp.status_code != 200:
+            return False, (
+                f"Failed to fetch operation status (HTTP {op_resp.status_code})"
+            )
+        data = op_resp.json()
+        op_status = data.get("status")
+
+        # ---- interpret state -----------------------------------------------
+        if op_status == "COMPLETED":
+            return True, ""
+        if op_status == "FAILED":
+            return False, data.get("error", "Operation failed")
+
+        # Not finished yet – wait a little and poll again (no live spinner)
+        time.sleep(3)
     
 
 @studio_app.command("detach")

{dayhoff_tools-1.6.5.dist-info → dayhoff_tools-1.6.7.dist-info}/METADATA

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

{dayhoff_tools-1.6.5.dist-info → dayhoff_tools-1.6.7.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=ohgcLv_y5UjeqBazh_BKnPGXSJn2rDqGCIb91VOUPiU,93633
+dayhoff_tools/cli/engine_commands.py,sha256=ge6GMrXUcc3xy_tt_Gn3bQUh5eqjYVEE4FDjoXO7-PM,95156
 dayhoff_tools/cli/main.py,sha256=tRN7WCBHg6uyNp6rA54pKTCoVmBntta2i0Yas3bUpZ4,4853
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=FRZTPrjsG_qmIIqoNxd1Q1vVkS_5w8aY33IrVYVNCLg,18131
@@ -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=heaYc64qplgN3_1WVPFmqj53goStioWwY5NqlWc4c0s,24453
-dayhoff_tools-1.6.5.dist-info/METADATA,sha256=8COj8yPA8o8aeykuMF9Jk_yIzp8wf8184BfST1oFLrs,2914
-dayhoff_tools-1.6.5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.6.5.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.6.5.dist-info/RECORD,,
+dayhoff_tools-1.6.7.dist-info/METADATA,sha256=_yxPKFxAXvOiL8eKqa3b4AKAOCk6F-bl_rQINQe44NU,2914
+dayhoff_tools-1.6.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.6.7.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.6.7.dist-info/RECORD,,