external-systems 0.104.0rc1__tar.gz → 0.106.0__tar.gz

{external_systems-0.104.0rc1 → external_systems-0.106.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: external-systems
-Version: 0.104.0rc1
+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.104.0rc1 → external_systems-0.106.0}/README.md

@@ -52,18 +52,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.104.0rc1 → external_systems-0.106.0}/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.104.0-rc1"
+__version__ = "0.106.0"

{external_systems-0.104.0rc1 → external_systems-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-0.104.0rc1 → external_systems-0.106.0}/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-0.104.0rc1 → external_systems-0.106.0}/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
@@ -75,13 +76,20 @@ class Source:
     def _https_connections(self) -> Mapping[str, HttpsConnection]:
         return frozendict(
             {
-                key: HttpsConnection(params, self._client_certificate, self._https_proxy_url, self._ca_bundle_path)
+                key: HttpsConnection(
+                    params, self._client_certificate, self._https_proxy_url, self.server_certificates_bundle_path
+                )
                 for key, params in self._source_parameters.https_connections.items()
             }
         )
 
-    @cached_property
-    def _ca_bundle_path(self) -> Optional[str]:
+    @property
+    def server_certificates_bundle_path(self) -> Optional[str]:
+        """
+        File path to the CA bundle file containing all server certificates required by the Source.
+        If no server certificates are defined on the Source, this will return None.
+        """
+
         if self._source_parameters.server_certificates is None:
             return None
 
@@ -185,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.104.0rc1 → external_systems-0.106.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "external-systems"
-version = "0.104.0-rc1"
+version = "0.106.0"
 description = "A Python library for interacting with Foundry Sources"
 authors = ["Palantir Technologies, Inc."]
 license = "Apache-2.0"