boto3-refresh-session 0.0.22__tar.gz → 0.0.24__tar.gz

{boto3_refresh_session-0.0.22 → boto3_refresh_session-0.0.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: boto3-refresh-session
-Version: 0.0.22
+Version: 0.0.24
 Summary: A simple Python package for refreshing boto3 sessions automatically.
 Home-page: https://github.com/michaelthomasletts/boto3-refresh-session
 License: MIT

{boto3_refresh_session-0.0.22 → boto3_refresh_session-0.0.24}/boto3_refresh_session/session.py

@@ -21,9 +21,12 @@ __all__ = ["AutoRefreshableSession"]
 from typing import Type
 
 from attrs import define, field
-from attrs.validators import instance_of, le, optional
+from attrs.validators import ge, instance_of, optional
 from boto3 import Session
-from botocore.credentials import RefreshableCredentials
+from botocore.credentials import (
+    DeferredRefreshableCredentials,
+    RefreshableCredentials,
+)
 from botocore.session import get_session
 
 
@@ -43,8 +46,13 @@ class AutoRefreshableSession:
         AWS role ARN.
     session_name : str
         Name for session.
+    defer_refresh : bool, optional
+        If ``True`` then temporary credentials are not automatically refreshed until
+        they are explicitly needed. If ``False`` then temporary credentials refresh
+        immediately upon expiration. Default is ``True``.
     ttl : int, optional
-        Number of seconds until temporary credentials expire, default 900.
+        Number of seconds until temporary credentials expire. Must be greater than or
+        equal to 900 seconds. Default is 900.
     session_kwargs : dict, optional
         Optional keyword arguments for :class:`boto3.session.Session`.
     client_kwargs : dict, optional
@@ -61,6 +69,10 @@ class AutoRefreshableSession:
     Check the :ref:`authorization documentation <authorization>` for additional
     information concerning how to authorize access to AWS.
 
+    The default ``defer_refresh`` parameter value results in temporary credentials not
+    being refreshed until they are explicitly requested; that is more efficient than
+    refreshing expired temporary credentials automatically after they expire.
+
     Examples
     --------
     Here's how to initialize this object:
@@ -76,8 +88,9 @@ class AutoRefreshableSession:
     region: str = field(validator=instance_of(str))
     role_arn: str = field(validator=instance_of(str))
     session_name: str = field(validator=instance_of(str))
+    defer_refresh: bool = field(default=True, validator=instance_of(bool))
     ttl: int = field(
-        default=900, validator=optional([instance_of(int), le(900)])
+        default=900, validator=optional([instance_of(int), ge(900)])
     )
     session_kwargs: dict = field(
         default={}, validator=optional(instance_of(dict))
@@ -88,13 +101,19 @@ class AutoRefreshableSession:
     session: Type[Session] = field(init=False)
 
     def __attrs_post_init__(self):
-        __credentials = RefreshableCredentials.create_from_metadata(
-            metadata=self._get_credentials(),
-            refresh_using=self._get_credentials,
-            method="sts-assume-role",
-        )
         __session = get_session()
-        # https://github.com/boto/botocore/blob/f8a1dd0820b548a5e8dc05420b28b6f1c6e21154/botocore/session.py#L143
+
+        if not self.defer_refresh:
+            __credentials = RefreshableCredentials.create_from_metadata(
+                metadata=self._get_credentials(),
+                refresh_using=self._get_credentials,
+                method="sts-assume-role",
+            )
+        else:
+            __credentials = DeferredRefreshableCredentials(
+                refresh_using=self._get_credentials, method="sts-assume-role"
+            )
+
         __session._credentials = __credentials
         self.session = Session(botocore_session=__session)
 

{boto3_refresh_session-0.0.22 → boto3_refresh_session-0.0.24}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "boto3-refresh-session"
-version = "0.0.22"
+version = "0.0.24"
 description = "A simple Python package for refreshing boto3 sessions automatically."
 authors = [
     {name = "Mike Letts",email = "michaelthomasletts@gmail.com"}
@@ -42,4 +42,10 @@ line_length = 79
 ensure_newline_before_comments = true
 use_parentheses = true
 include_trailing_comma = true
-multi_line_output = 3
+multi_line_output = 3
+
+[tool.pytest.ini_options]
+log_cli = true
+log_cli_level = "INFO"
+log_cli_date_format = "%Y-%m-%d %H:%M:%S"
+log_cli_format = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"