From b40485a115f199cea3eb8212547dfb0dae00703b Mon Sep 17 00:00:00 2001 From: Alexander Romanov Date: Thu, 2 Apr 2020 14:06:45 +0200 Subject: [PATCH] Use AWS Session Token when available to build a session this for example is useful in scenarios where you run requests after assuming a role --- awswrangler/session.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/awswrangler/session.py b/awswrangler/session.py index 5553d1d58..709e73403 100644 --- a/awswrangler/session.py +++ b/awswrangler/session.py @@ -142,6 +142,8 @@ def _load_new_boto3_session(self) -> None: if self.aws_access_key_id and self.aws_secret_access_key: args["aws_access_key_id"] = self.aws_access_key_id args["aws_secret_access_key"] = self.aws_secret_access_key + if self.aws_session_token: + args['aws_session_token'] = self.aws_session_token self._boto3_session = boto3.Session(**args) self._profile_name = self._boto3_session.profile_name credentials = self._boto3_session.get_credentials()