external-systems 0.105.0__py3-none-any.whl → 0.106.0__py3-none-any.whl

external_systems/_version.py

@@ -16,4 +16,4 @@
 # The version is set during the publishing step (since we can't know the version in advance)
 # using the autorelease bot
 
-__version__ = "0.105.0"
+__version__ = "0.106.0"

external_systems/sources/__init__.py

@@ -16,6 +16,7 @@ from ._api import (
     AwsCredentials,
     ClientCertificate,
     ClientCertificateFilePaths,
+    GcpOauthCredentials,
     HttpsConnectionParameters,
     SourceCredentials,
     SourceParameters,
@@ -27,6 +28,7 @@ from ._sources import Source
 __all__ = [
     "ClientCertificate",
     "ClientCertificateFilePaths",
+    "GcpOauthCredentials",
     "HttpsConnection",
     "HttpsConnectionParameters",
     "Source",

external_systems/sources/_api.py

@@ -66,8 +66,14 @@ class AwsCredentials:
     expiration: Optional[datetime] = None
 
 
+@dataclass(frozen=True)
+class GcpOauthCredentials:
+    access_token: str
+    expiration: Optional[datetime] = None
+
+
 # Each new credential type must include an expiration field
-SourceCredentials = Union[AwsCredentials]
+SourceCredentials = Union[AwsCredentials, GcpOauthCredentials]
 
 
 @dataclass(frozen=True)

external_systems/sources/_sources.py

@@ -16,9 +16,10 @@ import logging
 import os
 import random
 import socket
+import warnings
 from functools import cached_property
 from tempfile import NamedTemporaryFile
-from typing import Any, Mapping, Optional, Tuple
+from typing import Any, Mapping, Optional, Tuple, cast
 
 import urllib3.util
 from frozendict import frozendict
@@ -192,15 +193,29 @@ class Source:
 
     def get_aws_credentials(self) -> Refreshable[AwsCredentials]:
         """
-        Get the AWS credentials from the Source.
+        DEPRECATED: Use get_session_credentials instead.
+        """
+        warnings.warn(
+            "get_aws_credentials is deprecated and will be removed in a future release. "
+            "Use get_session_credentials instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        if self._maybe_refreshable_resolved_source_credentials is None:
+            raise ValueError("Resolved source credentials are not present on the Source.")
+        if not isinstance(self._maybe_refreshable_resolved_source_credentials.get(), AwsCredentials):
+            raise ValueError("Resolved source credentials are not of type AwsCredentials.")
+        return cast(Refreshable[AwsCredentials], self._maybe_refreshable_resolved_source_credentials)
 
+    def get_session_credentials(self) -> Refreshable[SourceCredentials]:
+        """
         Supported Sources:
             - S3
+            - BigQuery
+            - Google Cloud Storage
         """
         if self._maybe_refreshable_resolved_source_credentials is None:
             raise ValueError("Resolved source credentials are not present on the Source.")
-        if not isinstance(self._maybe_refreshable_resolved_source_credentials.get(), AwsCredentials):
-            raise ValueError("Resolved source credentials are not of type AwsCredentials.")
         return self._maybe_refreshable_resolved_source_credentials
 
     def get_secret(self, key: str) -> str:

{external_systems-0.105.0.dist-info → external_systems-0.106.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: external-systems
-Version: 0.105.0
+Version: 0.106.0
 Summary: A Python library for interacting with Foundry Sources
 License: Apache-2.0
 Keywords: Palantir,Foundry,Sources,Compute Modules,Python Functions,Transforms
@@ -72,18 +72,21 @@ my_source: Source = ...
 some_secret: str = my_source.get_secret("SECRET_NAME")
 ```
 
-For sources using session credentials we support credentials generation and refresh management. Currently on an S3 source you can access session credentials using `get_aws_credentials()`. This method will throw if the source is not pre-configured with `AwsCredentials`
+For sources using session credentials we support credentials generation and refresh management. This can be done by using `get_session_credentials` which supports `S3`, `BigQuery`, `Google Cloud Storage` sources.
 
 _Session credentials may not be available in all Foundry runtime environments_
 
 ```python
-from external_systems.sources import Source, Refreshable, AwsCredentials
+from external_systems.sources import Source, Refreshable, SourceCredentials, AwsCredentials
 
 s3_source: Source = ...
 
-refreshable_credentials: Refreshable[AwsCredentials] = s3_source.get_aws_credentials()
+refreshable_credentials: Refreshable[SourceCredentials] = s3_source.get_session_credentials()
 
-session_credentials: AwsCredentials = refreshable_credentials.get()
+session_credentials: SourceCredentials = refreshable_credentials.get()
+
+if not isinstance(session_credentials, AwsCredentials):
+    raise ...
 ```
 
 ## On-prem Connectivity with [Foundry Agent Proxy](https://www.palantir.com/docs/foundry/data-connection/agent-proxy-runtime)

{external_systems-0.105.0.dist-info → external_systems-0.106.0.dist-info}/RECORD

@@ -1,15 +1,15 @@
 external_systems/__init__.py,sha256=xXDUDD6_qRO-nHuXZx-fXp0R0vc3N_OOsB1F5mF_kpU,651
-external_systems/_version.py,sha256=-N277YomEI4sXteddhJUup5RsSUv1pQb8O-9AAtDiQQ,749
+external_systems/_version.py,sha256=XybDtttIh2RAnNN-6aMq9sh0VI5lwaASQ2EA1--tSg0,749
 external_systems/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-external_systems/sources/__init__.py,sha256=aqXbMIy_pnyIC1uPRzQfApLCbhYB4N8iRFnpOX4RdAk,1156
-external_systems/sources/_api.py,sha256=NV7oNIgzSWz4ROFW8uPpUJjDN5vfFzdTs3yKC37S39k,3262
+external_systems/sources/__init__.py,sha256=QGbNcUcv8QhVJwYraMV6Q3Uy3GgZdirA8KYl6Sss2tk,1208
+external_systems/sources/_api.py,sha256=7hUwW5O4v36jZnBKerHgRChSyrvHeJc4511oYBZRrGk,3400
 external_systems/sources/_connections.py,sha256=h82npEks29NALAAfMHrXcSnB7TY_OLeXgL3QN9Cssus,2509
 external_systems/sources/_proxies.py,sha256=hwgMBpOUdlhyRLgCyXE7v9X5v-1a655PDFqHJtZ9c74,6576
 external_systems/sources/_refreshable.py,sha256=0pa5XW0_2gTMW-ZymKhlhh3Kka_bWTWffThKvrTTifc,4421
 external_systems/sources/_sockets.py,sha256=XH51adVnloqg8XYVHPxl6K8R_q2BCh6WZ77cq1L8nRg,4473
-external_systems/sources/_sources.py,sha256=DlzoLXYtjLct1CC3h86-sfEwph5CuZe54wbhX98D9p8,11100
+external_systems/sources/_sources.py,sha256=2sranCdLgb_IO8pHMPqwoGqWfmECiMro96piN6iVaZ4,11783
 external_systems/sources/_utils.py,sha256=EmUKKiKRDOZ2cZs7FZ4qCSan_lq1K_tquh3kxleB-jA,1004
-external_systems-0.105.0.dist-info/LICENSE.txt,sha256=NAk6Uc9K_N_J5V75k9qECpzUnO-ujT-mKK_jk_mboUE,569
-external_systems-0.105.0.dist-info/METADATA,sha256=y7gqCMh6BgeH4672a9ZIW0ynAagT8YBNnKna7lxB38g,4765
-external_systems-0.105.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-external_systems-0.105.0.dist-info/RECORD,,
+external_systems-0.106.0.dist-info/LICENSE.txt,sha256=NAk6Uc9K_N_J5V75k9qECpzUnO-ujT-mKK_jk_mboUE,569
+external_systems-0.106.0.dist-info/METADATA,sha256=GohOFkVFmTYZ94j3folThj4TVBRKbPTEbbFYpR2UHDg,4809
+external_systems-0.106.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+external_systems-0.106.0.dist-info/RECORD,,

{external_systems-0.105.0.dist-info → external_systems-0.106.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.2
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any