feat: Added ability to specify custom AMI for EMR.

batch/library/emr

@@ -100,6 +100,7 @@ class ElasticMapreduceCluster():
     def get_boto_base_specs(self, arguments):
         ami_version = arguments.pop('ami_version', DEFAULT_AMI_VERSION)
         release_label = arguments.pop('release_label')
+        custom_ami = arguments.pop('custom_ami')
         log_uri = arguments.pop('log_uri')
         job_flow_role = arguments.pop('job_flow_role')
         service_role = arguments.pop('service_role')
@@ -136,6 +137,9 @@ class ElasticMapreduceCluster():
             if applications is not None:
                 parameters['NewSupportedProducts'] = self.get_boto_application_specs(applications)
 
+        if custom_ami:
+            parameters['CustomAmiId'] = custom_ami
+
         return parameters
 
     def get_boto_instance_specs(self, arguments):
@@ -526,6 +530,7 @@ def main():
             log_uri              = dict(default=None, type='str'),
             ami_version          = dict(type='str'),
             release_label        = dict(default=None, type='str'),
+            custom_ami           = dict(default=None, type='str'),
             visible_to_all_users = dict(type='bool'),
             job_flow_role        = dict(default=None, type='str'),
             service_role         = dict(default='EMR_DefaultRole', type='str'),

batch/provision.yml

@@ -39,6 +39,7 @@
     - role: emr
     - ami_version: 2.4.11
     - release_label: ''
+    - custom_ami: ''
     - vpc_subnet_id: !!null
     - user_info: []
     - log_uri: !!null
@@ -57,6 +58,7 @@
         job_flow_role: "{{ role }}"
         ami_version: "{{ ami_version }}"
         release_label: "{{ release_label }}"
+        custom_ami: "{{ custom_ami }}"
         instance_groups: "{{ instance_groups }}"
         instance_fleets: "{{ instance_fleets }}"
         bootstrap_actions: "{{ bootstrap_actions }}"