aws-ar 1.0__tar.gz

aws-ar-1.0/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.1
+Name: aws-ar
+Version: 1.0
+Author: Shashank Dubey
+Requires-Dist: boto3

aws-ar-1.0/aws-ar/aws_ar.py

@@ -0,0 +1,34 @@
+import argparse
+import subprocess
+
+def assume_role(arn, profile, new_profile):
+    
+    sts_output = subprocess.check_output(['aws', 'sts', 'assume-role', '--role-arn', arn, '--role-session-name', 'AssumeRoleSession', '--profile', profile])
+
+    sts_output_lines = sts_output.decode().splitlines()
+    for line in sts_output_lines:
+        if 'AccessKeyId' in line:
+            access_key = line.split(':')[1].strip().strip(',').strip('"').strip('"')
+        elif 'SecretAccessKey' in line:
+            secret_key = line.split(':')[1].strip().strip(',').strip('"').strip('"')
+        elif 'SessionToken' in line:
+            session_token = line.split(':')[1].strip().strip(',').strip('"').strip('"')
+
+    subprocess.run(['aws', 'configure', 'set', 'aws_access_key_id', access_key, '--profile', new_profile])
+    subprocess.run(['aws', 'configure', 'set', 'aws_secret_access_key', secret_key, '--profile', new_profile])
+    subprocess.run(['aws', 'configure', 'set', 'aws_session_token', session_token, '--profile', new_profile])
+
+def main():
+    parser = argparse.ArgumentParser(description='AWS Assume Role Utility')
+    parser.add_argument('--role-arn', required=True, help='ARN of the role to assume')
+    parser.add_argument('--profile', help='Name of the profile to use for assuming the role (default: default)')
+    parser.add_argument('--new-profile', required=True, help='Name of the new profile to create')
+
+    args = parser.parse_args()
+
+    assume_role(args.role_arn, args.profile or 'default',args.new_profile)
+
+    print(f'New profile {args.new_profile} configured with assumed role credentials.')
+
+if __name__ == "__main__":
+    main()

aws-ar-1.0/aws_ar.egg-info/PKG-INFO

@@ -0,0 +1,5 @@
+Metadata-Version: 2.1
+Name: aws-ar
+Version: 1.0
+Author: Shashank Dubey
+Requires-Dist: boto3

aws-ar-1.0/aws_ar.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+setup.py
+aws-ar/__init__.py
+aws-ar/aws_ar.py
+aws_ar.egg-info/PKG-INFO
+aws_ar.egg-info/SOURCES.txt
+aws_ar.egg-info/dependency_links.txt
+aws_ar.egg-info/entry_points.txt
+aws_ar.egg-info/requires.txt
+aws_ar.egg-info/top_level.txt

aws-ar-1.0/aws_ar.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

aws-ar-1.0/aws_ar.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+aws-ar = aws_ar.aws_ar:main

aws-ar-1.0/aws_ar.egg-info/requires.txt

@@ -0,0 +1 @@
+boto3

aws-ar-1.0/aws_ar.egg-info/top_level.txt

@@ -0,0 +1 @@
+aws-ar

aws-ar-1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

aws-ar-1.0/setup.py

@@ -0,0 +1,16 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='aws-ar',
+    version='1.0',
+    author='Shashank Dubey',
+    packages=find_packages(),
+    entry_points={
+        'console_scripts': [
+            'aws-ar = aws_ar.aws_ar:main'
+        ]
+    },
+    install_requires=[
+        'boto3',
+    ],
+)