diff --git a/generate/container.py b/generate/container.py index 0802237..094a65a 100644 --- a/generate/container.py +++ b/generate/container.py @@ -467,6 +467,12 @@ class Container: secrets += secret.create() return secrets + def create_environment(self) -> str: + """Wrap self.env.create().""" + if self.env is None: + return "" + return self.env.create() + def create_container(self) -> str: """Create podman container.""" cmd = "" @@ -541,3 +547,9 @@ class Container: for secret in self.secrets: secrets += secret.remove() return secrets + + def remove_environment(self) -> str: + """Wrap self.env.remove().""" + if self.env is None: + return "" + return self.env.remove()