datamasque-python 1.5.0__py3-none-any.whl → 1.5.1__py3-none-any.whl

datamasque/client/base.py

@@ -138,6 +138,7 @@ class BaseClient:
         self.verify_ssl = connection_config.verify_ssl
         self.token_source = connection_config.token_source
         self._session = _build_session(self.verify_ssl)
+        self._timeout = (connection_config.connect_timeout_seconds, connection_config.read_timeout_seconds)
         if connection_config.spcs_pat:
             install_spcs_gateway_auth(self._session, connection_config.spcs_pat)
 
@@ -252,8 +253,11 @@ class BaseClient:
                             params=params,
                             headers=headers,
                             files=files_payload,
+                            timeout=self._timeout,
                         )
-                    return self._session.request(method, url, json=data, params=params, headers=headers)
+                    return self._session.request(
+                        method, url, json=data, params=params, headers=headers, timeout=self._timeout
+                    )
             except requests.RequestException as e:
                 raise DataMasqueTransportError(f"Failed to reach DataMasque server at {url}: {e}") from e
 

datamasque/client/ifm.py

@@ -85,6 +85,7 @@ class DataMasqueIfmClient:
         # One session for both admin-server (JWT login/refresh) and IFM (data plane)
         # traffic -- different hosts, but a single session handles per-host pooling.
         self._session = _build_session(self.verify_ssl)
+        self._timeout = (connection_config.connect_timeout_seconds, connection_config.read_timeout_seconds)
 
     def authenticate(self) -> None:
         """Obtain an access (and refresh) token from the admin server, or via `token_source`."""
@@ -101,6 +102,7 @@ class DataMasqueIfmClient:
                 response = self._session.post(
                     login_url,
                     json={"username": self.username, "password": self.password},
+                    timeout=self._timeout,
                 )
         except requests.RequestException as e:
             raise DataMasqueTransportError(f"Failed to reach admin server at {login_url}: {e}") from e
@@ -127,6 +129,7 @@ class DataMasqueIfmClient:
                 response = self._session.post(
                     refresh_url,
                     json={"refresh": self.refresh_token},
+                    timeout=self._timeout,
                 )
         except requests.RequestException as e:
             raise DataMasqueTransportError(f"Failed to reach admin server at {refresh_url}: {e}") from e
@@ -194,6 +197,7 @@ class DataMasqueIfmClient:
                         json=json_body,
                         params=params,
                         headers={"Authorization": f"Bearer {self.access_token}"},
+                        timeout=self._timeout,
                     )
             except requests.RequestException as e:
                 raise DataMasqueTransportError(f"Failed to reach IFM server at {url}: {e}") from e

datamasque/client/models/dm_instance.py

@@ -4,6 +4,9 @@ from pydantic import BaseModel, ConfigDict, model_validator
 
 from datamasque.client.exceptions import DataMasqueUserError
 
+DEFAULT_CONNECT_TIMEOUT_SECONDS = 10.0
+DEFAULT_READ_TIMEOUT_SECONDS = 300.0
+
 
 class DataMasqueInstanceConfig(BaseModel):
     """
@@ -42,6 +45,10 @@ class DataMasqueInstanceConfig(BaseModel):
     Create the token in Snowsight (User profile → Programmatic access tokens)
     for an account that can reach the SPCS app.
     Leave unset for instances that are not hosted on SPCS."""
+    connect_timeout_seconds: float = DEFAULT_CONNECT_TIMEOUT_SECONDS
+    """How long a request may take to connect, including the TLS handshake."""
+    read_timeout_seconds: float = DEFAULT_READ_TIMEOUT_SECONDS
+    """How long a request may wait for the server between bytes of its response."""
 
     @model_validator(mode="after")
     def _validate_auth_source(self) -> "DataMasqueInstanceConfig":

datamasque/client/models/ifm.py

@@ -6,6 +6,7 @@ from typing import Any, Callable, Optional
 from pydantic import BaseModel, ConfigDict, model_validator
 
 from datamasque.client.exceptions import DataMasqueUserError
+from datamasque.client.models.dm_instance import DEFAULT_CONNECT_TIMEOUT_SECONDS, DEFAULT_READ_TIMEOUT_SECONDS
 
 
 class RulesetPlanOptions(BaseModel):
@@ -167,6 +168,10 @@ class DataMasqueIfmInstanceConfig(BaseModel):
     password: Optional[str] = None
     verify_ssl: bool = True
     token_source: Optional[Callable[[], str]] = None
+    connect_timeout_seconds: float = DEFAULT_CONNECT_TIMEOUT_SECONDS
+    """How long a request may take to connect, including the TLS handshake."""
+    read_timeout_seconds: float = DEFAULT_READ_TIMEOUT_SECONDS
+    """How long a request may wait for the server between bytes of its response."""
 
     @model_validator(mode="after")
     def _validate_auth_source(self) -> "DataMasqueIfmInstanceConfig":

{datamasque_python-1.5.0.dist-info → datamasque_python-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: datamasque-python
-Version: 1.5.0
+Version: 1.5.1
 Summary: Official Python client for the DataMasque data-masking API.
 Project-URL: Homepage, https://datamasque.com/
 Project-URL: Documentation, https://datamasque-python.readthedocs.io/

{datamasque_python-1.5.0.dist-info → datamasque_python-1.5.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 datamasque/client/__init__.py,sha256=E763HBSVk3NOMI-sAchHwwAB1eG40A_6FN6vixNVJ4s,9021
-datamasque/client/base.py,sha256=cXU4dluL9e1N1F2pV48iIvebv6OHt7Kh4Ka2y9vkOPo,14296
+datamasque/client/base.py,sha256=VMMn34yQRLZA44MHWrJe1pFLLRF-Z3LaA1gVtOiOyWk,14524
 datamasque/client/connections.py,sha256=EFinx8fJRme0mTxuWY3d29UnmUFbsQhMaUQT0Ma2PK4,2885
 datamasque/client/discovery.py,sha256=xidiAj5LdVSGQy6pjta_2Z_aJmuPPohblyNDfua_Mxk,28859
 datamasque/client/discovery_config_libraries.py,sha256=K6kFi7kL4SGIPKNIg1TBMIwHUeyfXI3Lyg5sknMMCD4,7285
@@ -7,7 +7,7 @@ datamasque/client/discovery_configs.py,sha256=-iSDi7g8PAYXJ8aGCusGOVtE1UWD4D_pNx
 datamasque/client/dmclient.py,sha256=0W6s26gUl9EZRL1kdk0R_7hgZSAOesY3H5wd-tN-QIE,1940
 datamasque/client/exceptions.py,sha256=3JI_LU8K3P2BCJDjGewzj3yg6oMd_7KuQ-zwSLR8Bfs,3956
 datamasque/client/files.py,sha256=vc2tZgT8RdwuCObIQ7H-Qjb80H1PBFqD0nG5Y1tOA_o,3503
-datamasque/client/ifm.py,sha256=uIMxpLIPvDiDO1m4bxezixNIUFIFY0MXWRMQNvTbpTA,11858
+datamasque/client/ifm.py,sha256=KobUTxKvjDzQFPvwr7Ei3S1-UjDquSKVY0K7s9Mic1w,12099
 datamasque/client/license.py,sha256=pluYaSU168OC6_laB9bM9H3Vuuxs8wa9gujCJvtoJCk,1392
 datamasque/client/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datamasque/client/rg_configs.py,sha256=jcrbw-oMhj6o3Wsf8xy3peVo7DzeQ8EhhAsfcEwC3lc,5816
@@ -24,10 +24,10 @@ datamasque/client/models/data_selection.py,sha256=9hRpkQQIQlKGFYDYQCrreggsv96dh6
 datamasque/client/models/discovery.py,sha256=BvAK9zj0jwUc1O4DSluMcjeoYhF4HKOkJmO4uAhzet4,18222
 datamasque/client/models/discovery_config.py,sha256=8sqs9x_X0H_hR1fJDEFmOdJyGRDwgbdb3eddEAe98WE,2420
 datamasque/client/models/discovery_config_library.py,sha256=m0wI51Q4aHysLNHyNTWH3KrDLV_rxtoLNeKQsF9RYFk,1436
-datamasque/client/models/dm_instance.py,sha256=6h0WNlI1qt6jkrYFapIvuSJFkhRCcwbpx0PLG5NsrGw,2219
+datamasque/client/models/dm_instance.py,sha256=Je1Ld66HxY_V7xFZxFSeqPE9ALg3VTpU0VNPVWcG-7w,2591
 datamasque/client/models/files.py,sha256=7hG3Q1-XYb1PF88l3ppsmBp2tjNtvQvv-RNTrOa7Nts,2784
 datamasque/client/models/git.py,sha256=F0KdMFe2vMyL-9tp3SGKGeOlbD1gbcjQn0jpOzNzybc,1782
-datamasque/client/models/ifm.py,sha256=j0Ef2BZYTk6MNZO6HS6mW0qWs1_wWD9PlUP_Y6WyBQI,5563
+datamasque/client/models/ifm.py,sha256=1IVQfJsw3NOqK8vgVVcdtWdJHkw6IZI3PMlhkvwBNI0,5969
 datamasque/client/models/license.py,sha256=OqIn4Sx3ATBStgt5KNiCOBChTTFYLBsyGj7inAl8oB4,2113
 datamasque/client/models/pagination.py,sha256=egg9aO2cf6KUDwDANPu5RxpJbdRKdwUdCQAtusnuf1c,651
 datamasque/client/models/rg_config.py,sha256=g6g0qMTeR6OcM4Ji8fHrDOnOM5fsz1UYxntn_mv_k88,1792
@@ -38,7 +38,7 @@ datamasque/client/models/safe_data_preview.py,sha256=BiI9d13KdQrTin2JfbjiRiODjns
 datamasque/client/models/status.py,sha256=DXcucL_jRsOQ0vaV0t8CJt6crvCIOLBn07KuwpwW-4A,3247
 datamasque/client/models/table_reference.py,sha256=55fKdHjC2TQnUmbp0XiWqfeFGXqcq4wXjbWgMaN3p3c,3835
 datamasque/client/models/user.py,sha256=UGAUzgJkf78m24_zFXXoA99zdut48BXkX_ivV8yq1Vc,2043
-datamasque_python-1.5.0.dist-info/METADATA,sha256=8PhvPpCLdPzElNVWl_4CBcBFlTDsPiwly09VSmvoH6s,4497
-datamasque_python-1.5.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
-datamasque_python-1.5.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-datamasque_python-1.5.0.dist-info/RECORD,,
+datamasque_python-1.5.1.dist-info/METADATA,sha256=vHITd4Mix_f1Q4bX8jsWolqQzhsQ_8hViMUlrlEZ4Uc,4497
+datamasque_python-1.5.1.dist-info/WHEEL,sha256=2iZrJkDIkeDh0k9SikKMqFuRP-ua6mb4Ir1AsER4BEw,87
+datamasque_python-1.5.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+datamasque_python-1.5.1.dist-info/RECORD,,

{datamasque_python-1.5.0.dist-info → datamasque_python-1.5.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.32.4
+Generator: hatchling 1.33.0
 Root-Is-Purelib: true
 Tag: py3-none-any