alternative for assuming the role

Scout24 · Nov 19, 2015 · 610de93 · 610de93
1 parent f8275bb
commit 610de93
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/src/integrationtest/python/assume_role.py b/src/integrationtest/python/assume_role.py
@@ -1,15 +1,17 @@
 #!/usr/bin/env python
 
-import boto3
+#import boto3
 import os
+import sys
+import json
 
 
-def assume_role():
-    invoker_role_arn = os.environ["INVOKER_ROLE_ARN"]
-    sts_client = boto3.client('sts')
-    return sts_client.assume_role(
-        RoleArn=invoker_role_arn,
-        RoleSessionName='integration_test')['Credentials']
+#def assume_role():
+#    invoker_role_arn = os.environ["INVOKER_ROLE_ARN"]
+#    sts_client = boto3.client('sts')
+#    return sts_client.assume_role(
+#        RoleArn=invoker_role_arn,
+#        RoleSessionName='integration_test')['Credentials']
 
 
 def format_(credentials):
@@ -22,4 +24,4 @@ def format_(credentials):
 
 
 if __name__ == "__main__":
-    format_(assume_role())
+    format_(json.loads(sys.stdin.read())['Credentials'])