dayhoff-tools 1.5.0__py3-none-any.whl → 1.5.2__py3-none-any.whl

dayhoff_tools/cli/engine_commands.py

@@ -1,6 +1,7 @@
 """Engine and Studio management commands for DHT CLI."""
 
 import json
+import shutil
 import subprocess
 import sys
 import time
@@ -365,6 +366,19 @@ def get_ssh_public_key() -> str:
     )
 
 
+def check_session_manager_plugin():
+    """Check if AWS Session Manager Plugin is available and warn if not."""
+    if shutil.which("session-manager-plugin") is None:
+        console.print(
+            "[bold red]⚠️  AWS Session Manager Plugin not found![/bold red]\n"
+            "SSH connections to engines require the Session Manager Plugin.\n"
+            "Please install it following the setup guide:\n"
+            "[link]https://github.com/dayhofflabs/nutshell/blob/main/REFERENCE/setup_guides/new-laptop.md[/link]"
+        )
+        return False
+    return True
+
+
 def update_ssh_config_entry(engine_name: str, instance_id: str, ssh_user: str):
     """Add or update a single SSH config entry for the given SSH user."""
     config_path = Path.home() / ".ssh" / "config"
@@ -850,6 +864,10 @@ def ssh_engine(
     Pass `--admin` to connect with the underlying [`ec2-user`] account for break-glass or debugging.
     """
     username = check_aws_sso()
+    
+    # Check for Session Manager Plugin
+    if not check_session_manager_plugin():
+        raise typer.Exit(1)
 
     # Get all engines to resolve name
     response = make_api_request("GET", "/engines")
@@ -887,6 +905,10 @@ def config_ssh(
     """Update SSH config with available engines."""
     username = check_aws_sso()
 
+    # Only check for Session Manager Plugin if we're not just cleaning
+    if not clean and not check_session_manager_plugin():
+        raise typer.Exit(1)
+
     if clean:
         console.print("Removing all managed SSH entries...")
     else:
@@ -1568,6 +1590,10 @@ def attach_studio(
     """Attach your studio to an engine."""
     username = check_aws_sso()
     
+    # Check for Session Manager Plugin since we'll update SSH config
+    if not check_session_manager_plugin():
+        raise typer.Exit(1)
+    
     # Use specified user if provided, otherwise use current user
     target_user = user if user else username
     

dayhoff_tools/warehouse.py

@@ -6,8 +6,6 @@ from io import StringIO
 from pathlib import Path
 from zoneinfo import ZoneInfo
 
-from ruamel.yaml import YAML
-
 # Import cloud helper lazily inside functions to avoid heavy deps at module load
 
 
@@ -135,6 +133,8 @@ def format_yaml_with_meta_spacing(yaml_str: str) -> str:
 
 def update_dvc_files(directory):
     """Traverse directory and update .dvc files with human-readable size, preserving existing formatting"""
+    from ruamel.yaml import YAML
+
     yaml = YAML()
     yaml.preserve_quotes = True
     yaml.indent(mapping=2, sequence=4, offset=2)
@@ -249,6 +249,8 @@ def import_from_warehouse(
         subprocess.run(command, check=True)
 
     # Copy meta section from warehouse_path to final_path.dvc
+    from ruamel.yaml import YAML
+
     yaml = YAML()
     yaml.preserve_quotes = True
     yaml.indent(mapping=2, sequence=4, offset=2)
@@ -309,6 +311,8 @@ def add_to_warehouse(
 
     # Process each ancestor .dvc file
     ancestors = []
+    from ruamel.yaml import YAML
+
     yaml_loader = YAML()
     yaml_loader.preserve_quotes = True
     yaml_loader.indent(mapping=2, sequence=4, offset=2)

{dayhoff_tools-1.5.0.dist-info → dayhoff_tools-1.5.2.dist-info}/METADATA

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

{dayhoff_tools-1.5.0.dist-info → dayhoff_tools-1.5.2.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=jXaUXbdpZKssFbMdwT6brvTuBmLkYdIHcjt3L5ZP6Ys,86313
+dayhoff_tools/cli/engine_commands.py,sha256=FvQTWb43x2Ns5rOW--ryfWkzzqvSzcpExDKyERY53Zk,87320
 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
@@ -26,8 +26,8 @@ dayhoff_tools/intake/structure.py,sha256=ufN3gAodQxhnt7psK1VTQeu9rKERmo_PhoxIbB4
 dayhoff_tools/intake/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJqE4,16456
 dayhoff_tools/logs.py,sha256=DKdeP0k0kliRcilwvX0mUB2eipO5BdWUeHwh-VnsICs,838
 dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
-dayhoff_tools/warehouse.py,sha256=6HxM8JO8NlYXxXApjptkNfnGrRLMScVeQI_xdFu4His,23062
-dayhoff_tools-1.5.0.dist-info/METADATA,sha256=9tvIpdKfGWRFN_caTCipQd0Bi40znW9pa_zGY4-ZVfM,2824
-dayhoff_tools-1.5.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.5.0.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.5.0.dist-info/RECORD,,
+dayhoff_tools/warehouse.py,sha256=fV3goH2cH1Y0oLpGERnu4p70P2JfByJHjBh_oMRv9C0,23134
+dayhoff_tools-1.5.2.dist-info/METADATA,sha256=sb8H_nJZYquFb8E_uvrj2xT3stsRS4x_uHdxQk7z15A,2824
+dayhoff_tools-1.5.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.5.2.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.5.2.dist-info/RECORD,,