wherobots-python-sdk 0.2.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

wherobots/__init__.py

@@ -1,6 +1,7 @@
 """Wherobots Python SDK (Jobs API surface)."""
 
 from wherobots.__version__ import __version__
+from wherobots.api.files import FilesAPI
 from wherobots.client import WherobotsJob
 from wherobots.config import WherobotsConfig
 from wherobots.enums import (
@@ -39,6 +40,7 @@ from wherobots.models import (
     RunPythonPayload,
     RunView,
     StorageIntegration,
+    TemporaryAwsCredentials,
     UtilizationStats,
 )
 
@@ -50,6 +52,7 @@ __all__ = [
     # Client
     "WherobotsJob",
     "Job",
+    "FilesAPI",
     "WherobotsConfig",
     # Enums
     "Runtime",
@@ -85,5 +88,6 @@ __all__ = [
     "PyPiDependency",
     "FileDependency",
     "StorageIntegration",
+    "TemporaryAwsCredentials",
     "UtilizationStats",
 ]

wherobots/__version__.py

@@ -1,5 +1,5 @@
 """Version information."""
 
-__version__ = "0.2.1"
+__version__ = "0.4.0"
 __author__ = "Wherobots"
 __email__ = "support@wherobots.com"

wherobots/api/base.py

@@ -2,6 +2,8 @@
 
 ``BaseClient`` wraps a ``requests.Session`` with:
 * Auth header construction (``X-API-Key``).
+* Client attribution via the advisory ``X-Wherobots-Client`` header,
+  set once on the session so every outgoing request carries it.
 * Automatic retry with exponential back-off for transient errors
   (429 Too Many Requests, 503 Service Unavailable, connection errors).
 * Unified error handling that maps HTTP failures into
@@ -11,15 +13,16 @@
 from __future__ import annotations
 
 import warnings
-from typing import Any
+from typing import Any  # noqa: TID251
 
 import requests
 from requests.adapters import HTTPAdapter
 from urllib3.util.retry import Retry
 
 from wherobots.__version__ import __version__
-from wherobots.config import WherobotsConfig
+from wherobots.config import WherobotsConfig  # noqa: TC001
 from wherobots.exceptions import WherobotsAPIError, WherobotsConfigError
+from wherobots.utils.client_header import CLIENT_HEADER, client_header_value
 from wherobots.utils.logger import get_logger
 
 logger = get_logger("api")
@@ -45,7 +48,7 @@ class BaseClient:
             retries.
     """
 
-    def __init__(
+    def __init__(  # noqa: D107
         self,
         config: WherobotsConfig,
         max_retries: int = _DEFAULT_RETRIES,
@@ -94,11 +97,15 @@ class BaseClient:
         # Intentionally not mounting on http:// — API keys must never
         # be transmitted over plaintext HTTP.
 
-        # Common headers
+        # Common headers. ``X-Wherobots-Client`` is set here, on the
+        # session, so it rides along with every request this client makes
+        # instead of having to be threaded through each call site. Auth
+        # headers are merged last so nothing can displace them.
         self._session.headers.update(
             {
                 "Content-Type": "application/json",
                 "User-Agent": f"wherobots-python-sdk/{__version__}",
+                CLIENT_HEADER: client_header_value(config.client_chain),
                 **self._auth_headers(config.api_key),
             }
         )
@@ -154,7 +161,7 @@ class BaseClient:
                 timeout=self._timeout,
             )
             response.raise_for_status()
-            return response
+            return response  # noqa: TRY300
         except requests.HTTPError as exc:
             response = exc.response
             status_code = response.status_code if response is not None else None
@@ -224,15 +231,15 @@ class BaseClient:
         self._session.close()
         self._closed = True
 
-    def __enter__(self) -> BaseClient:
+    def __enter__(self) -> BaseClient:  # noqa: PYI034
         """Enter the runtime context (context-manager protocol)."""
         return self
 
-    def __exit__(self, *exc: Any) -> None:
+    def __exit__(self, *exc: object) -> None:
         """Exit the runtime context, closing the session."""
         self.close()
 
-    def __del__(self) -> None:
+    def __del__(self) -> None:  # noqa: D105
         # Mirror ``open()``/``socket``: warn callers who let the client
         # get GC'd without closing it, so connection pools don't leak
         # silently. Guarded against partial __init__ failure and
@@ -242,7 +249,7 @@ class BaseClient:
                 return
             if getattr(self, "_session", None) is None:
                 return
-        except Exception:
+        except Exception:  # noqa: BLE001
             return
         try:
             warnings.warn(
@@ -252,5 +259,5 @@ class BaseClient:
                 stacklevel=2,
             )
             self._session.close()
-        except Exception:
+        except Exception:  # noqa: BLE001, S110
             pass