#!/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()