datacosmos 0.0.16__tar.gz → 0.0.17__tar.gz

{datacosmos-0.0.16 → datacosmos-0.0.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datacosmos
-Version: 0.0.16
+Version: 0.0.17
 Summary: A library for interacting with DataCosmos from Python code
 Author-email: Open Cosmos <support@open-cosmos.com>
 Classifier: Programming Language :: Python :: 3

{datacosmos-0.0.16 → datacosmos-0.0.17}/datacosmos/datacosmos_client.py

@@ -1,9 +1,10 @@
 """Client to interact with the Datacosmos API with authentication and request handling."""
 from __future__ import annotations
 
+import logging
 import threading
 from datetime import datetime, timedelta, timezone
-from typing import Any, Optional
+from typing import Any, Callable, List, Optional
 
 import requests
 from requests.exceptions import ConnectionError, HTTPError, RequestException, Timeout
@@ -21,6 +22,11 @@ from datacosmos.auth.m2m_authenticator import M2MAuthenticator
 from datacosmos.config.config import Config
 from datacosmos.exceptions.datacosmos_exception import DatacosmosException
 
+_log = logging.getLogger(__name__)
+
+RequestHook = Callable[[str, str, Any, Any], None]
+ResponseHook = Callable[[requests.Response], None]
+
 
 class DatacosmosClient:
     """Client to interact with the Datacosmos API with authentication and request handling."""
@@ -31,18 +37,24 @@ class DatacosmosClient:
         self,
         config: Optional[Config | Any] = None,
         http_session: Optional[requests.Session | OAuth2Session] = None,
+        request_hooks: Optional[List[RequestHook]] = None,
+        response_hooks: Optional[List[ResponseHook]] = None,
     ):
         """Initialize the DatacosmosClient.
 
         Args:
             config (Optional[Config]): Configuration object (only needed when SDK creates its own session).
             http_session (Optional[requests.Session]): Pre-authenticated session.
+            request_hooks (Optional[List[RequestHook]]): A list of functions to be called before each request.
+            response_hooks (Optional[List[ResponseHook]]): A list of functions to be called after each successful response.
         """
         self.config = self._coerce_config(config)
         self.token: Optional[str] = None
         self.token_expiry: Optional[datetime] = None
         self._refresh_lock = threading.Lock()
         self._authenticator: Optional[BaseAuthenticator] = None
+        self._request_hooks = request_hooks or []
+        self._response_hooks = response_hooks or []
 
         if http_session is not None:
             self._init_with_injected_session(http_session)
@@ -172,11 +184,40 @@ class DatacosmosClient:
     def request(
         self, method: str, url: str, *args: Any, **kwargs: Any
     ) -> requests.Response:
-        """Send an HTTP request using the authenticated session (with auto-refresh and retries)."""
+        """Send an HTTP request using the authenticated session (with auto-refresh and retries).
+
+        Args:
+            method (str): The HTTP method (e.g., "GET", "POST").
+            url (str): The URL for the request.
+            *args: Positional arguments for requests.request().
+            **kwargs: Keyword arguments for requests.request().
+
+        Returns:
+            requests.Response: The HTTP response.
+
+        Raises:
+            DatacosmosException: For any HTTP or request-related errors.
+        """
         self._refresh_token_if_needed()
+
+        # Call pre-request hooks
+        for hook in self._request_hooks:
+            try:
+                hook(method, url, *args, **kwargs)
+            except Exception:
+                _log.error("Request hook failed.", exc_info=True)
+
         try:
             response = self._http_client.request(method, url, *args, **kwargs)
             response.raise_for_status()
+
+            # Call post-response hooks on success
+            for hook in self._response_hooks:
+                try:
+                    hook(response)
+                except Exception:
+                    _log.error("Response hook failed.", exc_info=True)
+
             return response
         except HTTPError as e:
             status = getattr(e.response, "status_code", None)

{datacosmos-0.0.16 → datacosmos-0.0.17}/datacosmos.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datacosmos
-Version: 0.0.16
+Version: 0.0.17
 Summary: A library for interacting with DataCosmos from Python code
 Author-email: Open Cosmos <support@open-cosmos.com>
 Classifier: Programming Language :: Python :: 3

{datacosmos-0.0.16 → datacosmos-0.0.17}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "datacosmos"
-version = "0.0.16"
+version = "0.0.17"
 authors = [
     { name="Open Cosmos", email="support@open-cosmos.com" },
 ]