boto3-refresh-session 2.0.5__tar.gz → 2.0.6__tar.gz

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3-refresh-session
-Version: 2.0.5
+Version: 2.0.6
 Summary: A simple Python package for refreshing the temporary security credentials in a boto3.session.Session object automatically.
 License: MIT
 Keywords: boto3,botocore,aws,sts,ecs,credentials,token,refresh

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/boto3_refresh_session/__init__.py

@@ -1,7 +1,10 @@
+from .methods.custom import CustomRefreshableSession
+from .methods.ecs import ECSRefreshableSession
+from .methods.sts import STSRefreshableSession
 from .session import RefreshableSession
 
 __all__ = ["RefreshableSession"]
-__version__ = "2.0.5"
+__version__ = "2.0.6"
 __title__ = "boto3-refresh-session"
 __author__ = "Mike Letts"
 __maintainer__ = "Mike Letts"

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/boto3_refresh_session/methods/custom.py

@@ -6,7 +6,7 @@ from typing import Any, Callable
 
 from ..exceptions import BRSError
 from ..session import BaseRefreshableSession
-from ..utils import TemporaryCredentials
+from ..utils import RefreshMethod, TemporaryCredentials
 
 
 class CustomRefreshableSession(BaseRefreshableSession, registry_key="custom"):
@@ -66,6 +66,8 @@ class CustomRefreshableSession(BaseRefreshableSession, registry_key="custom"):
         defer_refresh: bool | None = None,
         **kwargs,
     ):
+        self.defer_refresh = defer_refresh is not False
+        self.refresh_method: RefreshMethod = "custom"
         super().__init__(**kwargs)
 
         self._custom_get_credentials = custom_credentials_method
@@ -75,12 +77,6 @@ class CustomRefreshableSession(BaseRefreshableSession, registry_key="custom"):
             else {}
         )
 
-        self.initialize(
-            credentials_method=self._get_credentials,
-            defer_refresh=defer_refresh is not False,
-            refresh_method="custom",
-        )
-
     def _get_credentials(self) -> TemporaryCredentials:
         credentials: TemporaryCredentials = self._custom_get_credentials(
             **self._custom_get_credentials_args

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/boto3_refresh_session/methods/ecs.py

@@ -8,7 +8,7 @@ import requests
 
 from ..exceptions import BRSError
 from ..session import BaseRefreshableSession
-from ..utils import TemporaryCredentials
+from ..utils import RefreshMethod, TemporaryCredentials
 
 _ECS_CREDENTIALS_RELATIVE_URI = "AWS_CONTAINER_CREDENTIALS_RELATIVE_URI"
 _ECS_CREDENTIALS_FULL_URI = "AWS_CONTAINER_CREDENTIALS_FULL_URI"
@@ -35,18 +35,14 @@ class ECSRefreshableSession(BaseRefreshableSession, registry_key="ecs"):
     """
 
     def __init__(self, defer_refresh: bool | None = None, **kwargs):
-        super().__init__(**kwargs)
+        self.defer_refresh: bool = defer_refresh is not False
+        self.refresh_method: RefreshMethod = "ecs-container-metadata"
+        super().__init__(**kwargs)  # mounting refreshable credentials
 
         self._endpoint = self._resolve_endpoint()
         self._headers = self._build_headers()
         self._http = self._init_http_session()
 
-        self.initialize(
-            credentials_method=self._get_credentials,
-            defer_refresh=defer_refresh is not False,
-            refresh_method="ecs-container-metadata",
-        )
-
     def _resolve_endpoint(self) -> str:
         uri = os.environ.get(_ECS_CREDENTIALS_FULL_URI) or os.environ.get(
             _ECS_CREDENTIALS_RELATIVE_URI

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/boto3_refresh_session/methods/sts.py

@@ -6,7 +6,12 @@ from typing import Any
 
 from ..exceptions import BRSWarning
 from ..session import BaseRefreshableSession
-from ..utils import AssumeRoleParams, STSClientParams, TemporaryCredentials
+from ..utils import (
+    AssumeRoleParams,
+    RefreshMethod,
+    STSClientParams,
+    TemporaryCredentials,
+)
 
 
 class STSRefreshableSession(BaseRefreshableSession, registry_key="sts"):
@@ -42,7 +47,10 @@ class STSRefreshableSession(BaseRefreshableSession, registry_key="sts"):
         sts_client_kwargs: STSClientParams | None = None,
         **kwargs,
     ):
-        super().__init__(**kwargs)
+        self.defer_refresh: bool = defer_refresh is not False
+        self.refresh_method: RefreshMethod = "sts-assume-role"
+        super().__init__(**kwargs)  # mounting refreshable credentials
+
         self.assume_role_kwargs = assume_role_kwargs
 
         if sts_client_kwargs is not None:
@@ -59,13 +67,6 @@ class STSRefreshableSession(BaseRefreshableSession, registry_key="sts"):
         else:
             self._sts_client = self.client(service_name="sts")
 
-        # mounting refreshable credentials
-        self.initialize(
-            credentials_method=self._get_credentials,
-            defer_refresh=defer_refresh is not False,
-            refresh_method="sts-assume-role",
-        )
-
     def _get_credentials(self) -> TemporaryCredentials:
         temporary_credentials = self._sts_client.assume_role(
             **self.assume_role_kwargs

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/boto3_refresh_session/utils.py

@@ -115,22 +115,16 @@ class BRSSession(Session):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
 
-    def initialize(
-        self,
-        credentials_method: Callable,
-        defer_refresh: bool,
-        refresh_method: RefreshMethod,
-    ):
-        # determining how exactly to refresh expired temporary credentials
-        if not defer_refresh:
+    def __post_init__(self):
+        if not self.defer_refresh:
             self._credentials = RefreshableCredentials.create_from_metadata(
-                metadata=credentials_method(),
-                refresh_using=credentials_method,
-                method=refresh_method,
+                metadata=self._get_credentials(),
+                refresh_using=self._get_credentials,
+                method=self.refresh_method,
             )
         else:
             self._credentials = DeferredRefreshableCredentials(
-                refresh_using=credentials_method, method=refresh_method
+                refresh_using=self._get_credentials, method=self.refresh_method
             )
 
     def refreshable_credentials(self) -> RefreshableTemporaryCredentials:

{boto3_refresh_session-2.0.5 → boto3_refresh_session-2.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "boto3-refresh-session"
-version = "2.0.5"
+version = "2.0.6"
 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"}