Add stack delete confirmation prompt

src/zenml/cli/stack.py

@@ -564,9 +564,17 @@ def describe_stack(stack_name: Optional[str]) -> None:
 
 @stack.command("delete")
 @click.argument("stack_name", type=str)
-def delete_stack(stack_name: str) -> None:
+@click.option("--yes", "-y", is_flag=True, required=False)
+def delete_stack(stack_name: str, yes: bool = False) -> None:
     """Delete a stack."""
     cli_utils.print_active_profile()
+    confirmation = yes or cli_utils.confirmation(
+        f"This will delete stack '{stack_name}' from your repository. \n"
+        "Are you sure you want to proceed?"
+    )
+
+    if not confirmation:
+        cli_utils.declare("Stack deletion cancelled.")
 
     with console.status(f"Deleting stack '{stack_name}'...\n"):
         cfg = GlobalConfiguration()
@@ -588,8 +596,8 @@ def delete_stack(stack_name: str) -> None:
             )
             return
 
-        Repository().deregister_stack(stack_name)
-        cli_utils.declare(f"Deleted stack '{stack_name}'.")
+    Repository().deregister_stack(stack_name)
+    cli_utils.declare(f"Deleted stack '{stack_name}'.")
 
 
 @stack.command("set")

tests/unit/cli/test_stack.py

@@ -17,6 +17,7 @@
 
 from zenml.artifact_stores import LocalArtifactStore
 from zenml.cli.stack import (
+    delete_stack,
     describe_stack,
     remove_stack_component,
     rename_stack,
@@ -203,3 +204,19 @@ def test_remove_non_core_component_from_stack_succeeds(clean_repo) -> None:
     )
     assert result.exit_code == 0
     assert clean_repo.active_stack.secrets_manager is None
+
+
+def test_deleting_stack_with_flag_succeeds(clean_repo) -> None:
+    """Test stack delete with flag succeeds."""
+    new_stack = Stack(
+        name="arias_new_stack",
+        orchestrator=clean_repo.active_stack.orchestrator,
+        metadata_store=clean_repo.active_stack.metadata_store,
+        artifact_store=clean_repo.active_stack.artifact_store,
+    )
+    clean_repo.register_stack(new_stack)
+    runner = CliRunner()
+    result = runner.invoke(delete_stack, ["arias_new_stack", "-y"])
+    assert result.exit_code == 0
+    with pytest.raises(KeyError):
+        clean_repo.get_stack("arias_new_stack")