dayhoff-tools 1.5.5__tar.gz → 1.5.7__tar.gz

{dayhoff_tools-1.5.5 → dayhoff_tools-1.5.7}/PKG-INFO

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

{dayhoff_tools-1.5.5 → dayhoff_tools-1.5.7}/dayhoff_tools/cli/engine_commands.py

@@ -415,7 +415,7 @@ def update_ssh_config_entry(engine_name: str, instance_id: str, ssh_user: str, i
 Host {engine_name} {SSH_MANAGED_COMMENT}
     HostName {instance_id}
     User {ssh_user}
-    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT={idle_timeout} aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p,idleTimeout={idle_timeout}'\"\n"""
+    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT={idle_timeout} aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p'\"\n"""
 
     # Check if entry already exists
     host_line = f"Host {engine_name} {SSH_MANAGED_COMMENT}"
@@ -1029,7 +1029,7 @@ def config_ssh(
                     f"Host {engine['name']} {SSH_MANAGED_COMMENT}",
                     f"    HostName {engine['instance_id']}",
                     f"    User {ssh_user}",
-                    f"    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT=600 aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p,idleTimeout=600'\"",
+                    f"    ProxyCommand sh -c \"AWS_SSM_IDLE_TIMEOUT=600 aws ssm start-session --target %h --document-name AWS-StartSSHSession --parameters 'portNumber=%p'\"",
                 ]
             )
 

{dayhoff_tools-1.5.5 → dayhoff_tools-1.5.7}/dayhoff_tools/warehouse.py

@@ -9,6 +9,25 @@ from zoneinfo import ZoneInfo
 # Import cloud helper lazily inside functions to avoid heavy deps at module load
 
 
+def _find_project_root() -> Path | None:
+    """
+    Find the project root by searching upwards from the current directory for
+    a `.git` directory or a `pyproject.toml` file.
+
+    Returns:
+        The path to the project root, or None if not found.
+    """
+    current_dir = Path.cwd().resolve()
+    while current_dir != current_dir.parent:
+        if (current_dir / ".git").is_dir() or (current_dir / "pyproject.toml").is_file():
+            return current_dir
+        current_dir = current_dir.parent
+    # Check the final directory in the hierarchy (e.g., '/')
+    if (current_dir / ".git").is_dir() or (current_dir / "pyproject.toml").is_file():
+        return current_dir
+    return None
+
+
 def _warn_if_gcp_default_sa(force_prompt: bool = False) -> None:
     """Warn the user when the active gcloud principal is the default VM service
     account.  See detailed docstring later in file (duplicate for early
@@ -528,11 +547,17 @@ def import_from_warehouse_typer() -> None:
     import questionary
 
     # Ensure execution from root directory
-    cwd = Path(os.getcwd())
-    if cwd.parent.name != "workspaces" or str(cwd.parent.parent) != cwd.root:
-        raise Exception(
-            f"This command must be executed from the repo's root directory (/workspaces/reponame). Current directory: {cwd}"
+    project_root = _find_project_root()
+    cwd = Path.cwd()
+    if not project_root or project_root != cwd:
+        error_msg = (
+            "This command must be run from the project's root directory, which is"
+            " expected to contain a `.git` folder or a `pyproject.toml` file.\n"
+            f"Current directory: {cwd}"
         )
+        if project_root:
+            error_msg += f"\nDetected project root: {project_root}"
+        raise Exception(error_msg)
 
     # Use questionary for prompts instead of typer
     warehouse_path = questionary.text("Warehouse path:").ask()
@@ -574,11 +599,17 @@ def get_from_warehouse_typer() -> None:
     import questionary
 
     # Ensure execution from root directory
-    cwd = Path(os.getcwd())
-    if cwd.parent.name != "workspaces" or str(cwd.parent.parent) != cwd.root:
-        raise Exception(
-            f"This command must be executed from the repo's root directory (/workspaces/reponame). Current directory: {cwd}"
+    project_root = _find_project_root()
+    cwd = Path.cwd()
+    if not project_root or project_root != cwd:
+        error_msg = (
+            "This command must be run from the project's root directory, which is"
+            " expected to contain a `.git` folder or a `pyproject.toml` file.\n"
+            f"Current directory: {cwd}"
         )
+        if project_root:
+            error_msg += f"\nDetected project root: {project_root}"
+        raise Exception(error_msg)
 
     # Use questionary for prompts instead of typer
     warehouse_path = questionary.text("Warehouse path:").ask()
@@ -619,11 +650,17 @@ def add_to_warehouse_typer() -> None:
     import questionary
 
     # Ensure execution from root directory
-    cwd = Path(os.getcwd())
-    if cwd.parent.name != "workspaces" or str(cwd.parent.parent) != cwd.root:
-        raise Exception(
-            f"This command must be executed from the repo's root directory (/workspaces/reponame). Current directory: {cwd}"
+    project_root = _find_project_root()
+    cwd = Path.cwd()
+    if not project_root or project_root != cwd:
+        error_msg = (
+            "This command must be run from the project's root directory, which is"
+            " expected to contain a `.git` folder or a `pyproject.toml` file.\n"
+            f"Current directory: {cwd}"
         )
+        if project_root:
+            error_msg += f"\nDetected project root: {project_root}"
+        raise Exception(error_msg)
 
     # Prompt for the data file path
     warehouse_path = questionary.text("Data file to be registered:").ask()

{dayhoff_tools-1.5.5 → dayhoff_tools-1.5.7}/pyproject.toml

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