wandelbots_api_client 26.6.0.dev43__py3-none-any.whl → 26.6.0.dev45__py3-none-any.whl

wandelbots_api_client/__init__.py

@@ -1,48 +0,0 @@
-# coding: utf-8
-
-# flake8: noqa
-
-"""
-Wandelbots NOVA API
-
-Interact with robots in an easy and intuitive way.
-
-The version of the OpenAPI document: 1.6.0 dev
-Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-Do not edit the class manually.
-"""  # noqa: E501
-
-__version__ = "26.6.0.dev43"
-
-from . import models
-from . import api
-from . import api_client
-from . import exceptions
-from . import configuration
-
-from .api import *
-
-# import ApiClient
-from .api_response import ApiResponse
-from .api_client import ApiClient
-from .configuration import Configuration
-from .exceptions import OpenApiException
-from .exceptions import ApiTypeError
-from .exceptions import ApiValueError
-from .exceptions import ApiKeyError
-from .exceptions import ApiAttributeError
-from .exceptions import ApiException
-
-
-__all__ = [
-    "ApiResponse",
-    "ApiClient",
-    "Configuration",
-    "OpenApiException",
-    "ApiTypeError",
-    "ApiValueError",
-    "ApiKeyError",
-    "ApiAttributeError",
-    "ApiException",
-]

wandelbots_api_client/authorization.py

@@ -1,243 +1,238 @@
-from typing import Optional
+"""OAuth2 Device Code Flow implementation for Auth0 and Entra ID."""
 
-import asyncio
 import aiohttp
+import asyncio
+from typing import Any
+from abc import ABC, abstractmethod
 import pydantic
 
 
-class Auth0Config(pydantic.BaseModel):
-    """Configuration for Auth0 authentication"""
+class DeviceCodeFlowConfig(pydantic.BaseModel, ABC):
+    """Base class for OAuth2 device code flow authentication."""
 
-    domain: Optional[str] = None
-    client_id: Optional[str] = None
-    audience: Optional[str] = None
+    scope: str = "openid profile email offline_access"
 
-    @classmethod
-    def default(cls) -> "Auth0Config":
-        """Create Auth0Config from environment variables"""
-        return cls(
-            domain="",
-            client_id="",
-            audience="",
-        )
+    class Config:
+        arbitrary_types_allowed = True
 
+    @abstractmethod
     def is_complete(self) -> bool:
-        """Check if all required fields are set and not None"""
-        return bool(self.domain and self.client_id and self.audience)
-
-    def get_validated_config(self) -> tuple[str, str, str]:
-        """Get validated config values, ensuring they are not None"""
-        if not self.is_complete():
-            raise ValueError("Auth0 configuration is incomplete")
-        return self.domain, self.client_id, self.audience
-
-
-class Auth0DeviceCodeInfo(pydantic.BaseModel):
-    """
-    Model to store device code information.
-
-    Attributes:
-        device_code (str): The device code.
-        user_code (str): The user code.
-        verification_uri (str): The verification URI.
-        expires_in (int): The expiration time in seconds.
-        interval (int): The interval time in seconds (default is 5).
-    """
-
-    device_code: str
-    user_code: str
-    verification_uri: str
-    expires_in: int
-    interval: int = pydantic.Field(default=5)
-
-
-class Auth0TokenInfo(pydantic.BaseModel):
-    """
-    Model to store token information.
-
-    Attributes:
-        access_token (str): The access token.
-        refresh_token (str, optional): The refresh token.
-    """
-
-    access_token: str
-    refresh_token: Optional[str] = None
-
-
-class Auth0DeviceAuthorization:
-    """
-    Class to handle Auth0 device authorization.
-
-    Methods:
-        __init__(auth0_config: Auth0Config | None = None):
-            Initializes the Auth0DeviceAuthorization instance with the given parameters.
-        request_device_code():
-            Requests a device code from Auth0.
-        display_user_instructions():
-            Displays instructions for the user to authenticate.
-        poll_token_endpoint():
-            Polls the token endpoint to obtain an access token.
-        refresh_access_token(refresh_token: str):
-            Refreshes the access token using the refresh token.
-    """
-
-    def __init__(self, auth0_config: Optional[Auth0Config] = None):
+        """Check if all required fields are set."""
+        pass
+
+    @property
+    @abstractmethod
+    def token_endpoint(self) -> str:
+        """Get the token endpoint URL."""
+        pass
+
+    @property
+    @abstractmethod
+    def device_endpoint(self) -> str:
+        """Get the device code endpoint URL."""
+        pass
+
+    @abstractmethod
+    def _build_device_code_request(self) -> dict[str, str]:
+        """Build the device code request data."""
+        pass
+
+    async def request_device_code(self) -> dict[str, Any]:
         """
-        Initialize with Auth0Config from env vars or passed config.
-
-        Args:
-            auth0_config: Optional Auth0Config object. If not provided,
-                         will be created using default configuration.
-        """
-        try:
-            self.params = auth0_config or Auth0Config.default()
-            if not self.params.is_complete():
-                raise ValueError("Auth0 configuration is incomplete")
-
-            domain, client_id, audience = self.params.get_validated_config()
-            self.auth0_domain = domain
-            self.auth0_client_id = client_id
-            self.auth0_audience = audience
-
-        except ValueError as e:
-            raise ValueError(f"Error: Auth0 configuration is invalid: {e}")
-
-        self.headers = {"content-type": "application/x-www-form-urlencoded"}
-        self.device_code_info: Optional[Auth0DeviceCodeInfo] = None
-        self.refresh_token: Optional[str] = None
-        self.interval = 5
-        self.attempts = 10
-
-    async def request_device_code(self) -> Auth0DeviceCodeInfo:
-        """
-        Requests a device code from Auth0.
+        Request device code from the authorization server.
 
         Returns:
-            Auth0DeviceCodeInfo: The device code information.
+            Device code response with verification_uri, user_code, etc.
 
         Raises:
-            Exception: If there is an error requesting the device code.
-        """
-        device_code_url = f"https://{self.auth0_domain}/oauth/device/code"
-        data = {
-            "client_id": self.auth0_client_id,
-            "scope": "openid profile email",
-            "audience": self.auth0_audience,
-        }
-
-        async with aiohttp.ClientSession() as session:
-            async with session.post(device_code_url, headers=self.headers, data=data) as response:
-                if response.status == 200:
-                    response_data = await response.json()
-                    self.device_code_info = Auth0DeviceCodeInfo(**response_data)
-                    self.interval = self.device_code_info.interval
-                    return self.device_code_info
-                else:
-                    error_data = await response.json()
-                    raise Exception("Error requesting device code:", error_data)
-
-    def get_device_code_info(self) -> Optional[Auth0DeviceCodeInfo]:
+            ValueError: If configuration is incomplete
+            Exception: If device code request fails
         """
-        Returns the device code information.
+        if not self.is_complete():
+            raise ValueError(f"{self.__class__.__name__} configuration is incomplete")
 
-        Returns:
-            Optional[Auth0DeviceCodeInfo]: The device code information.
-        """
-        return self.device_code_info
+        data = self._build_device_code_request()
 
-    def display_user_instructions(self) -> None:
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                self.device_endpoint,
+                data=data,
+                headers={"Content-Type": "application/x-www-form-urlencoded"},
+            ) as response:
+                if response.status != 200:
+                    error_text = await response.text()
+                    raise Exception(f"Device code request failed: {error_text}")
+
+                return await response.json()
+
+    async def poll_token_endpoint(self, device_code: str, interval: int = 5, expires_in: int = 900) -> dict[str, str]:
         """
-        Displays instructions for the user to authenticate.
+        Poll the token endpoint until authorization is complete.
 
-        Raises:
-            Exception: If device code information is not available.
-        """
-        if self.device_code_info:
-            verification_uri = f"{self.device_code_info.verification_uri}?user_code={self.device_code_info.user_code}"
-            user_code = self.device_code_info.user_code
-            print(f"Please visit {verification_uri} and validate the code {user_code} to authenticate.")
-        else:
-            raise Exception("Device code information is not available.")
-
-    async def poll_token_endpoint(self):
-        """
-        Polls the token endpoint to obtain an access token.
+        Args:
+            device_code: Device code from request_device_code
+            interval: Polling interval in seconds
+            expires_in: Expiration time in seconds
 
         Returns:
-            str: The access token.
+            Token response with access_token, refresh_token, etc.
 
         Raises:
-            Exception: If there is an error polling the token endpoint.
+            TimeoutError: If authorization times out
+            Exception: If authorization fails
         """
-        if not self.device_code_info:
-            raise Exception("Device code information is not available.")
-
-        token_url = f"https://{self.auth0_domain}/oauth/token"
-        token_data = {
-            "grant_type": "urn:ietf:params:oauth:grant-type:device_code",
-            "device_code": self.device_code_info.device_code,
-            "client_id": self.auth0_client_id,
-        }
+        start_time = asyncio.get_event_loop().time()
+
+        while (asyncio.get_event_loop().time() - start_time) < expires_in:
+            await asyncio.sleep(interval)
+
+            async with aiohttp.ClientSession() as session:
+                async with session.post(
+                    self.token_endpoint,
+                    data={
+                        "grant_type": "urn:ietf:params:oauth:grant-type:device_code",
+                        "device_code": device_code,
+                        "client_id": self._get_client_id(),
+                    },
+                    headers={"Content-Type": "application/x-www-form-urlencoded"},
+                ) as response:
+                    if response.status == 200:
+                        return await response.json()
 
-        async with aiohttp.ClientSession() as session:
-            while self.attempts > 0:
-                self.attempts = self.attempts - 1
-                async with session.post(token_url, headers=self.headers, data=token_data) as token_response:
-                    if token_response.status == 200:
-                        # If the response status is 200, it means the access token is successfully obtained.
-                        response_data = await token_response.json()
-                        token_info = Auth0TokenInfo(**response_data)
-                        self.refresh_token = token_info.refresh_token
-                        return token_info.access_token
-                    elif token_response.status == 400:
-                        # If the response status is 400, check the error type.
-                        error_data = await token_response.json()
-                        error = error_data.get("error")
-                        if error == "authorization_pending":
-                            # If the error is 'authorization_pending', it means the user has not yet authorized.
-                            await asyncio.sleep(self.interval)
-                        elif error == "slow_down":
-                            # If the error is 'slow_down', it means the server requests to slow down polling.
-                            self.interval += 5
-                            await asyncio.sleep(self.interval)
-                    elif token_response.status == 403:
-                        # If the response status is 403, it means the request is forbidden, wait and retry.
-                        await asyncio.sleep(self.interval)
+                    error_data = await response.json()
+                    error = error_data.get("error")
+
+                    if error == "authorization_pending":
+                        continue
+                    elif error == "slow_down":
+                        interval += 5
+                        continue
                     else:
-                        # For other status codes, raise an exception with the error details.
-                        error_data = await token_response.json()
-                        raise Exception("Error:", token_response.status, error_data)
-        raise Exception("Error: It was not able to authenticate. Please try again.")
+                        raise Exception(f"Authorization failed: {error}")
+
+        raise TimeoutError("Device authorization timed out")
 
-    async def refresh_access_token(self, refresh_token: str):
+    async def refresh_token(self, refresh_token: str) -> dict[str, str]:
         """
-        Refreshes the access token using the refresh token.
+        Refresh access token using refresh token.
 
         Args:
-            refresh_token (str): The refresh token.
+            refresh_token: Refresh token from previous authentication
 
         Returns:
-            str: The new access token.
+            New token response with access_token and potentially new refresh_token
 
         Raises:
-            Exception: If there is an error refreshing the access token.
+            Exception: If token refresh fails
         """
-        if not refresh_token:
-            raise Exception("Refresh token is not available.")
-
-        token_url = f"https://{self.auth0_domain}/oauth/token"
-        token_data = {
-            "grant_type": "refresh_token",
-            "client_id": self.auth0_client_id,
-            "refresh_token": refresh_token,
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                self.token_endpoint,
+                data={
+                    "grant_type": "refresh_token",
+                    "refresh_token": refresh_token,
+                    "client_id": self._get_client_id(),
+                },
+                headers={"Content-Type": "application/x-www-form-urlencoded"},
+            ) as response:
+                if response.status != 200:
+                    error_text = await response.text()
+                    raise Exception(f"Token refresh failed: {error_text}")
+
+                return await response.json()
+
+    @abstractmethod
+    def _get_client_id(self) -> str:
+        """Get the client ID for the configuration."""
+        pass
+
+
+class Auth0Config(DeviceCodeFlowConfig):
+    """Configuration for Auth0 device code flow authentication."""
+
+    domain: str | None = None
+    client_id: str | None = None
+    audience: str | None = None
+
+    @classmethod
+    def default(cls) -> "Auth0Config":
+        """Provide the default Auth0 configuration."""
+        return cls(
+            domain="",
+            client_id="",
+            audience="",
+        )
+
+    def is_complete(self) -> bool:
+        """Check if all required fields are set."""
+        return bool(self.domain and self.client_id and self.audience)
+
+    @property
+    def token_endpoint(self) -> str:
+        """Get the token endpoint URL."""
+        if not self.domain:
+            raise ValueError("Domain is not set")
+        return f"https://{self.domain}/oauth/token"
+
+    @property
+    def device_endpoint(self) -> str:
+        """Get the device code endpoint URL."""
+        if not self.domain:
+            raise ValueError("Domain is not set")
+        return f"https://{self.domain}/oauth/device/code"
+
+    def _get_client_id(self) -> str:
+        """Get the client ID for the configuration."""
+        if not self.client_id:
+            raise ValueError("Client ID is not set")
+        return self.client_id
+
+    def _build_device_code_request(self) -> dict[str, str]:
+        """Build the device code request data."""
+        data = {
+            "client_id": self.client_id,
+            "scope": self.scope,
         }
+        if self.audience:
+            data["audience"] = self.audience
+        return data
 
-        async with aiohttp.ClientSession() as session:
-            async with session.post(token_url, headers=self.headers, data=token_data) as response:
-                if response.status == 200:
-                    response_data = await response.json()
-                    token_info = Auth0TokenInfo(**response_data)
-                    return token_info.access_token
-                else:
-                    error_data = await response.json()
-                    raise Exception("Error refreshing access token:", error_data)
+
+class EntraIDConfig(DeviceCodeFlowConfig):
+    """Configuration for Microsoft Entra ID (Azure AD) device code flow authentication."""
+
+    client_id: str | None = None
+    tenant_id: str | None = None
+
+    def is_complete(self) -> bool:
+        """Check if all required fields are set."""
+        return bool(self.client_id and self.tenant_id)
+
+    @property
+    def token_endpoint(self) -> str:
+        """Get the token endpoint URL."""
+        if not self.tenant_id:
+            raise ValueError("Tenant ID is not set")
+        return f"https://login.microsoftonline.com/{self.tenant_id}/oauth2/v2.0/token"
+
+    @property
+    def device_endpoint(self) -> str:
+        """Get the device code endpoint URL."""
+        if not self.tenant_id:
+            raise ValueError("Tenant ID is not set")
+        return f"https://login.microsoftonline.com/{self.tenant_id}/oauth2/v2.0/devicecode"
+
+    def _get_client_id(self) -> str:
+        """Get the client ID for the configuration."""
+        if not self.client_id:
+            raise ValueError("Client ID is not set")
+        return self.client_id
+
+    def _build_device_code_request(self) -> dict[str, str]:
+        """Build the device code request data."""
+        if not self.client_id:
+            raise ValueError("Client ID is not set")
+        return {
+            "client_id": self.client_id,
+            "scope": self.scope,
+        }

wandelbots_api_client/v2/__init__.py

@@ -13,7 +13,7 @@ Generated by OpenAPI Generator (https://openapi-generator.tech)
 Do not edit the class manually.
 """  # noqa: E501
 
-__version__ = "26.6.0.dev43"
+__version__ = "26.6.0.dev45"
 
 from . import api
 from . import api_client

wandelbots_api_client/v2/api_client.py

@@ -85,7 +85,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = "Wandelbots-Nova-API-Python-Client/26.6.0.dev43"
+        self.user_agent = "Wandelbots-Nova-API-Python-Client/26.6.0.dev45"
         self.client_side_validation = configuration.client_side_validation
 
     async def __aenter__(self):

wandelbots_api_client/v2/configuration.py

@@ -512,7 +512,7 @@ class Configuration:
             "OS: {env}\n"
             "Python Version: {pyversion}\n"
             "Version of the API: 2.6.0 dev\n"
-            "SDK Package Version: 26.6.0.dev43".format(env=sys.platform, pyversion=sys.version)
+            "SDK Package Version: 26.6.0.dev45".format(env=sys.platform, pyversion=sys.version)
         )
 
     def get_host_settings(self) -> List[HostSetting]:

wandelbots_api_client/v2_pydantic/__init__.py

@@ -13,7 +13,7 @@ Generated by OpenAPI Generator (https://openapi-generator.tech)
 Do not edit the class manually.
 """  # noqa: E501
 
-__version__ = "26.6.0.dev43"
+__version__ = "26.6.0.dev45"
 
 from . import api
 from . import api_client

wandelbots_api_client/v2_pydantic/api_client.py

@@ -85,7 +85,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = "Wandelbots-Nova-API-Python-Client/26.6.0.dev43"
+        self.user_agent = "Wandelbots-Nova-API-Python-Client/26.6.0.dev45"
         self.client_side_validation = configuration.client_side_validation
 
     async def __aenter__(self):

wandelbots_api_client/v2_pydantic/configuration.py

@@ -512,7 +512,7 @@ class Configuration:
             "OS: {env}\n"
             "Python Version: {pyversion}\n"
             "Version of the API: 2.6.0 dev\n"
-            "SDK Package Version: 26.6.0.dev43".format(env=sys.platform, pyversion=sys.version)
+            "SDK Package Version: 26.6.0.dev45".format(env=sys.platform, pyversion=sys.version)
         )
 
     def get_host_settings(self) -> List[HostSetting]:

wandelbots_api_client/v2_pydantic/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmpfmlj1vyj
-#   timestamp: 2026-06-30T11:22:54+00:00
+#   filename:  tmpr5wlf014
+#   timestamp: 2026-06-30T16:55:31+00:00
 
 from __future__ import annotations
 

wandelbots_api_client/v2_pydantic/models/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  models.yaml
-#   timestamp: 2026-06-30T11:22:54+00:00
+#   timestamp: 2026-06-30T16:55:31+00:00
 
 from __future__ import annotations
 from pydantic import AnyUrl, AwareDatetime, BaseModel, EmailStr, Field, RootModel

wandelbots_api_client-26.6.0.dev45.dist-info/METADATA

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: wandelbots_api_client
+Version: 26.6.0.dev45
+Summary: Wandelbots Python Client: Interact with robots in an easy and intuitive way. 
+Author: Copyright (c) 2025 Wandelbots GmbH
+Author-email: Copyright (c) 2025 Wandelbots GmbH <contact@wandelbots.com>
+License-Expression: Apache-2.0
+Classifier: Typing :: Typed
+Classifier: Topic :: Software Development
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX
+Classifier: Operating System :: Unix
+Classifier: Operating System :: MacOS
+Requires-Dist: urllib3>=1.25.3
+Requires-Dist: python-dateutil>=2.8.2
+Requires-Dist: aiohttp>=3.8.4
+Requires-Dist: aiohttp-retry>=2.8.3
+Requires-Dist: pydantic>=2
+Requires-Dist: websockets
+Requires-Dist: typing-extensions>=4.7.1
+Requires-Dist: furl
+Requires-Dist: pyhumps
+Requires-Dist: pydantic-core
+Requires-Dist: annotated-types
+Requires-Dist: pydantic[email]
+Requires-Python: >=3.9
+Project-URL: homepage, https://wandelbots.com/
+Description-Content-Type: text/markdown
+
+# wandelbots-api-client.v2
+Interact with robots in an easy and intuitive way.
+
+- Compatible API version: 2.6.0 dev (can be found at the home screen of your instance -> API)
+- Package version: 26.6.0.dev45
+
+## Requirements.
+Python >=3.9, Python < 4.0
+
+## Installation
+```
+pip install wandelbots-api-client
+```
+
+OR
+
+pyproject.toml:
+```
+[tool.poetry.dependencies]
+wandelbots-api-client = "*"
+```
+
+