"""Connector for Switchboard AWS access."""
import boto3

from switchboard_consumer import config

sts = boto3.client('sts')


def get_switchboard_credentials():
    """Get credentials for interacting with Switchboard AWS."""
    assumed_role = sts.assume_role(
        RoleArn=config.SWITCHBOARD_ROLE_ARN,
        RoleSessionName='switchboard-graphql'
    )
    credentials = assumed_role['Credentials']
    return {
        'aws_access_key_id': credentials['AccessKeyId'],
        'aws_secret_access_key': credentials['SecretAccessKey'],
        'aws_session_token': credentials['SessionToken'],
    }