from __future__ import unicode_literals import json import re from datetime import datetime from functools import wraps import pytz from urllib.parse import urlparse from moto.core.responses import AWSServiceSpec from moto.core.responses import BaseResponse from moto.core.responses import xml_to_json_response from moto.core.utils import tags_from_query_string from .exceptions import EmrError from .models import emr_backends from .utils import steps_from_query_string, Unflattener, ReleaseLabel def generate_boto3_response(operation): """The decorator to convert an XML response to JSON, if the request is determined to be from boto3. Pass the API action as a parameter. """ def _boto3_request(method): @wraps(method) def f(self, *args, **kwargs): rendered = method(self, *args, **kwargs) if "json" in self.headers.get("Content-Type", []): self.response_headers.update( { "x-amzn-requestid": "2690d7eb-ed86-11dd-9877-6fad448a8419", "date": datetime.now(pytz.utc).strftime( "%a, %d %b %Y %H:%M:%S %Z" ), "content-type": "application/x-amz-json-1.1", } ) resp = xml_to_json_response(self.aws_service_spec, operation, rendered) return "" if resp is None else json.dumps(resp) return rendered return f return _boto3_request class ElasticMapReduceResponse(BaseResponse): # EMR end points are inconsistent in the placement of region name # in the URL, so parsing it out needs to be handled differently region_regex = [ re.compile(r"elasticmapreduce\.(.+?)\.amazonaws\.com"), re.compile(r"(.+?)\.elasticmapreduce\.amazonaws\.com"), ] aws_service_spec = AWSServiceSpec("data/emr/2009-03-31/service-2.json") def get_region_from_url(self, request, full_url): parsed = urlparse(full_url) for regex in self.region_regex: match = regex.search(parsed.netloc) if match: return match.group(1) return self.default_region @property def backend(self): return emr_backends[self.region] @generate_boto3_response("AddInstanceGroups") def add_instance_groups(self): jobflow_id = self._get_param("JobFlowId") instance_groups = self._get_list_prefix("InstanceGroups.member") for item in instance_groups: item["instance_count"] = int(item["instance_count"]) # Adding support to EbsConfiguration self._parse_ebs_configuration(item) # Adding support for auto_scaling_policy Unflattener.unflatten_complex_params(item, "auto_scaling_policy") instance_groups = self.backend.add_instance_groups(jobflow_id, instance_groups) template = self.response_template(ADD_INSTANCE_GROUPS_TEMPLATE) return template.render(instance_groups=instance_groups) @generate_boto3_response("AddJobFlowSteps") def add_job_flow_steps(self): job_flow_id = self._get_param("JobFlowId") steps = self.backend.add_job_flow_steps( job_flow_id, steps_from_query_string(self._get_list_prefix("Steps.member")) ) template = self.response_template(ADD_JOB_FLOW_STEPS_TEMPLATE) return template.render(steps=steps) @generate_boto3_response("AddTags") def add_tags(self): cluster_id = self._get_param("ResourceId") tags = tags_from_query_string(self.querystring, prefix="Tags") self.backend.add_tags(cluster_id, tags) template = self.response_template(ADD_TAGS_TEMPLATE) return template.render() def cancel_steps(self): raise NotImplementedError @generate_boto3_response("CreateSecurityConfiguration") def create_security_configuration(self): name = self._get_param("Name") security_configuration = self._get_param("SecurityConfiguration") resp = self.backend.create_security_configuration( name=name, security_configuration=security_configuration ) template = self.response_template(CREATE_SECURITY_CONFIGURATION_TEMPLATE) return template.render(name=name, creation_date_time=resp.creation_date_time) @generate_boto3_response("DescribeSecurityConfiguration") def describe_security_configuration(self): name = self._get_param("Name") security_configuration = self.backend.get_security_configuration(name=name) template = self.response_template(DESCRIBE_SECURITY_CONFIGURATION_TEMPLATE) return template.render(security_configuration=security_configuration) @generate_boto3_response("DeleteSecurityConfiguration") def delete_security_configuration(self): name = self._get_param("Name") self.backend.delete_security_configuration(name=name) template = self.response_template(DELETE_SECURITY_CONFIGURATION_TEMPLATE) return template.render() @generate_boto3_response("DescribeCluster") def describe_cluster(self): cluster_id = self._get_param("ClusterId") cluster = self.backend.get_cluster(cluster_id) template = self.response_template(DESCRIBE_CLUSTER_TEMPLATE) return template.render(cluster=cluster) @generate_boto3_response("DescribeJobFlows") def describe_job_flows(self): created_after = self._get_param("CreatedAfter") created_before = self._get_param("CreatedBefore") job_flow_ids = self._get_multi_param("JobFlowIds.member") job_flow_states = self._get_multi_param("JobFlowStates.member") clusters = self.backend.describe_job_flows( job_flow_ids, job_flow_states, created_after, created_before ) template = self.response_template(DESCRIBE_JOB_FLOWS_TEMPLATE) return template.render(clusters=clusters) @generate_boto3_response("DescribeStep") def describe_step(self): cluster_id = self._get_param("ClusterId") step_id = self._get_param("StepId") step = self.backend.describe_step(cluster_id, step_id) template = self.response_template(DESCRIBE_STEP_TEMPLATE) return template.render(step=step) @generate_boto3_response("ListBootstrapActions") def list_bootstrap_actions(self): cluster_id = self._get_param("ClusterId") marker = self._get_param("Marker") bootstrap_actions, marker = self.backend.list_bootstrap_actions( cluster_id, marker ) template = self.response_template(LIST_BOOTSTRAP_ACTIONS_TEMPLATE) return template.render(bootstrap_actions=bootstrap_actions, marker=marker) @generate_boto3_response("ListClusters") def list_clusters(self): cluster_states = self._get_multi_param("ClusterStates.member") created_after = self._get_param("CreatedAfter") created_before = self._get_param("CreatedBefore") marker = self._get_param("Marker") clusters, marker = self.backend.list_clusters( cluster_states, created_after, created_before, marker ) template = self.response_template(LIST_CLUSTERS_TEMPLATE) return template.render(clusters=clusters, marker=marker) @generate_boto3_response("ListInstanceGroups") def list_instance_groups(self): cluster_id = self._get_param("ClusterId") marker = self._get_param("Marker") instance_groups, marker = self.backend.list_instance_groups( cluster_id, marker=marker ) template = self.response_template(LIST_INSTANCE_GROUPS_TEMPLATE) return template.render(instance_groups=instance_groups, marker=marker) @generate_boto3_response("ListInstances") def list_instances(self): cluster_id = self._get_param("ClusterId") marker = self._get_param("Marker") instance_group_id = self._get_param("InstanceGroupId") instance_group_types = self._get_param("InstanceGroupTypes") instances, marker = self.backend.list_instances( cluster_id, marker=marker, instance_group_id=instance_group_id, instance_group_types=instance_group_types, ) template = self.response_template(LIST_INSTANCES_TEMPLATE) return template.render(instances=instances, marker=marker) @generate_boto3_response("ListSteps") def list_steps(self): cluster_id = self._get_param("ClusterId") marker = self._get_param("Marker") step_ids = self._get_multi_param("StepIds.member") step_states = self._get_multi_param("StepStates.member") steps, marker = self.backend.list_steps( cluster_id, marker=marker, step_ids=step_ids, step_states=step_states ) template = self.response_template(LIST_STEPS_TEMPLATE) return template.render(steps=steps, marker=marker) @generate_boto3_response("ModifyCluster") def modify_cluster(self): cluster_id = self._get_param("ClusterId") step_concurrency_level = self._get_param("StepConcurrencyLevel") cluster = self.backend.modify_cluster(cluster_id, step_concurrency_level) template = self.response_template(MODIFY_CLUSTER_TEMPLATE) return template.render(cluster=cluster) @generate_boto3_response("ModifyInstanceGroups") def modify_instance_groups(self): instance_groups = self._get_list_prefix("InstanceGroups.member") for item in instance_groups: item["instance_count"] = int(item["instance_count"]) instance_groups = self.backend.modify_instance_groups(instance_groups) template = self.response_template(MODIFY_INSTANCE_GROUPS_TEMPLATE) return template.render(instance_groups=instance_groups) @generate_boto3_response("RemoveTags") def remove_tags(self): cluster_id = self._get_param("ResourceId") tag_keys = self._get_multi_param("TagKeys.member") self.backend.remove_tags(cluster_id, tag_keys) template = self.response_template(REMOVE_TAGS_TEMPLATE) return template.render() @generate_boto3_response("RunJobFlow") def run_job_flow(self): instance_attrs = dict( master_instance_type=self._get_param("Instances.MasterInstanceType"), slave_instance_type=self._get_param("Instances.SlaveInstanceType"), instance_count=self._get_int_param("Instances.InstanceCount", 1), ec2_key_name=self._get_param("Instances.Ec2KeyName"), ec2_subnet_id=self._get_param("Instances.Ec2SubnetId"), hadoop_version=self._get_param("Instances.HadoopVersion"), availability_zone=self._get_param( "Instances.Placement.AvailabilityZone", self.backend.region_name + "a" ), keep_job_flow_alive_when_no_steps=self._get_bool_param( "Instances.KeepJobFlowAliveWhenNoSteps", False ), termination_protected=self._get_bool_param( "Instances.TerminationProtected", False ), emr_managed_master_security_group=self._get_param( "Instances.EmrManagedMasterSecurityGroup" ), emr_managed_slave_security_group=self._get_param( "Instances.EmrManagedSlaveSecurityGroup" ), service_access_security_group=self._get_param( "Instances.ServiceAccessSecurityGroup" ), additional_master_security_groups=self._get_multi_param( "Instances.AdditionalMasterSecurityGroups.member." ), additional_slave_security_groups=self._get_multi_param( "Instances.AdditionalSlaveSecurityGroups.member." ), ) kwargs = dict( name=self._get_param("Name"), log_uri=self._get_param("LogUri"), job_flow_role=self._get_param("JobFlowRole"), service_role=self._get_param("ServiceRole"), steps=steps_from_query_string(self._get_list_prefix("Steps.member")), visible_to_all_users=self._get_bool_param("VisibleToAllUsers", False), instance_attrs=instance_attrs, ) bootstrap_actions = self._get_list_prefix("BootstrapActions.member") if bootstrap_actions: for ba in bootstrap_actions: args = [] idx = 1 keyfmt = "script_bootstrap_action._args.member.{0}" key = keyfmt.format(idx) while key in ba: args.append(ba.pop(key)) idx += 1 key = keyfmt.format(idx) ba["args"] = args ba["script_path"] = ba.pop("script_bootstrap_action._path") kwargs["bootstrap_actions"] = bootstrap_actions configurations = self._get_list_prefix("Configurations.member") if configurations: for idx, config in enumerate(configurations, 1): for key in list(config.keys()): if key.startswith("properties."): config.pop(key) config["properties"] = {} map_items = self._get_map_prefix( "Configurations.member.{0}.Properties.entry".format(idx) ) config["properties"] = map_items kwargs["configurations"] = configurations release_label = self._get_param("ReleaseLabel") ami_version = self._get_param("AmiVersion") if release_label: kwargs["release_label"] = release_label if ami_version: message = ( "Only one AMI version and release label may be specified. " "Provided AMI: {0}, release label: {1}." ).format(ami_version, release_label) raise EmrError( error_type="ValidationException", message=message, template="error_json", ) else: if ami_version: kwargs["requested_ami_version"] = ami_version kwargs["running_ami_version"] = ami_version else: kwargs["running_ami_version"] = "1.0.0" custom_ami_id = self._get_param("CustomAmiId") if custom_ami_id: kwargs["custom_ami_id"] = custom_ami_id if release_label and ( ReleaseLabel(release_label) < ReleaseLabel("emr-5.7.0") ): message = "Custom AMI is not allowed" raise EmrError( error_type="ValidationException", message=message, template="error_json", ) elif ami_version: message = "Custom AMI is not supported in this version of EMR" raise EmrError( error_type="ValidationException", message=message, template="error_json", ) step_concurrency_level = self._get_param("StepConcurrencyLevel") if step_concurrency_level: kwargs["step_concurrency_level"] = step_concurrency_level security_configuration = self._get_param("SecurityConfiguration") if security_configuration: kwargs["security_configuration"] = security_configuration kerberos_attributes = {} kwargs["kerberos_attributes"] = kerberos_attributes realm = self._get_param("KerberosAttributes.Realm") if realm: kerberos_attributes["Realm"] = realm kdc_admin_password = self._get_param("KerberosAttributes.KdcAdminPassword") if kdc_admin_password: kerberos_attributes["KdcAdminPassword"] = kdc_admin_password cross_realm_principal_password = self._get_param( "KerberosAttributes.CrossRealmTrustPrincipalPassword" ) if cross_realm_principal_password: kerberos_attributes[ "CrossRealmTrustPrincipalPassword" ] = cross_realm_principal_password ad_domain_join_user = self._get_param("KerberosAttributes.ADDomainJoinUser") if ad_domain_join_user: kerberos_attributes["ADDomainJoinUser"] = ad_domain_join_user ad_domain_join_password = self._get_param( "KerberosAttributes.ADDomainJoinPassword" ) if ad_domain_join_password: kerberos_attributes["ADDomainJoinPassword"] = ad_domain_join_password cluster = self.backend.run_job_flow(**kwargs) applications = self._get_list_prefix("Applications.member") if applications: self.backend.add_applications(cluster.id, applications) else: self.backend.add_applications( cluster.id, [{"Name": "Hadoop", "Version": "0.18"}] ) instance_groups = self._get_list_prefix("Instances.InstanceGroups.member") if instance_groups: for ig in instance_groups: ig["instance_count"] = int(ig["instance_count"]) # Adding support to EbsConfiguration self._parse_ebs_configuration(ig) # Adding support for auto_scaling_policy Unflattener.unflatten_complex_params(ig, "auto_scaling_policy") instance_group_result = self.backend.add_instance_groups( cluster.id, instance_groups ) for i in range(0, len(instance_group_result)): self.backend.add_instances( cluster.id, instance_groups[i], instance_group_result[i] ) tags = self._get_list_prefix("Tags.member") if tags: self.backend.add_tags( cluster.id, dict((d["key"], d["value"]) for d in tags) ) template = self.response_template(RUN_JOB_FLOW_TEMPLATE) return template.render(cluster=cluster) def _has_key_prefix(self, key_prefix, value): for key in value: # iter on both keys and values if key.startswith(key_prefix): return True return False def _parse_ebs_configuration(self, instance_group): key_ebs_config = "ebs_configuration" ebs_configuration = dict() # Filter only EBS config keys for key in instance_group: if key.startswith(key_ebs_config): ebs_configuration[key] = instance_group[key] if len(ebs_configuration) > 0: # Key that should be extracted ebs_optimized = "ebs_optimized" ebs_block_device_configs = "ebs_block_device_configs" volume_specification = "volume_specification" size_in_gb = "size_in_gb" volume_type = "volume_type" iops = "iops" volumes_per_instance = "volumes_per_instance" key_ebs_optimized = "{0}._{1}".format(key_ebs_config, ebs_optimized) # EbsOptimized config if key_ebs_optimized in ebs_configuration: instance_group.pop(key_ebs_optimized) ebs_configuration[ebs_optimized] = ebs_configuration.pop( key_ebs_optimized ) # Ebs Blocks ebs_blocks = [] idx = 1 keyfmt = "{0}._{1}.member.{{}}".format( key_ebs_config, ebs_block_device_configs ) key = keyfmt.format(idx) while self._has_key_prefix(key, ebs_configuration): vlespc_keyfmt = "{0}._{1}._{{}}".format(key, volume_specification) vol_size = vlespc_keyfmt.format(size_in_gb) vol_iops = vlespc_keyfmt.format(iops) vol_type = vlespc_keyfmt.format(volume_type) ebs_block = dict() ebs_block[volume_specification] = dict() if vol_size in ebs_configuration: instance_group.pop(vol_size) ebs_block[volume_specification][size_in_gb] = int( ebs_configuration.pop(vol_size) ) if vol_iops in ebs_configuration: instance_group.pop(vol_iops) ebs_block[volume_specification][iops] = ebs_configuration.pop( vol_iops ) if vol_type in ebs_configuration: instance_group.pop(vol_type) ebs_block[volume_specification][ volume_type ] = ebs_configuration.pop(vol_type) per_instance = "{0}._{1}".format(key, volumes_per_instance) if per_instance in ebs_configuration: instance_group.pop(per_instance) ebs_block[volumes_per_instance] = int( ebs_configuration.pop(per_instance) ) if len(ebs_block) > 0: ebs_blocks.append(ebs_block) idx += 1 key = keyfmt.format(idx) if len(ebs_blocks) > 0: ebs_configuration[ebs_block_device_configs] = ebs_blocks instance_group[key_ebs_config] = ebs_configuration @generate_boto3_response("SetTerminationProtection") def set_termination_protection(self): termination_protection = self._get_bool_param("TerminationProtected") job_ids = self._get_multi_param("JobFlowIds.member") self.backend.set_termination_protection(job_ids, termination_protection) template = self.response_template(SET_TERMINATION_PROTECTION_TEMPLATE) return template.render() @generate_boto3_response("SetVisibleToAllUsers") def set_visible_to_all_users(self): visible_to_all_users = self._get_param("VisibleToAllUsers") job_ids = self._get_multi_param("JobFlowIds.member") self.backend.set_visible_to_all_users(job_ids, visible_to_all_users) template = self.response_template(SET_VISIBLE_TO_ALL_USERS_TEMPLATE) return template.render() @generate_boto3_response("TerminateJobFlows") def terminate_job_flows(self): job_ids = self._get_multi_param("JobFlowIds.member.") self.backend.terminate_job_flows(job_ids) template = self.response_template(TERMINATE_JOB_FLOWS_TEMPLATE) return template.render() @generate_boto3_response("PutAutoScalingPolicy") def put_auto_scaling_policy(self): cluster_id = self._get_param("ClusterId") cluster = self.backend.get_cluster(cluster_id) instance_group_id = self._get_param("InstanceGroupId") auto_scaling_policy = self._get_param("AutoScalingPolicy") instance_group = self.backend.put_auto_scaling_policy( instance_group_id, auto_scaling_policy ) template = self.response_template(PUT_AUTO_SCALING_POLICY) return template.render( cluster_id=cluster_id, cluster=cluster, instance_group=instance_group ) @generate_boto3_response("RemoveAutoScalingPolicy") def remove_auto_scaling_policy(self): cluster_id = self._get_param("ClusterId") instance_group_id = self._get_param("InstanceGroupId") instance_group = self.backend.put_auto_scaling_policy(instance_group_id, None) template = self.response_template(REMOVE_AUTO_SCALING_POLICY) return template.render(cluster_id=cluster_id, instance_group=instance_group) ADD_INSTANCE_GROUPS_TEMPLATE = """ {% for instance_group in instance_groups %} {{ instance_group.id }} {% endfor %} 2690d7eb-ed86-11dd-9877-6fad448a8419 """ ADD_JOB_FLOW_STEPS_TEMPLATE = """ {% for step in steps %} {{ step.id }} {% endfor %} df6f4f4a-ed85-11dd-9877-6fad448a8419 """ ADD_TAGS_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """ DESCRIBE_CLUSTER_TEMPLATE = """ {% for application in cluster.applications %} {{ application.name }} {{ application.version }} {% endfor %} {{ (not cluster.keep_job_flow_alive_when_no_steps)|lower }} {% for configuration in cluster.configurations %} {{ configuration['classification'] }} {% for key, value in configuration['properties'].items() %} {{ key }} {{ value }} {% endfor %} {% endfor %} {% if cluster.custom_ami_id is not none %} {{ cluster.custom_ami_id }} {% endif %} {% for each in cluster.additional_master_security_groups %} {{ each }} {% endfor %} {% for each in cluster.additional_slave_security_groups %} {{ each }} {% endfor %} {{ cluster.availability_zone }} {{ cluster.ec2_key_name }} {{ cluster.ec2_subnet_id }} {{ cluster.role }} {{ cluster.master_security_group }} {{ cluster.slave_security_group }} {{ cluster.service_access_security_group }} {{ cluster.id }} {% if 'Realm' in cluster.kerberos_attributes%} {{ cluster.kerberos_attributes['Realm'] }} {% endif %} {% if 'KdcAdminPassword' in cluster.kerberos_attributes%} {{ cluster.kerberos_attributes['KdcAdminPassword'] }} {% endif %} {% if 'CrossRealmTrustPrincipalPassword' in cluster.kerberos_attributes%} {{ cluster.kerberos_attributes['CrossRealmTrustPrincipalPassword'] }} {% endif %} {% if 'ADDomainJoinUser' in cluster.kerberos_attributes%} {{ cluster.kerberos_attributes['ADDomainJoinUser'] }} {% endif %} {% if 'ADDomainJoinPassword' in cluster.kerberos_attributes%} {{ cluster.kerberos_attributes['ADDomainJoinPassword'] }} {% endif %} {{ cluster.log_uri }} ec2-184-0-0-1.us-west-1.compute.amazonaws.com {{ cluster.name }} {{ cluster.normalized_instance_hours }} {% if cluster.release_label is not none %} {{ cluster.release_label }} {% endif %} {% if cluster.requested_ami_version is not none %} {{ cluster.requested_ami_version }} {% endif %} {% if cluster.running_ami_version is not none %} {{ cluster.running_ami_version }} {% endif %} {% if cluster.security_configuration is not none %} {{ cluster.security_configuration }} {% endif %} {{ cluster.service_role }} {{ cluster.state }} {% if cluster.last_state_change_reason is not none %} {{ cluster.last_state_change_reason }} {% endif %} USER_REQUEST {{ cluster.creation_datetime.isoformat() }} {% if cluster.end_datetime is not none %} {{ cluster.end_datetime.isoformat() }} {% endif %} {% if cluster.ready_datetime is not none %} {{ cluster.ready_datetime.isoformat() }} {% endif %} {% for tag_key, tag_value in cluster.tags.items() %} {{ tag_key }} {{ tag_value }} {% endfor %} {{ cluster.termination_protected|lower }} {{ cluster.visible_to_all_users|lower }} {{ cluster.step_concurrency_level }} {{ cluster.arn }} aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee """ DESCRIBE_JOB_FLOWS_TEMPLATE = """ {% for cluster in clusters %} {% if cluster.running_ami_version is not none %} {{ cluster.running_ami_version }} {% endif %} {% if cluster.bootstrap_actions %} {% for bootstrap_action in cluster.bootstrap_actions %} {{ bootstrap_action.name }} {% for arg in bootstrap_action.args %} {{ arg | escape }} {% endfor %} {{ bootstrap_action.script_path | escape }} {% endfor %} {% endif %} {{ cluster.creation_datetime.isoformat() }} {% if cluster.end_datetime is not none %} {{ cluster.end_datetime.isoformat() }} {% endif %} {% if cluster.last_state_change_reason is not none %} {{ cluster.last_state_change_reason }} {% endif %} {% if cluster.ready_datetime is not none %} {{ cluster.ready_datetime.isoformat() }} {% endif %} {% if cluster.start_datetime is not none %} {{ cluster.start_datetime.isoformat() }} {% endif %} {{ cluster.state }} {% if cluster.ec2_key_name is not none %} {{ cluster.ec2_key_name }} {% endif %} {% if cluster.ec2_subnet_id is not none %} {{ cluster.ec2_subnet_id }} {% endif %} {{ cluster.hadoop_version }} {{ cluster.instance_count }} {% for instance_group in cluster.instance_groups %} {% if instance_group.bid_price is not none %} {{ instance_group.bid_price }} {% endif %} {{ instance_group.creation_datetime.isoformat() }} {% if instance_group.end_datetime is not none %} {{ instance_group.end_datetime.isoformat() }} {% endif %} {{ instance_group.id }} {{ instance_group.num_instances }} {{ instance_group.role }} {{ instance_group.num_instances }} {{ instance_group.type }} {{ instance_group.market }} {{ instance_group.name }} {% if instance_group.ready_datetime is not none %} {{ instance_group.ready_datetime.isoformat() }} {% endif %} {% if instance_group.start_datetime is not none %} {{ instance_group.start_datetime.isoformat() }} {% endif %} {{ instance_group.state }} {% endfor %} {{ cluster.keep_job_flow_alive_when_no_steps|lower }} {{ cluster.master_instance_id }} {{ cluster.master_instance_type }} ec2-184-0-0-1.{{ cluster.region }}.compute.amazonaws.com {{ cluster.normalized_instance_hours }} {{ cluster.availability_zone }} {{ cluster.slave_instance_type }} {{ cluster.termination_protected|lower }} {{ cluster.id }} {{ cluster.role }} {{ cluster.log_uri }} {{ cluster.name }} {{ cluster.service_role }} {% for step in cluster.steps %} {{ step.creation_datetime.isoformat() }} {% if step.end_datetime is not none %} {{ step.end_datetime.isoformat() }} {% endif %} {% if step.last_state_change_reason is not none %} {{ step.last_state_change_reason }} {% endif %} {% if step.ready_datetime is not none %} {{ step.ready_datetime.isoformat() }} {% endif %} {% if step.start_datetime is not none %} {{ step.start_datetime.isoformat() }} {% endif %} {{ step.state }} {{ step.action_on_failure }} {{ step.jar }} {{ step.main_class }} {% for arg in step.args %} {{ arg | escape }} {% endfor %} {{ step.name | escape }} {% endfor %} {{ cluster.visible_to_all_users|lower }} {% endfor %} 9cea3229-ed85-11dd-9877-6fad448a8419 """ DESCRIBE_STEP_TEMPLATE = """ {{ step.action_on_failure }} {% for arg in step.args %} {{ arg | escape }} {% endfor %} {{ step.jar }} {% for key, val in step.properties.items() %} {{ key }} {{ val | escape }} {% endfor %} {{ step.id }} {{ step.name | escape }} {{ step.state }} {{ step.state_change_reason }} {{ step.creation_datetime.isoformat() }} {% if step.end_datetime is not none %} {{ step.end_datetime.isoformat() }} {% endif %} {% if step.ready_datetime is not none %} {{ step.start_datetime.isoformat() }} {% endif %} df6f4f4a-ed85-11dd-9877-6fad448a8419 """ LIST_BOOTSTRAP_ACTIONS_TEMPLATE = """ {% for bootstrap_action in bootstrap_actions %} {% for arg in bootstrap_action.args %} {{ arg | escape }} {% endfor %} {{ bootstrap_action.name }} {{ bootstrap_action.script_path }} {% endfor %} {% if marker is not none %} {{ marker }} {% endif %} df6f4f4a-ed85-11dd-9877-6fad448a8419 """ LIST_CLUSTERS_TEMPLATE = """ {% for cluster in clusters %} {{ cluster.id }} {{ cluster.name }} {{ cluster.normalized_instance_hours }} {{ cluster.state }} USER_REQUEST {% if cluster.last_state_change_reason is not none %} {{ cluster.last_state_change_reason }} {% endif %} {{ cluster.creation_datetime.isoformat() }} {% if cluster.end_datetime is not none %} {{ cluster.end_datetime.isoformat() }} {% endif %} {% if cluster.ready_datetime is not none %} {{ cluster.ready_datetime.isoformat() }} {% endif %} {{ cluster.arn }} {% endfor %} {% if marker is not none %} {{ marker }} {% endif %} 2690d7eb-ed86-11dd-9877-6fad448a8418 """ LIST_INSTANCE_GROUPS_TEMPLATE = """ {% for instance_group in instance_groups %} {% if instance_group.bid_price is not none %} {{ instance_group.bid_price }} {% endif %} {% if instance_group.ebs_configuration is not none %} {% for ebs_block_device in instance_group.ebs_configuration.ebs_block_device_configs %} {% for i in range(ebs_block_device.volumes_per_instance) %} {{ebs_block_device.volume_specification.volume_type}} {{ebs_block_device.volume_specification.iops}} {{ebs_block_device.volume_specification.size_in_gb}} /dev/sd{{i}} {% endfor %} {% endfor %} {% endif %} {% if instance_group.auto_scaling_policy is not none %} {% if instance_group.auto_scaling_policy.constraints is not none %} {% if instance_group.auto_scaling_policy.constraints.min_capacity is not none %} {{instance_group.auto_scaling_policy.constraints.min_capacity}} {% endif %} {% if instance_group.auto_scaling_policy.constraints.max_capacity is not none %} {{instance_group.auto_scaling_policy.constraints.max_capacity}} {% endif %} {% endif %} {% if instance_group.auto_scaling_policy.rules is not none %} {% for rule in instance_group.auto_scaling_policy.rules %} {% if 'name' in rule %} {{rule['name']}} {% endif %} {% if 'description' in rule %} {{rule['description']}} {% endif %} {% if 'action' in rule %} {% if 'market' in rule['action'] %} {{rule['action']['market']}} {% endif %} {% if 'simple_scaling_policy_configuration' in rule['action'] %} {% if 'adjustment_type' in rule['action']['simple_scaling_policy_configuration'] %} {{rule['action']['simple_scaling_policy_configuration']['adjustment_type']}} {% endif %} {% if 'scaling_adjustment' in rule['action']['simple_scaling_policy_configuration'] %} {{rule['action']['simple_scaling_policy_configuration']['scaling_adjustment']}} {% endif %} {% if 'cool_down' in rule['action']['simple_scaling_policy_configuration'] %} {{rule['action']['simple_scaling_policy_configuration']['cool_down']}} {% endif %} {% endif %} {% endif %} {% if 'trigger' in rule %} {% if 'cloud_watch_alarm_definition' in rule['trigger'] %} {% if 'comparison_operator' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['comparison_operator']}} {% endif %} {% if 'evaluation_periods' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['evaluation_periods']}} {% endif %} {% if 'metric_name' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['metric_name']}} {% endif %} {% if 'namespace' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['namespace']}} {% endif %} {% if 'period' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['period']}} {% endif %} {% if 'statistic' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['statistic']}} {% endif %} {% if 'threshold' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['threshold']}} {% endif %} {% if 'unit' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['unit']}} {% endif %} {% if 'dimensions' in rule['trigger']['cloud_watch_alarm_definition'] %} {% for dimension in rule['trigger']['cloud_watch_alarm_definition']['dimensions'] %} {% if 'key' in dimension %} {{dimension['key']}} {% endif %} {% if 'value' in dimension %} {{dimension['value']}} {% endif %} {% endfor %} {% endif %} {% endif %} {% endif %} {% endfor %} {% endif %} {% if instance_group.auto_scaling_policy.status is not none %} {% if 'state' in instance_group.auto_scaling_policy.status %} {{instance_group.auto_scaling_policy.status['state']}} {% endif %} {% endif %} {% endif %} {% if instance_group.ebs_optimized is not none %} {{ instance_group.ebs_optimized }} {% endif %} {{ instance_group.id }} {{ instance_group.role }} {{ instance_group.type }} {{ instance_group.market }} {{ instance_group.name }} {{ instance_group.num_instances }} {{ instance_group.num_instances }} {{ instance_group.state }} {% if instance_group.state_change_reason is not none %} {{ instance_group.state_change_reason }} {% endif %} USER_REQUEST {{ instance_group.creation_datetime.isoformat() }} {% if instance_group.end_datetime is not none %} {{ instance_group.end_datetime.isoformat() }} {% endif %} {% if instance_group.ready_datetime is not none %} {{ instance_group.ready_datetime.isoformat() }} {% endif %} {% endfor %} {% if marker is not none %} {{ marker }} {% endif %} 8296d8b8-ed85-11dd-9877-6fad448a8419 """ LIST_INSTANCES_TEMPLATE = """ {% for instance in instances %} {{ instance.id }} {{ instance.ec2_instance_id }} {{ instance.details.public_dns }} {{ instance.details.public_ip }} {{ instance.details.private_dns }} {{ instance.details.private_ip }} {{ instance.instance_group.id }} {{ instance.instance_fleet_id }} {{ instance.instance_group.market }} {{ instance.details.instance_type }} {% for volume in instance.details.block_device_mapping %} {{ volume }} {{ instance.details.block_device_mapping[volume].volume_id }} {% endfor %} {{ instance.instance_group.state }} {% if instance.state_change_reason is not none %} {{ instance.state_change_reason }} {% endif %} {{ instance.instance_group.creation_datetime.isoformat() }} {% if instance.instance_group.end_datetime is not none %} {{ instance.instance_group.end_datetime.isoformat() }} {% endif %} {% if instance.instance_group.ready_datetime is not none %} {{ instance.instance_group.ready_datetime.isoformat() }} {% endif %} {% endfor %} 4248c46c-71c0-4772-b155-0e992dc30027 """ LIST_STEPS_TEMPLATE = """ {% for step in steps %} {{ step.action_on_failure }} {% for arg in step.args %} {{ arg | escape }} {% endfor %} {{ step.jar | escape }} {% for key, val in step.properties.items() %} {{ key }} {{ val | escape }} {% endfor %} {{ step.id }} {{ step.name | escape }} {{ step.state }} {{ step.state_change_reason }} {{ step.creation_datetime.isoformat() }} {% if step.end_datetime is not none %} {{ step.end_datetime.isoformat() }} {% endif %} {% if step.start_datetime is not none %} {{ step.start_datetime.isoformat() }} {% endif %} {% endfor %} {% if marker is not none %} {{ marker }} {% endif %} df6f4f4a-ed85-11dd-9877-6fad448a8419 """ MODIFY_CLUSTER_TEMPLATE = """ {{ cluster.step_concurrency_level }} 0751c837-e78d-4aef-95c9-9c4d29a092ff """ MODIFY_INSTANCE_GROUPS_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """ REMOVE_TAGS_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """ RUN_JOB_FLOW_TEMPLATE = """ {{ cluster.id }} {{ cluster.arn }} 8296d8b8-ed85-11dd-9877-6fad448a8419 """ SET_TERMINATION_PROTECTION_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """ SET_VISIBLE_TO_ALL_USERS_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """ TERMINATE_JOB_FLOWS_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """ PUT_AUTO_SCALING_POLICY = """ {{cluster_id}} {{instance_group.id}} {% if instance_group.auto_scaling_policy is not none %} {% if instance_group.auto_scaling_policy.constraints is not none %} {% if instance_group.auto_scaling_policy.constraints.min_capacity is not none %} {{instance_group.auto_scaling_policy.constraints.min_capacity}} {% endif %} {% if instance_group.auto_scaling_policy.constraints.max_capacity is not none %} {{instance_group.auto_scaling_policy.constraints.max_capacity}} {% endif %} {% endif %} {% if instance_group.auto_scaling_policy.rules is not none %} {% for rule in instance_group.auto_scaling_policy.rules %} {% if 'name' in rule %} {{rule['name']}} {% endif %} {% if 'description' in rule %} {{rule['description']}} {% endif %} {% if 'action' in rule %} {% if 'market' in rule['action'] %} {{rule['action']['market']}} {% endif %} {% if 'simple_scaling_policy_configuration' in rule['action'] %} {% if 'adjustment_type' in rule['action']['simple_scaling_policy_configuration'] %} {{rule['action']['simple_scaling_policy_configuration']['adjustment_type']}} {% endif %} {% if 'scaling_adjustment' in rule['action']['simple_scaling_policy_configuration'] %} {{rule['action']['simple_scaling_policy_configuration']['scaling_adjustment']}} {% endif %} {% if 'cool_down' in rule['action']['simple_scaling_policy_configuration'] %} {{rule['action']['simple_scaling_policy_configuration']['cool_down']}} {% endif %} {% endif %} {% endif %} {% if 'trigger' in rule %} {% if 'cloud_watch_alarm_definition' in rule['trigger'] %} {% if 'comparison_operator' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['comparison_operator']}} {% endif %} {% if 'evaluation_periods' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['evaluation_periods']}} {% endif %} {% if 'metric_name' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['metric_name']}} {% endif %} {% if 'namespace' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['namespace']}} {% endif %} {% if 'period' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['period']}} {% endif %} {% if 'statistic' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['statistic']}} {% endif %} {% if 'threshold' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['threshold']}} {% endif %} {% if 'unit' in rule['trigger']['cloud_watch_alarm_definition'] %} {{rule['trigger']['cloud_watch_alarm_definition']['unit']}} {% endif %} {% if 'dimensions' in rule['trigger']['cloud_watch_alarm_definition'] %} {% for dimension in rule['trigger']['cloud_watch_alarm_definition']['dimensions'] %} {% if 'key' in dimension %} {{dimension['key']}} {% endif %} {% if 'value' in dimension %} {{dimension['value']}} {% endif %} {% endfor %} {% endif %} {% endif %} {% endif %} {% endfor %} {% endif %} {% if instance_group.auto_scaling_policy.status is not none %} {% if 'state' in instance_group.auto_scaling_policy.status %} {{instance_group.auto_scaling_policy.status['state']}} {% endif %} {% endif %} {% endif %} {{ cluster.arn }} d47379d9-b505-49af-9335-a68950d82535 """ REMOVE_AUTO_SCALING_POLICY = """ c04a1042-5340-4c0a-a7b5-7779725ce4f7 """ CREATE_SECURITY_CONFIGURATION_TEMPLATE = """ {{name}} {{creation_date_time}} 2690d7eb-ed86-11dd-9877-6fad448a8419 """ DESCRIBE_SECURITY_CONFIGURATION_TEMPLATE = """ {{security_configuration['name']}} {{security_configuration['security_configuration']}} {{security_configuration['creation_date_time']}} 2690d7eb-ed86-11dd-9877-6fad448a8419 """ DELETE_SECURITY_CONFIGURATION_TEMPLATE = """ 2690d7eb-ed86-11dd-9877-6fad448a8419 """