25 lines
788 B
Python
Executable File
25 lines
788 B
Python
Executable File
#!/usr/bin/env python3.13
|
|
from botocore.exceptions import ClientError
|
|
import boto3
|
|
import sys
|
|
|
|
def main() -> None:
|
|
# TODO implement
|
|
sts_client = boto3.client('sts')
|
|
try:
|
|
assumed_role_object = sts_client.assume_role(
|
|
RoleArn="arn:aws:iam::" + sys.argv[1] + ":role/" + sys.argv[2],
|
|
RoleSessionName=sys.argv[2]
|
|
)
|
|
print(f'''
|
|
export AWS_ACCESS_KEY_ID={assumed_role_object['Credentials']['AccessKeyId']}
|
|
export AWS_SECRET_ACCESS_KEY{assumed_role_object['Credentials']['SecretAccessKey']}
|
|
export AWS_SESSION_TOKEN={assumed_role_object['Credentials']['SessionToken']}
|
|
export AWS_DEFAULT_REGION=ap-east-1")
|
|
''')
|
|
except ClientError as e:
|
|
print(e)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|