From 3f5ccffc4a9357f36713f224f0ad4f8f1842d219 Mon Sep 17 00:00:00 2001
From: Enno Tensing <tenno@suij.in>
Date: Tue, 22 Jul 2025 13:17:35 +0200
Subject: [PATCH] generate: container: Add {create,remove}_environment wrappers

Add {create,remove}_environment wrappers to the Container class, so that
the generate script can access them.

Signed-off-by: Enno Tensing <tenno@suij.in>
---
 generate/container.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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()