fixes and improvements

Author: pinzon

localstack_utils/container.py

@@ -23,21 +23,20 @@
 class Container:
     @staticmethod
     def create_localstack_container(
-        pull_new_image: bool,
-        image_name: str = None,
+        *args,
+        pull_new_image: bool = False,
+        image_name: str = LOCALSTACK_IMAGE_NAME,
         image_tag: str = LATEST_TAG,
         gateway_listen: str = "0.0.0.0:4566",
-        environment_variables: dict = None,
-        bind_ports: dict = None,
+        environment_variables: dict | None = None,
+        bind_ports: dict | None = None,
+        **kwargs,
     ):
         environment_variables = environment_variables or {}
         environment_variables["GATEWAY_LISTEN"] = gateway_listen
 
-        image_name_or_default = image_name or LOCALSTACK_IMAGE_NAME
         image_exists = (
-            True
-            if len(DOCKER_CLIENT.images.list(name=image_name_or_default))
-            else False
+            True if len(DOCKER_CLIENT.images.list(name=image_name)) else False
         )
 
         bind_ports = bind_ports or {}
@@ -46,10 +45,10 @@ def create_localstack_container(
 
         if pull_new_image or not image_exists:
             logging.info("Pulling latest image")
-            DOCKER_CLIENT.images.pull(image_name_or_default, image_tag)
+            DOCKER_CLIENT.images.pull(image_name, image_tag)
 
         return DOCKER_CLIENT.containers.run(
-            image_name_or_default,
+            image_name,
             ports=bind_ports,
             environment=environment_variables,
             detach=True,