def assume_role()

in lib/custom_resources/create_and_attach_scp.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

def assume_role(arn, session_name):
    """

    :param arn: ORG role to be assumed
    :param session_name: Session name
    :return: session object for the role
    """
    response = sts_client.assume_role(RoleArn=arn, RoleSessionName=session_name)
    session = boto3.Session(aws_access_key_id=response['Credentials']['AccessKeyId'],
                            aws_secret_access_key=response['Credentials']['SecretAccessKey'],
                            aws_session_token=response['Credentials']['SessionToken'])
    return session