dayhoff-tools 1.13.1__tar.gz → 1.13.3__tar.gz

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/PKG-INFO

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

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/dayhoff_tools/cli/engines_studios/engine-studio-cli.md

@@ -321,8 +321,8 @@ dh engine2 list --env sand
 
 **Output:**
 ```
-Engines for AWS Account dev
 
+Engines for AWS Account dev
 ╭──────────────┬─────────────┬─────────────┬─────────────┬─────────────────────╮
 │ Name         │ State       │ User        │ Type        │ Instance ID         │
 ├──────────────┼─────────────┼─────────────┼─────────────┼─────────────────────┤
@@ -330,8 +330,8 @@ Engines for AWS Account dev
 │ bob-training │ running     │ bob         │ a10g        │ i-0fedcba987654321  │
 │ batch-worker │ stopped     │ charlie     │ cpumax      │ i-0abc123def456789  │
 ╰──────────────┴─────────────┴─────────────┴─────────────┴─────────────────────╯
+Total: 3
 
-Total: 3 engine(s)
 ```
 
 **Formatting:**
@@ -755,8 +755,8 @@ dh studio2 list
 
 **Output:**
 ```
-Studios for AWS Account dev
 
+Studios for AWS Account dev
 ╭────────┬──────────────┬──────────────┬───────────┬───────────────────────────╮
 │ User   │ Status       │ Attached To  │ Size      │ Studio ID                 │
 ├────────┼──────────────┼──────────────┼───────────┼───────────────────────────┤
@@ -764,8 +764,8 @@ Studios for AWS Account dev
 │ bob    │ available    │ -            │ 200GB     │ vol-0fedcba987654321      │
 │ carol  │ attaching    │ carol-gpu    │ 150GB     │ vol-0abc123def456789      │
 ╰────────┴──────────────┴──────────────┴───────────┴───────────────────────────╯
+Total: 3
 
-Total: 3 studio(s)
 ```
 
 **Formatting:**

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/dayhoff_tools/cli/engines_studios/engine_commands.py

@@ -123,7 +123,9 @@ def engine_cli():
 @click.option(
     "--skip-ssh-config", is_flag=True, help="Don't automatically update SSH config"
 )
-@click.option("--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments"
+)
 @click.option(
     "--env",
     default=None,
@@ -222,7 +224,9 @@ def launch_engine(
 @click.option(
     "--skip-ssh-config", is_flag=True, help="Don't automatically update SSH config"
 )
-@click.option("--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments"
+)
 @click.option(
     "--env",
     default=None,
@@ -345,7 +349,9 @@ def start_engine(
 
 @engine_cli.command("stop")
 @click.argument("name_or_id")
-@click.option("--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments")
+@click.option(
+    "--yes", "-y", is_flag=True, help="Skip confirmation for non-dev environments"
+)
 @click.option(
     "--env",
     default=None,
@@ -581,14 +587,16 @@ def list_engines(env: Optional[str]):
         engines = result.get("engines", [])
 
         # Show account header with blue account name
-        click.echo(f"Engines for AWS Account \033[34m{env}\033[0m\n")
+        click.echo(f"\nEngines for AWS Account \033[34m{env}\033[0m")
 
         if not engines:
-            click.echo("No engines found")
+            click.echo("No engines found\n")
             return
 
         # Calculate dynamic width for Name column (longest name + 2 for padding)
-        max_name_len = max((len(engine.get("name", "unknown")) for engine in engines), default=4)
+        max_name_len = max(
+            (len(engine.get("name", "unknown")) for engine in engines), default=4
+        )
         name_width = max(max_name_len + 2, len("Name") + 2)
 
         # Fixed widths for other columns
@@ -597,11 +605,20 @@ def list_engines(env: Optional[str]):
         type_width = 12
         id_width = 20
 
-        # Calculate total width for separator line
-        total_width = name_width + state_width + user_width + type_width + id_width + 9  # +9 for separators and spaces
-
         # Table top border
-        click.echo("╭" + "─" * (name_width + 1) + "┬" + "─" * (state_width + 1) + "┬" + "─" * (user_width + 1) + "┬" + "─" * (type_width + 1) + "┬" + "─" * (id_width + 1) + "╮")
+        click.echo(
+            "╭"
+            + "─" * (name_width + 1)
+            + "┬"
+            + "─" * (state_width + 1)
+            + "┬"
+            + "─" * (user_width + 1)
+            + "┬"
+            + "─" * (type_width + 1)
+            + "┬"
+            + "─" * (id_width + 1)
+            + "╮"
+        )
 
         # Table header
         click.echo(
@@ -609,7 +626,19 @@ def list_engines(env: Optional[str]):
         )
 
         # Header separator
-        click.echo("├" + "─" * (name_width + 1) + "┼" + "─" * (state_width + 1) + "┼" + "─" * (user_width + 1) + "┼" + "─" * (type_width + 1) + "┼" + "─" * (id_width + 1) + "┤")
+        click.echo(
+            "├"
+            + "─" * (name_width + 1)
+            + "┼"
+            + "─" * (state_width + 1)
+            + "┼"
+            + "─" * (user_width + 1)
+            + "┼"
+            + "─" * (type_width + 1)
+            + "┼"
+            + "─" * (id_width + 1)
+            + "┤"
+        )
 
         # Table rows
         for engine in engines:
@@ -621,11 +650,11 @@ def list_engines(env: Optional[str]):
 
             # Truncate if needed
             if len(name) > name_width - 1:
-                name = name[:name_width - 1]
+                name = name[: name_width - 1]
             if len(user) > user_width - 1:
-                user = user[:user_width - 1]
+                user = user[: user_width - 1]
             if len(engine_type) > type_width - 1:
-                engine_type = engine_type[:type_width - 1]
+                engine_type = engine_type[: type_width - 1]
 
             # Color the name (blue)
             name_display = f"\033[34m{name:<{name_width}}\033[0m"
@@ -648,9 +677,21 @@ def list_engines(env: Optional[str]):
             )
 
         # Table bottom border
-        click.echo("╰" + "─" * (name_width + 1) + "┴" + "─" * (state_width + 1) + "┴" + "─" * (user_width + 1) + "┴" + "─" * (type_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
+        click.echo(
+            "╰"
+            + "─" * (name_width + 1)
+            + "┴"
+            + "─" * (state_width + 1)
+            + "┴"
+            + "─" * (user_width + 1)
+            + "┴"
+            + "─" * (type_width + 1)
+            + "┴"
+            + "─" * (id_width + 1)
+            + "╯"
+        )
 
-        click.echo(f"\nTotal: {len(engines)} engine(s)")
+        click.echo(f"Total: {len(engines)}\n")
 
     except Exception as e:
         click.echo(f"✗ Error: {e}", err=True)

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/dayhoff_tools/cli/engines_studios/simulators/engine_list_simulator.py

@@ -24,10 +24,10 @@ def format_list_output(engines: list[dict[str, Any]], env: str = "dev") -> None:
     """Format and print engine list output matching the actual CLI."""
 
     # Header with blue account name
-    print(f"Engines for AWS Account {colorize(env, '34')}\n")
+    print(f"\nEngines for AWS Account {colorize(env, '34')}")
 
     if not engines:
-        print("No engines found")
+        print("No engines found\n")
         return
 
     # Calculate dynamic width for Name column (longest name + 2 for padding)
@@ -128,7 +128,7 @@ def format_list_output(engines: list[dict[str, Any]], env: str = "dev") -> None:
         + "╯"
     )
 
-    print(f"\nTotal: {len(engines)} engine(s)")
+    print(f"Total: {len(engines)}\n")
 
 
 def generate_scenarios() -> dict[str, dict[str, Any]]:

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/dayhoff_tools/cli/engines_studios/simulators/studio_list_simulator.py

@@ -24,10 +24,10 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
     """Format and print studio list output matching the actual CLI."""
 
     # Header with blue account name
-    print(f"Studios for AWS Account {colorize(env, '34')}\n")
+    print(f"\nStudios for AWS Account {colorize(env, '34')}")
 
     if not studios:
-        print("No studios found")
+        print("No studios found\n")
         return
 
     # Calculate dynamic width for User column (longest user + 2 for padding)
@@ -104,7 +104,7 @@ def format_list_output(studios: list[dict[str, Any]], engines_map: dict[str, str
     # Table bottom border
     print("╰" + "─" * (user_width + 1) + "┴" + "─" * (status_width + 1) + "┴" + "─" * (attached_width + 1) + "┴" + "─" * (size_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
 
-    print(f"\nTotal: {len(studios)} studio(s)")
+    print(f"Total: {len(studios)}\n")
 
 
 def generate_scenarios() -> dict[str, dict[str, Any]]:

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/dayhoff_tools/cli/engines_studios/studio_commands.py

@@ -258,8 +258,11 @@ def list_studios(env: Optional[str]):
         result = client.list_studios()
         studios = result.get("studios", [])
 
+        # Show account header with blue account name
+        click.echo(f"\nStudios for AWS Account \033[34m{env}\033[0m")
+
         if not studios:
-            click.echo("No studios found")
+            click.echo("No studios found\n")
             return
 
         # Get all engines to map instance IDs to names
@@ -346,7 +349,7 @@ def list_studios(env: Optional[str]):
         # Table bottom border
         click.echo("╰" + "─" * (user_width + 1) + "┴" + "─" * (status_width + 1) + "┴" + "─" * (attached_width + 1) + "┴" + "─" * (size_width + 1) + "┴" + "─" * (id_width + 1) + "╯")
 
-        click.echo(f"\nTotal: {len(studios)} studio(s)")
+        click.echo(f"Total: {len(studios)}\n")
 
     except Exception as e:
         click.echo(f"✗ Error: {e}", err=True)

{dayhoff_tools-1.13.1 → dayhoff_tools-1.13.3}/pyproject.toml

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