def get_session_for_role()

in enrichment_function/import_findings/helper.py [0:0]

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

    def get_session_for_role(self, role_arn: str):
        role = boto3.client('sts').assume_role(RoleArn=role_arn, RoleSessionName='switch-role')
        credentials = role['Credentials']
        aws_access_key_id = credentials['AccessKeyId']
        aws_secret_access_key = credentials['SecretAccessKey']
        aws_session_token = credentials['SessionToken']
        return boto3.session.Session(
            aws_access_key_id = aws_access_key_id,
            aws_secret_access_key = aws_secret_access_key,
            aws_session_token = aws_session_token)