Merge

Author: yolanfery

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "2.4.0"
+  ".": "2.5.0"
 }

CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [2.5.0](https://github.com/BLSQ/openhexa-sdk-python/compare/v2.4.0...v2.5.0) (2025-06-20)
+
+
+### Features
+
+* add a "country" property to the current_workspace ([#273](https://github.com/BLSQ/openhexa-sdk-python/issues/273)) ([d942273](https://github.com/BLSQ/openhexa-sdk-python/commit/d942273c8ab7c8b9d41f6228dc783a9d2ec4b0b9))
+
 ## [2.4.0](https://github.com/BLSQ/openhexa-sdk-python/compare/v2.3.1...v2.4.0) (2025-06-16)
 
 

openhexa/cli/graphql/graphql_client/__init__.py

@@ -643,7 +643,6 @@
     DenyAccessmodAccessRequestDenyAccessmodAccessRequest,
 )
 from .disable_two_factor import DisableTwoFactor, DisableTwoFactorDisableTwoFactor
-from .dummy import Dummy
 from .enable_two_factor import EnableTwoFactor, EnableTwoFactorEnableTwoFactor
 from .enums import (
     AccessmodAccessibilityAnalysisAlgorithm,
@@ -812,6 +811,11 @@
     GenerateWorkspaceToken,
     GenerateWorkspaceTokenGenerateWorkspaceToken,
 )
+from .get_countries import (
+    GetCountries,
+    GetCountriesWorkspace,
+    GetCountriesWorkspaceCountries,
+)
 from .input_types import (
     AddPipelineOutputInput,
     AddToFavoritesInput,
@@ -2326,7 +2330,6 @@
     "DisableTwoFactorDisableTwoFactor",
     "DisableTwoFactorError",
     "DisableTwoFactorInput",
-    "Dummy",
     "EnableTwoFactor",
     "EnableTwoFactorEnableTwoFactor",
     "EnableTwoFactorError",
@@ -2370,6 +2373,9 @@
     "GenerateWorkspaceTokenError",
     "GenerateWorkspaceTokenGenerateWorkspaceToken",
     "GenerateWorkspaceTokenInput",
+    "GetCountries",
+    "GetCountriesWorkspace",
+    "GetCountriesWorkspaceCountries",
     "GraphQLClientError",
     "GraphQLClientGraphQLError",
     "GraphQLClientGraphQLMultiError",

openhexa/cli/graphql/graphql_client/client.py

@@ -80,7 +80,6 @@
 from .delete_workspace_member import DeleteWorkspaceMember
 from .deny_accessmod_access_request import DenyAccessmodAccessRequest
 from .disable_two_factor import DisableTwoFactor
-from .dummy import Dummy
 from .enable_two_factor import EnableTwoFactor
 from .enums import (
     AccessmodFilesetMode,
@@ -94,6 +93,7 @@
 from .generate_new_database_password import GenerateNewDatabasePassword
 from .generate_pipeline_webhook_url import GeneratePipelineWebhookUrl
 from .generate_workspace_token import GenerateWorkspaceToken
+from .get_countries import GetCountries
 from .input_types import (
     AddPipelineOutputInput,
     AddToFavoritesInput,
@@ -11724,17 +11724,24 @@ def workspaces(
         data = self.get_data(response)
         return Workspaces.model_validate(data)
 
-    def dummy(self, **kwargs: Any) -> Dummy:
+    def get_countries(self, workspace_slug: str, **kwargs: Any) -> GetCountries:
         query = gql(
             """
-            query dummy {
-              __typename
+            query getCountries($workspaceSlug: String!) {
+              workspace(slug: $workspaceSlug) {
+                countries {
+                  code
+                  name
+                  flag
+                  alpha3
+                }
+              }
             }
             """
         )
-        variables: Dict[str, object] = {}
+        variables: Dict[str, object] = {"workspaceSlug": workspace_slug}
         response = self.execute(
-            query=query, operation_name="dummy", variables=variables, **kwargs
+            query=query, operation_name="getCountries", variables=variables, **kwargs
         )
         data = self.get_data(response)
-        return Dummy.model_validate(data)
+        return GetCountries.model_validate(data)

openhexa/cli/graphql/graphql_client/get_countries.py

@@ -0,0 +1,27 @@
+# Generated by ariadne-codegen
+# Source: openhexa/cli/graphql/queries
+
+from typing import List, Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+
+
+class GetCountries(BaseModel):
+    workspace: Optional["GetCountriesWorkspace"]
+
+
+class GetCountriesWorkspace(BaseModel):
+    countries: List["GetCountriesWorkspaceCountries"]
+
+
+class GetCountriesWorkspaceCountries(BaseModel):
+    code: str
+    name: str
+    flag: str
+    alpha_3: str = Field(alias="alpha3")
+
+
+GetCountries.model_rebuild()
+GetCountriesWorkspace.model_rebuild()

openhexa/cli/graphql/queries/queries.graphql

@@ -1,3 +1,10 @@
-query dummy {
-    __typename
+query getCountries($workspaceSlug:String!){
+    workspace(slug: $workspaceSlug) {
+        countries {
+            code
+            name
+            flag
+            alpha3
+        }
+    }
 }

openhexa/sdk/workspaces/current_workspace.py

@@ -7,6 +7,7 @@
 from dataclasses import fields, make_dataclass
 from warnings import warn
 
+from openhexa.cli.api import GetCountriesWorkspaceCountries
 from openhexa.utils import stringcase
 
 from ..datasets import Dataset
@@ -59,6 +60,16 @@ def slug(self) -> str:
         except KeyError:
             raise WorkspaceConfigError("The workspace slug is not available in this environment.")
 
+    @property
+    def countries(self) -> list[GetCountriesWorkspaceCountries]:
+        """The countries of the workspace."""
+        from openhexa.cli.api import OpenHexaClient
+
+        try:
+            return OpenHexaClient().get_countries(workspace_slug=self.slug).workspace.countries
+        except KeyError:
+            raise WorkspaceConfigError("The workspace countries are not available in this environment.")
+
     @property
     def database_host(self) -> str:
         """The workspace database host."""

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "openhexa.sdk"
-version = "2.4.0"
+version = "2.5.0"
 description = "OpenHEXA SDK"
 
 authors = [{ name = "Bluesquare", email = "dev@bluesquarehub.com" }]
@@ -97,7 +97,7 @@ ignore = [
     "ANN202",
     "ANN204",
     "ANN205",
-    "ANN401",
+    "ANN401"
 ]
 
 [tool.ruff.pycodestyle]