dayhoff-tools 1.5.0__py3-none-any.whl → 1.5.1__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-1.5.0.dist-info → dayhoff_tools-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.5.0
+Version: 1.5.1
 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.1.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
@@ -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=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-1.5.1.dist-info/METADATA,sha256=5NW6Kdr32qJQ6kFkhf0zDSi7ZmuYRMPil638Yg9sWjU,2824
+dayhoff_tools-1.5.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.5.1.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.5.1.dist-info/RECORD,,