Allow options for the stack methods

SpareCores · Jul 26, 2024 · 92f6bac · 92f6bac
1 parent a372762
commit 92f6bac
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sparecores-runner"
-version = "0.0.20"
+version = "0.0.21"
 requires-python = ">= 3.9"
 dependencies = [
     "click",

diff --git a/src/sc_runner/runner.py b/src/sc_runner/runner.py
@@ -46,7 +46,7 @@ def pulumi_stack(
     return stack
 
 
-def create(vendor, pulumi_opts, resource_opts):
+def create(vendor, pulumi_opts, resource_opts, stack_opts=dict(on_output=print)):
     resource_f = getattr(resources, f"{resources.PREFIX}{vendor}")
     if not pulumi_opts.get("stack_name"):
         pulumi_opts["stack_name"] = get_stack_name(vendor, resource_f, resource_opts)
@@ -55,24 +55,24 @@ def pulumi_program():
         return resource_f(**resource_opts)
 
     stack = pulumi_stack(pulumi_program, **pulumi_opts)
-    stack.up(on_output=print)
+    stack.up(**stack_opts)
 
 
-def destroy(vendor, pulumi_opts, resource_opts):
+def destroy(vendor, pulumi_opts, resource_opts, stack_opts=dict(on_output=print)):
     resource_f = getattr(resources, f"{resources.PREFIX}{vendor}")
     if not pulumi_opts.get("stack_name"):
         pulumi_opts["stack_name"] = get_stack_name(vendor, resource_f, resource_opts)
     stack = pulumi_stack(lambda: None, **pulumi_opts)
-    stack.up(on_output=print)
+    stack.up(**stack_opts)
 
 
-def destroy_stack(vendor, pulumi_opts, resource_opts):
+def destroy_stack(vendor, pulumi_opts, resource_opts, stack_opts=dict(on_output=print)):
     resource_f = getattr(resources, f"{resources.PREFIX}{vendor}")
     if not pulumi_opts.get("stack_name"):
         pulumi_opts["stack_name"] = get_stack_name(vendor, resource_f, resource_opts)
     stack = pulumi_stack(lambda: None, **pulumi_opts)
-    stack.refresh(on_output=print)
-    stack.destroy(on_output=print)
+    stack.refresh(**stack_opts)
+    stack.destroy(**stack_opts)
     stack.workspace.remove_stack(stack.name)