another one

alexrf45 · Dec 1, 2024 · 757fce4 · 757fce4
1 parent 493853e
commit 757fce4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/aegis-wrapper/aegis/cli.py b/aegis-wrapper/aegis/cli.py
@@ -7,10 +7,9 @@
 
 def main():
     parser = argparse.ArgumentParser(description="Aegis Docker Wrapper TUI")
-    parser.add_argument('command', choices=[
-                        'start', 'stop', 'destroy', 'pull'],
+    parser.add_argument('command', choices=['start', 'stop', 'destroy', 'pull'],
                         help="Action to perform")
-    # parser.add_argument('-img', help="Docker image to use")
+    parser.add_argument('-img', '--image', help="Docker image to use"),
     parser.add_argument('-n', '--name', help="project/container name")
     parser.add_argument('--host-network', action='store_true',
                         help="Enable host networking")

diff --git a/aegis-wrapper/aegis/docker_handler.py b/aegis-wrapper/aegis/docker_handler.py
@@ -56,6 +56,7 @@ def pull_image(self, image_name: str):
             progress.remove_task(task)
 
     def start_container(self,
+                        image_name: str,
                         project_name: str,
                         name: str,
                         host_network: bool,
@@ -70,6 +71,7 @@ def start_container(self,
                 'bind': '/tmp/.X11-unix', 'mode': 'rw'}
 
         container = self.client.containers.run(
+            image=image_name,
             name=project_name,
             command="zsh",
             detach=False,