removed superfluous session invocation

Author: michaelthomasletts

boto3_refresh_session/session.py

@@ -22,7 +22,7 @@
 
 from attrs import define, field
 from attrs.validators import ge, instance_of, optional
-from boto3 import Session
+from boto3 import Session, client
 from botocore.credentials import (
     DeferredRefreshableCredentials,
     RefreshableCredentials,
@@ -115,7 +115,9 @@ def __attrs_post_init__(self):
             )
 
         __session._credentials = __credentials
-        self.session = Session(botocore_session=__session)
+        self.session = Session(
+            botocore_session=__session, **self.session_kwargs
+        )
 
     def _get_credentials(self) -> dict:
         """Returns temporary credentials via AWS STS.
@@ -126,11 +128,10 @@ def _get_credentials(self) -> dict:
             AWS temporary credentials.
         """
 
-        __session = Session(region_name=self.region, **self.session_kwargs)
-        __client = __session.client(
+        __sts_client = client(
             service_name="sts", region_name=self.region, **self.client_kwargs
         )
-        __temporary_credentials = __client.assume_role(
+        __temporary_credentials = __sts_client.assume_role(
             RoleArn=self.role_arn,
             RoleSessionName=self.session_name,
             DurationSeconds=self.ttl,

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "boto3-refresh-session"
-version = "0.0.25"
+version = "0.0.26"
 description = "A simple Python package for refreshing boto3 sessions automatically."
 authors = [
     {name = "Mike Letts",email = "[email protected]"}

tests/test_session.py

@@ -16,6 +16,7 @@ def test_defer_refresh():
         defer_refresh=True,
     ).session
     s3 = session.client(service_name="s3")
+    s3.list_buckets()
 
     # testing defer_refresh = False
     logger.info("Testing AutoRefreshableSession with defer_refresh = False")
@@ -26,3 +27,4 @@ def test_defer_refresh():
         defer_refresh=False,
     ).session
     s3 = session.client(service_name="s3")
+    s3.list_buckets()