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

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

Hide file tree

Showing 2 changed files with 13 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,15 @@ 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:
+            try:
+                env_vars["WORKSPACE_DOCKER_IMAGE"] = local_workspace_config["image"]["name"]
+            except KeyError:
+                exception_message = (
+                    "Invalid local workspace database config. Please make sure you provide the following "
+                    "key: image.name."
+                )
+                raise LocalWorkspaceConfigError(exception_message)
+
     return env_vars