def validate_and_build_instance_fleets()

in awscli/customizations/emr/instancefleetsutils.py [0:0]

• 38 lines of code
• 13 McCabe index (conditional complexity)

def validate_and_build_instance_fleets(parsed_instance_fleets):
    """
    Helper method that converts --instance-fleets option value in
    create-cluster to Amazon Elastic MapReduce InstanceFleetConfig
    data type.
    """
    instance_fleets = []
    for instance_fleet in parsed_instance_fleets:
        instance_fleet_config = {}

        keys = instance_fleet.keys()

        if 'Name' in keys:
            instance_fleet_config['Name'] = instance_fleet['Name']
        else:
            instance_fleet_config['Name'] = instance_fleet['InstanceFleetType']
        instance_fleet_config['InstanceFleetType'] = instance_fleet['InstanceFleetType']

        if 'TargetOnDemandCapacity' in keys:
            instance_fleet_config['TargetOnDemandCapacity'] = instance_fleet['TargetOnDemandCapacity']

        if 'TargetSpotCapacity' in keys:
            instance_fleet_config['TargetSpotCapacity'] = instance_fleet['TargetSpotCapacity']

        if 'InstanceTypeConfigs' in keys:
            instance_fleet_config['InstanceTypeConfigs'] = instance_fleet['InstanceTypeConfigs']

        if 'LaunchSpecifications' in keys:
            instanceFleetProvisioningSpecifications = instance_fleet['LaunchSpecifications']
            instance_fleet_config['LaunchSpecifications'] = {}

            if 'SpotSpecification' in instanceFleetProvisioningSpecifications:
                instance_fleet_config['LaunchSpecifications']['SpotSpecification'] = \
                    instanceFleetProvisioningSpecifications['SpotSpecification']

            if 'OnDemandSpecification' in instanceFleetProvisioningSpecifications:
                instance_fleet_config['LaunchSpecifications']['OnDemandSpecification'] = \
                    instanceFleetProvisioningSpecifications['OnDemandSpecification']

        if 'ResizeSpecifications' in keys:
            instanceFleetResizeSpecifications = instance_fleet['ResizeSpecifications']
            instance_fleet_config['ResizeSpecifications'] = {}

            if 'SpotResizeSpecification' in instanceFleetResizeSpecifications:
                instance_fleet_config['ResizeSpecifications']['SpotResizeSpecification'] = \
                    instanceFleetResizeSpecifications['SpotResizeSpecification']

            if 'OnDemandResizeSpecification' in instanceFleetResizeSpecifications:
                instance_fleet_config['ResizeSpecifications']['OnDemandResizeSpecification'] = \
                    instanceFleetResizeSpecifications['OnDemandResizeSpecification']
        
        if 'Context' in keys:
            instance_fleet_config['Context'] = instance_fleet['Context']

        instance_fleets.append(instance_fleet_config)
    return instance_fleets