boto3-refresh-session 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

boto3_refresh_session/session.py

@@ -6,6 +6,8 @@ credentials.
 """
 __all__ = ["RefreshableSession"]
 
+from typing import Any, Dict
+
 from boto3 import client
 from boto3.session import Session
 from botocore.credentials import (
@@ -29,7 +31,7 @@ class RefreshableSession(Session):
         Default is ``True``.
     sts_client_kwargs : dict, optional
         Optional keyword arguments for the :class:`STS.Client` object. Default is
-        an empty dictionary.
+        None.
 
     Other Parameters
     ----------------
@@ -94,9 +96,9 @@ class RefreshableSession(Session):
 
     def __init__(
         self,
-        assume_role_kwargs: dict,
+        assume_role_kwargs: Dict[Any],
         defer_refresh: bool = True,
-        sts_client_kwargs: dict = {},
+        sts_client_kwargs: Dict[Any] = None,
         **kwargs,
     ):
         # inheriting from boto3.session.Session
@@ -106,7 +108,10 @@ class RefreshableSession(Session):
         self.assume_role_kwargs = assume_role_kwargs
 
         # initializing the STS client
-        self._sts_client = client(service_name="sts", **sts_client_kwargs)
+        if sts_client_kwargs is not None:
+            self._sts_client = client(service_name="sts", **sts_client_kwargs)
+        else:
+            self._sts_client = client(service_name="sts")
 
         # determining how exactly to refresh expired temporary credentials
         if not defer_refresh:
@@ -122,7 +127,7 @@ class RefreshableSession(Session):
                 refresh_using=self._get_credentials, method="sts-assume-role"
             )
 
-    def _get_credentials(self) -> dict:
+    def _get_credentials(self) -> Dict[Any]:
         """Returns temporary credentials via AWS STS.
 
         Returns
@@ -132,14 +137,12 @@ class RefreshableSession(Session):
         """
 
         # fetching temporary credentials
-        _temporary_credentials = self._sts_client.assume_role(
+        temporary_credentials = self._sts_client.assume_role(
             **self.assume_role_kwargs
         )["Credentials"]
         return {
-            "access_key": _temporary_credentials.get("AccessKeyId"),
-            "secret_key": _temporary_credentials.get("SecretAccessKey"),
-            "token": _temporary_credentials.get("SessionToken"),
-            "expiry_time": _temporary_credentials.get(
-                "Expiration"
-            ).isoformat(),
+            "access_key": temporary_credentials.get("AccessKeyId"),
+            "secret_key": temporary_credentials.get("SecretAccessKey"),
+            "token": temporary_credentials.get("SessionToken"),
+            "expiry_time": temporary_credentials.get("Expiration").isoformat(),
         }

{boto3_refresh_session-1.0.0.dist-info → boto3_refresh_session-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3-refresh-session
-Version: 1.0.0
+Version: 1.0.2
 Summary: A simple Python package for refreshing the temporary security credentials in a boto3.session.Session object automatically.
 License: MIT
 Keywords: boto3,botocore,aws
@@ -83,13 +83,8 @@ assume_role_kwargs = {
   'DurationSeconds': '<your-selection>',
   ...
 }
-sts_client_kwargs = {
-  ...
-}
 session = brs.RefreshableSession(
-  assume_role_kwargs=assume_role_kwargs,
-  sts_client_kwargs=sts_client_kwargs,
-  region_name='us-east-1',
+  assume_role_kwargs=assume_role_kwargs
 )
 s3 = session.client(service_name='s3')
 buckets = s3.list_buckets()

boto3_refresh_session-1.0.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+boto3_refresh_session/__init__.py,sha256=NiaKXI1Ln9DU0cJOM3bYLEm0GNTyT7AV7D3itCGzfME,127
+boto3_refresh_session/session.py,sha256=rQHJDvOAz0glymYXQBfSt1keFvQAM_G8AQXgMDLZq2U,5349
+boto3_refresh_session-1.0.2.dist-info/LICENSE,sha256=I3ZYTXAjbIly6bm6J-TvFTuuHwTKws4h89QaY5c5HiY,1067
+boto3_refresh_session-1.0.2.dist-info/METADATA,sha256=0AiJd6DKWLDG2JSC0smJTKpdqCQMcgukMfyAsvgX4R4,4395
+boto3_refresh_session-1.0.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+boto3_refresh_session-1.0.2.dist-info/RECORD,,

boto3_refresh_session-1.0.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-boto3_refresh_session/__init__.py,sha256=NiaKXI1Ln9DU0cJOM3bYLEm0GNTyT7AV7D3itCGzfME,127
-boto3_refresh_session/session.py,sha256=flpqQ5E9Z6XdtFNpCO2Ew1ZVJDYGncF6PG7uwlA7IJ4,5234
-boto3_refresh_session-1.0.0.dist-info/LICENSE,sha256=I3ZYTXAjbIly6bm6J-TvFTuuHwTKws4h89QaY5c5HiY,1067
-boto3_refresh_session-1.0.0.dist-info/METADATA,sha256=5-yiWf1HbXj_aDfqBTqtpNzYn7tdrQxVlDjihudHQF4,4492
-boto3_refresh_session-1.0.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-boto3_refresh_session-1.0.0.dist-info/RECORD,,