dayhoff-tools 1.3.6__tar.gz → 1.3.7__tar.gz

{dayhoff_tools-1.3.6 → dayhoff_tools-1.3.7}/PKG-INFO

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

{dayhoff_tools-1.3.6 → dayhoff_tools-1.3.7}/dayhoff_tools/cli/engine_commands.py

@@ -163,7 +163,7 @@ def format_status(state: str, ready: Optional[bool]) -> str:
         if ready is True:
             return "[green]Running ✓[/green]"
         elif ready is False:
-            return "[yellow]Running ⚠ (Initializing...)[/yellow]"
+            return "[yellow]Running ⚠ (Bootstrapping...)[/yellow]"
         else:
             return "[green]Running[/green]"
     elif state.lower() == "stopped":
@@ -332,7 +332,6 @@ def launch_engine(
     else:
         error = response.json().get("error", "Unknown error")
         console.print(f"[red]❌ Failed to launch engine: {error}[/red]")
-        raise typer.Exit(1)
 
 
 @engine_app.command("list")
@@ -948,15 +947,43 @@ def create_ami(
 
             response = ec2.create_image(**create_params)
 
-        ami_id = response["ImageId"]
-        console.print(f"[green]✓ AMI creation initiated![/green]")
-        console.print(f"AMI ID: [cyan]{ami_id}[/cyan]")
-        console.print("\n[dim]The AMI creation process will continue in the background.[/dim]")
-        console.print("[dim]You can monitor progress in the EC2 Console under 'AMIs'.[/dim]")
-        console.print(
-            f"\nOnce complete, run [cyan]terraform apply[/cyan] in "
-            f"terraform/environments/dev to use the new AMI."
-        )
+            ami_id = response["ImageId"]
+            console.print(f"[green]✓ AMI creation initiated![/green]")
+            console.print(f"AMI ID: [cyan]{ami_id}[/cyan]")
+            
+            # Restore the source engine to a normal state
+            console.print("Restoring source engine state...")
+            restore_response = ssm.send_command(
+                InstanceIds=[engine["instance_id"]],
+                DocumentName="AWS-RunShellScript",
+                Parameters={
+                    "commands": [
+                        "sudo touch /opt/dayhoff/first_boot_complete.sentinel",
+                        "sudo systemctl restart engine-idle-detector.timer",
+                    ],
+                    "executionTimeout": ["60"],
+                },
+            )
+
+            # Quick wait to see if it failed immediately
+            time.sleep(1)
+            restore_command_id = restore_response["Command"]["CommandId"]
+            result = ssm.get_command_invocation(
+                CommandId=restore_command_id,
+                InstanceId=engine["instance_id"],
+            )
+            if result["Status"] not in ["Pending", "InProgress", "Success"]:
+                 console.print("[yellow]⚠️  Warning: Failed to restore source engine state.[/yellow]")
+            else:
+                console.print("[green]✓ Source engine restored to normal operation.[/green]")
+
+
+            console.print("\n[dim]The AMI creation process will continue in the background.[/dim]")
+            console.print("[dim]You can monitor progress in the EC2 Console under 'AMIs'.[/dim]")
+            console.print(
+                f"\nOnce complete, run [cyan]terraform apply[/cyan] in "
+                f"terraform/environments/dev to use the new AMI."
+            )
 
     except ClientError as e:
         console.print(f"[red]❌ Failed to create AMI: {e}[/red]")

{dayhoff_tools-1.3.6 → dayhoff_tools-1.3.7}/pyproject.toml

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