fixing bug

eamonnfaherty · Apr 3, 2019 · 9b3ba99 · 9b3ba99
1 parent bcfc2d7
commit 9b3ba99
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/betterboto/client.py b/betterboto/client.py
@@ -53,6 +53,7 @@ def __enter__(self):
     def __exit__(self, *args, **kwargs):
         self.clients = None
 
+
 class CrossAccountClientContextManager(object):
     def __init__(self, service_name, role_arn, role_session_name, **kwargs):
         super().__init__()
@@ -75,7 +76,7 @@ def __enter__(self):
             "aws_session_token": credentials['SessionToken'],
         }
         if self.kwargs is not None:
-            kwargs.update(kwargs)
+            kwargs.update(self.kwargs)
         self.client = boto3.client(**kwargs)
         self.client = make_better(self.service_name, self.client)
         return self.client

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="better-boto",
-    version="0.3.2",
+    version="0.3.3",
     author="Eamonn Faherty",
     author_email="python-packages@designandsolve.co.uk",
     description="Helpers to make using boto3 more enjoyable",