feat(workspace): PATHWAYS-634 update worksapce settings (#304)

Author: nazarfil

openhexa/graphql/graphql_client/client.py

@@ -677,6 +677,7 @@ def update_workspace(
                   slug
                   name
                   description
+                  configuration
                   countries {
                     code
                     alpha3

openhexa/graphql/graphql_client/input_types.py

@@ -609,6 +609,7 @@ class UpdateWebappInput(BaseModel):
 
 
 class UpdateWorkspaceInput(BaseModel):
+    configuration: Optional[Any] = None
     countries: Optional[List["CountryInput"]] = None
     description: Optional[str] = None
     docker_image: Optional[str] = Field(alias="dockerImage", default=None)

openhexa/graphql/graphql_client/update_workspace.py

@@ -1,7 +1,7 @@
 # Generated by ariadne-codegen
 # Source: openhexa/graphql/queries.graphql
 
-from typing import List, Optional
+from typing import Any, List, Optional
 
 from pydantic import Field
 
@@ -23,6 +23,7 @@ class UpdateWorkspaceUpdateWorkspaceWorkspace(BaseModel):
     slug: str
     name: str
     description: Optional[str]
+    configuration: Optional[Any]
     countries: List["UpdateWorkspaceUpdateWorkspaceWorkspaceCountries"]
 
 

openhexa/graphql/queries.graphql

@@ -262,6 +262,7 @@ mutation UpdateWorkspace($input: UpdateWorkspaceInput!) {
       slug
       name
       description
+      configuration
       countries {
         code
         alpha3

openhexa/graphql/schema.generated.graphql

@@ -3903,6 +3903,7 @@ enum UpdateWorkspaceError {
 
 """Represents the input for updating a workspace."""
 input UpdateWorkspaceInput {
+  configuration: JSON
   countries: [CountryInput!]
   description: String
   dockerImage: String

openhexa/sdk/workspaces/current_workspace.py

@@ -8,6 +8,7 @@
 from warnings import warn
 
 from openhexa.graphql.graphql_client import WorkspaceWorkspaceCountries
+from openhexa.graphql.graphql_client.input_types import UpdateWorkspaceInput
 from openhexa.utils import stringcase
 
 from ..datasets import Dataset
@@ -80,6 +81,46 @@ def configuration(self) -> dict[str, str | dict] | None:
             return None
         return OpenHexaClient().workspace(slug=self.slug).configuration
 
+    @configuration.setter
+    def configuration(self, value: dict[str, str | dict]) -> None:
+        """Set the workspace configuration.
+
+        Parameters
+        ----------
+        value : dict[str, str | dict]
+            The configuration dictionary to set for the workspace.
+            Keys must be strings and values can be strings or JSON objects.
+
+        Raises
+        ------
+        WorkspaceConfigError
+            If not connected to the API or if the update fails.
+
+        Examples
+        --------
+        >>> workspace.configuration = {
+        ...     "api_url": "https://api.example.com",
+        ...     "debug_mode": "true",
+        ...     "SNT_CONFIG": "VERY_GOOD_CONFIG"
+        ... }
+        """
+        if not self._connected:
+            raise WorkspaceConfigError("Cannot update configuration: not connected to the API.")
+
+        try:
+            client = OpenHexaClient()
+
+            input_data = UpdateWorkspaceInput(slug=self.slug, configuration=value)
+            result = client.update_workspace(input=input_data)
+
+            if not result.success:
+                raise WorkspaceConfigError("Failed to update workspace configuration.")
+
+        except Exception as e:
+            if isinstance(e, WorkspaceConfigError):
+                raise
+            raise WorkspaceConfigError(f"Failed to update workspace configuration: {str(e)}")
+
     @property
     def database_host(self) -> str:
         """The workspace database host."""

tests/test_workspace.py

@@ -557,3 +557,24 @@ def test_workspace_configuration(self, workspace):
         with mock.patch("openhexa.sdk.workspaces.current_workspace.OpenHexaClient") as mock_client:
             mock_client.return_value.workspace.return_value = mock_workspace_data
             assert workspace.configuration == mock_config
+
+    def test_workspace_configuration_setter(self, workspace):
+        """Test setting workspace configuration through property setter."""
+        new_config = {
+            "api_url": "https://new-api.example.com",
+            "environment": "development",
+            "features": {"feature1": True, "feature2": False},
+        }
+
+        mock_result = mock.Mock()
+        mock_result.success = True
+
+        with mock.patch("openhexa.sdk.workspaces.current_workspace.OpenHexaClient") as mock_client:
+            mock_client.return_value.update_workspace.return_value = mock_result
+
+            workspace.configuration = new_config
+
+            mock_client.return_value.update_workspace.assert_called_once()
+            call_args = mock_client.return_value.update_workspace.call_args[1]["input"]
+            assert call_args.slug == workspace.slug
+            assert call_args.configuration == new_config