feat(Workspaces): allow user to specify workspace docker image (#95)

BLSQ · Dec 6, 2023 · 2455a3e · 2455a3e
1 parent 93e277e
commit 2455a3e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/openhexa/cli/cli.py b/openhexa/cli/cli.py
@@ -410,6 +410,8 @@ def pipelines_run(
     if force_pull:
         cmd.extend(["--pull", "always"])
 
+    image = env_vars["WORKSPACE_DOCKER_IMAGE"] if env_vars.get("WORKSPACE_DOCKER_IMAGE") else image
+
     cmd.extend(
         [
             image,

diff --git a/openhexa/sdk/pipelines/utils.py b/openhexa/sdk/pipelines/utils.py
@@ -147,4 +147,8 @@ def get_local_workspace_config(path: Path):
                     for key, value in connection_config.items():
                         if key != "type":
                             env_vars[stringcase.constcase(f"{slug}_{key.lower()}")] = str(value)
+        # Workspace docker image
+        if "image" in local_workspace_config:
+            env_vars["WORKSPACE_DOCKER_IMAGE"] = local_workspace_config["image"]
+
     return env_vars