def lambda_handler()

in EC2 Auto Clean Room Forensics/Lambda-Functions/isolateInstance.py [0:0]

• 17 lines of code
• 5 McCabe index (conditional complexity)

def lambda_handler(event, context):

    instanceID = event.get('instanceID')
    response = 'FAILED'
    targetGroups = elbv2client.describe_target_groups()
    # Iterates ELB and gets the ELB where the instance is attached
    for key in targetGroups['TargetGroups']:
        targetGroupArn = key.get('TargetGroupArn')
        targets = elbv2client.describe_target_health(
            TargetGroupArn=targetGroupArn
        )

        instanceIDlist = []
        for instanceKey in targets['TargetHealthDescriptions']:
            instanceIDlist.append(instanceKey.get('Target').get('Id'))

        if instanceID in instanceIDlist:
            response = isolateInstance(instanceID, targetGroupArn)
    event['STATUS'] = response
    event['targetGroupArn'] = targetGroupArn
    return event