boto3-refresh-session 1.3.6__tar.gz → 1.3.7__tar.gz

{boto3_refresh_session-1.3.6 → boto3_refresh_session-1.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3-refresh-session
-Version: 1.3.6
+Version: 1.3.7
 Summary: A simple Python package for refreshing the temporary security credentials in a boto3.session.Session object automatically.
 License: MIT
 Keywords: boto3,botocore,aws
@@ -170,7 +170,7 @@ def your_custom_credential_getter(...):
 session = RefreshableSession(
     method="custom",
     custom_credentials_method=your_custom_credential_getter,
-    custom_credentials_methods_args=...,
+    custom_credentials_method_args=...,
     region_name=region_name,
     profile_name=profile_name,
     ...

{boto3_refresh_session-1.3.6 → boto3_refresh_session-1.3.7}/README.md

@@ -147,7 +147,7 @@ def your_custom_credential_getter(...):
 session = RefreshableSession(
     method="custom",
     custom_credentials_method=your_custom_credential_getter,
-    custom_credentials_methods_args=...,
+    custom_credentials_method_args=...,
     region_name=region_name,
     profile_name=profile_name,
     ...

{boto3_refresh_session-1.3.6 → boto3_refresh_session-1.3.7}/boto3_refresh_session/__init__.py

@@ -3,5 +3,5 @@ from .session import RefreshableSession
 from .sts import STSRefreshableSession
 
 __all__ = ["RefreshableSession"]
-__version__ = "1.3.6"
+__version__ = "1.3.7"
 __author__ = "Mike Letts"

{boto3_refresh_session-1.3.6 → boto3_refresh_session-1.3.7}/boto3_refresh_session/custom.py

@@ -4,6 +4,7 @@ __all__ = ["CustomRefreshableSession"]
 
 from typing import Any, Callable
 
+from .exceptions import BRSError
 from .session import BaseRefreshableSession
 
 
@@ -50,7 +51,7 @@ class CustomRefreshableSession(BaseRefreshableSession, method="custom"):
     >>> sess = RefreshableSession(
     >>>     method='custom',
     >>>     custom_credentials_method=your_custom_credential_getter,
-    >>>     custom_credentials_methods_args=...,
+    >>>     custom_credentials_method_args=...,
     >>> )
     """
 
@@ -77,11 +78,20 @@ class CustomRefreshableSession(BaseRefreshableSession, method="custom"):
         )
 
     def _get_credentials(self) -> dict[str, str]:
-        return self._custom_get_credentials(
+        credentials = self._custom_get_credentials(
             **self._custom_get_credentials_args
         )
+        required_keys = {"access_key", "secret_key", "token", "expiry_time"}
 
-    def get_identity(self):
+        if missing := required_keys - credentials.keys():
+            raise BRSError(
+                f"The dict returned by custom_credentials_method is missing these key-value pairs: "
+                f"{', '.join(repr(param) for param in missing)}. "
+            )
+
+        return credentials
+
+    def get_identity(self) -> dict[str, str]:
         """Returns metadata about the custom credential getter.
 
         Returns
@@ -90,7 +100,9 @@ class CustomRefreshableSession(BaseRefreshableSession, method="custom"):
             Dict containing information about the custom credential getter.
         """
 
-        return {
-            "method": "custom",
-            "source": repr(self._custom_get_credentials.__name__),
-        }
+        source = getattr(
+            self._custom_get_credentials,
+            "__name__",
+            repr(self._custom_get_credentials),
+        )
+        return {"method": "custom", "source": repr(source)}

{boto3_refresh_session-1.3.6 → boto3_refresh_session-1.3.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "boto3-refresh-session"
-version = "1.3.6"
+version = "1.3.7"
 description = "A simple Python package for refreshing the temporary security credentials in a boto3.session.Session object automatically."
 authors = [
     {name = "Mike Letts",email = "lettsmt@gmail.com"}