dh-cli 0.2.2__tar.gz → 0.2.3__tar.gz

{dh_cli-0.2.2 → dh_cli-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dh-cli
-Version: 0.2.2
+Version: 0.2.3
 Summary: Dayhoff Labs developer CLI
 Author-email: Dayhoff Labs <dev@dayhofflabs.com>
 License: # PolyForm Noncommercial License 1.0.0

{dh_cli-0.2.2 → dh_cli-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dh-cli"
-version = "0.2.2"
+version = "0.2.3"
 description = "Dayhoff Labs developer CLI"
 requires-python = ">=3.11"
 readme = "README.md"

{dh_cli-0.2.2 → dh_cli-0.2.3}/src/dh_cli/engines_studios/engine_commands.py

@@ -616,6 +616,7 @@ def list_engines(env: Optional[str]):
         state_width = 12
         user_width = 12
         type_width = 12
+        disk_width = 14
         id_width = 20
 
         # Table top border
@@ -629,13 +630,15 @@ def list_engines(env: Optional[str]):
             + "┬"
             + "─" * (type_width + 1)
             + "┬"
+            + "─" * (disk_width + 1)
+            + "┬"
             + "─" * (id_width + 1)
             + "╮"
         )
 
         # Table header
         click.echo(
-            f"│ {'Name':<{name_width}}│ {'State':<{state_width}}│ {'User':<{user_width}}│ {'Type':<{type_width}}│ {'Instance ID':<{id_width}}│"
+            f"│ {'Name':<{name_width}}│ {'State':<{state_width}}│ {'User':<{user_width}}│ {'Type':<{type_width}}│ {'Disk':>{disk_width}}│ {'Instance ID':<{id_width}}│"
         )
 
         # Header separator
@@ -649,6 +652,8 @@ def list_engines(env: Optional[str]):
             + "┼"
             + "─" * (type_width + 1)
             + "┼"
+            + "─" * (disk_width + 1)
+            + "┼"
             + "─" * (id_width + 1)
             + "┤"
         )
@@ -685,8 +690,22 @@ def list_engines(env: Optional[str]):
             # Color the instance ID (grey)
             instance_id_display = f"\033[90m{instance_id:<{id_width}}\033[0m"
 
+            # Color the disk usage
+            usage_pct = engine.get("boot_disk_usage_pct")
+            total_gb = engine.get("boot_disk_total_gb")
+            if usage_pct is not None and total_gb is not None:
+                disk_text = f"{usage_pct}% of {total_gb}G"
+                if usage_pct >= 90:
+                    disk_display = f"\033[31m{disk_text:>{disk_width}}\033[0m"
+                elif usage_pct >= 70:
+                    disk_display = f"\033[33m{disk_text:>{disk_width}}\033[0m"
+                else:
+                    disk_display = f"\033[32m{disk_text:>{disk_width}}\033[0m"
+            else:
+                disk_display = f"\033[90m{'-':>{disk_width}}\033[0m"
+
             click.echo(
-                f"│ {name_display}│ {state_display}│ {user:<{user_width}}│ {engine_type:<{type_width}}│ {instance_id_display}│"
+                f"│ {name_display}│ {state_display}│ {user:<{user_width}}│ {engine_type:<{type_width}}│ {disk_display}│ {instance_id_display}│"
             )
 
         # Table bottom border
@@ -700,6 +719,8 @@ def list_engines(env: Optional[str]):
             + "┴"
             + "─" * (type_width + 1)
             + "┴"
+            + "─" * (disk_width + 1)
+            + "┴"
             + "─" * (id_width + 1)
             + "╯"
         )

{dh_cli-0.2.2 → dh_cli-0.2.3}/src/dh_cli/engines_studios/studio_commands.py

@@ -299,9 +299,10 @@ def list_studios(env: Optional[str]):
             max_attached_len + 2, len("Attached To") + 2, 3
         )  # At least 3 for "-"
 
-        # Fixed widths for other columns - reordered to [User, Status, Attached To, Size, Studio ID]
+        # Fixed widths for other columns - reordered to [User, Status, Attached To, Size, Usage, Studio ID]
         status_width = 12
         size_width = 10
+        usage_width = 8
         id_width = 25
 
         # Table top border
@@ -315,13 +316,15 @@ def list_studios(env: Optional[str]):
             + "┬"
             + "─" * (size_width + 1)
             + "┬"
+            + "─" * (usage_width + 1)
+            + "┬"
             + "─" * (id_width + 1)
             + "╮"
         )
 
-        # Table header - reordered to [User, Status, Attached To, Size, Studio ID]
+        # Table header - reordered to [User, Status, Attached To, Size, Usage, Studio ID]
         click.echo(
-            f"│ {'User':<{user_width}}│ {'Status':<{status_width}}│ {'Attached To':<{attached_width}}│ {'Size':<{size_width}}│ {'Studio ID':<{id_width}}│"
+            f"│ {'User':<{user_width}}│ {'Status':<{status_width}}│ {'Attached To':<{attached_width}}│ {'Size':<{size_width}}│ {'Usage':>{usage_width}}│ {'Studio ID':<{id_width}}│"
         )
 
         # Header separator
@@ -335,6 +338,8 @@ def list_studios(env: Optional[str]):
             + "┼"
             + "─" * (size_width + 1)
             + "┼"
+            + "─" * (usage_width + 1)
+            + "┼"
             + "─" * (id_width + 1)
             + "┤"
         )
@@ -387,8 +392,21 @@ def list_studios(env: Optional[str]):
             # Color the studio ID (grey)
             studio_id_display = f"\033[90m{studio_id:<{id_width}}\033[0m"
 
+            # Format usage percentage
+            usage_pct = studio.get("usage_pct")
+            if usage_pct is not None:
+                usage_text = f"{usage_pct}%"
+                if usage_pct >= 90:
+                    usage_display = f"\033[31m{usage_text:>{usage_width}}\033[0m"
+                elif usage_pct >= 70:
+                    usage_display = f"\033[33m{usage_text:>{usage_width}}\033[0m"
+                else:
+                    usage_display = f"\033[32m{usage_text:>{usage_width}}\033[0m"
+            else:
+                usage_display = f"\033[90m{'-':>{usage_width}}\033[0m"
+
             click.echo(
-                f"│ {user_display}│ {status_display}│ {attached_display}│ {size:<{size_width}}│ {studio_id_display}│"
+                f"│ {user_display}│ {status_display}│ {attached_display}│ {size:<{size_width}}│ {usage_display}│ {studio_id_display}│"
             )
 
         # Table bottom border
@@ -402,6 +420,8 @@ def list_studios(env: Optional[str]):
             + "┴"
             + "─" * (size_width + 1)
             + "┴"
+            + "─" * (usage_width + 1)
+            + "┴"
             + "─" * (id_width + 1)
             + "╯"
         )