import json
import boto3

def lambda_handler(event, context):
    # TODO implement
    sts_client = boto3.client('sts')
    assumed_role_object=sts_client.assume_role(
        RoleArn="arn:aws:iam::111111111111:role/rolex",
        RoleSessionName="lambda"
    )
    print("export AWS_ACCESS_KEY_ID=" + assumed_role_object['Credentials']['AccessKeyId'])
    print("export AWS_SECRET_ACCESS_KEY=" + assumed_role_object['Credentials']['SecretAccessKey'])
    print("export AWS_SESSION_TOKEN=" + assumed_role_object['Credentials']['SessionToken'])
    print("export AWS_DEFAULT_REGION=ap-east-1")