Merge pull request #348 from eedgar/async

run aws session sts code asyncronously vastly improving the performance.

gimme_aws_creds/main.py

@@ -17,6 +17,8 @@
 import os
 import re
 import sys
+import concurrent.futures
+
 
 # extras
 import boto3
@@ -613,6 +615,7 @@ def aws_results(self):
                 'links': {'appLink': self.config.app_url}
             }
             aws_results.append(new_app_entry)
+            self.ui.info("Authentication Success!")
 
         # Use the gimme_creds_lambda service
         else:
@@ -705,7 +708,7 @@ def prepare_data(self, role, generate_credentials=False):
             except ClientError as ex:
                 if 'requested DurationSeconds exceeds the MaxSessionDuration' in ex.response['Error']['Message']:
                     self.ui.warning(
-                        "The requested session duration was too long for this role.  Falling back to 1 hour.")
+                        "The requested session duration was too long for the role {}.  Falling back to 1 hour.".format(role.role))
                     aws_creds = self._get_sts_creds(
                         self.aws_partition,
                         self.saml_data['SAMLResponse'],
@@ -770,12 +773,19 @@ def get_profile_name(self, cred_profile, include_path, naming_data, resolve_alia
 
     def iter_selected_aws_credentials(self):
         results = []
-        for role in self.aws_selected_roles:
+        aws_results = []
+
+        def generate_credentials_prepare_data(role):
             data = self.prepare_data(role, generate_credentials=True)
-            if not data:
+            return data
+
+        with concurrent.futures.ThreadPoolExecutor(max_workers=10) as executor:
+            aws_results = executor.map(generate_credentials_prepare_data, self.aws_selected_roles)
+        for ar in aws_results:
+            if not ar:
                 continue
-            results.append(data)
-            yield data
+            results.append(ar)
+            yield ar
 
         self._cache['selected_aws_credentials'] = results