def create_image_template( # pylint: disable=too-many-locals cmd, client, resource_group_name, image_template_name, location=None, source_dict=None, scripts_list=None, destinations_lists=None, build_timeout=None, tags=None, source=None, scripts=None, checksum=None, managed_image_destinations=None, # pylint: disable=unused-argument shared_image_destinations=None, no_wait=False): # pylint: disable=unused-argument, too-many-locals from azure.mgmt.imagebuilder.models import (ImageTemplate, ImageTemplateSharedImageVersionSource, ImageTemplatePlatformImageSource, ImageTemplateIsoSource, ImageTemplateManagedImageSource, # pylint: disable=line-too-long ImageTemplateShellCustomizer, ImageTemplatePowerShellCustomizer, ImageTemplateManagedImageDistributor, ImageTemplateSharedImageDistributor) # pylint: disable=line-too-long template_source, template_scripts, template_destinations = None, [], [] # create image template source settings if source_dict['type'] == _SourceType.PLATFORM_IMAGE: template_source = ImageTemplatePlatformImageSource(**source_dict) elif source_dict['type'] == _SourceType.ISO_URI: template_source = ImageTemplateIsoSource(**source_dict) elif source_dict['type'] == _SourceType.MANAGED_IMAGE: template_source = ImageTemplateManagedImageSource(**source_dict) elif source_dict['type'] == _SourceType.SIG_VERSION: template_source = ImageTemplateSharedImageVersionSource(**source_dict) # create image template customizer settings # Script structure can be found in _parse_script's function definition for script in scripts_list: script.pop("is_url") script["script_uri"] = script.pop("script") if script["type"] == ScriptType.SHELL: template_scripts.append(ImageTemplateShellCustomizer(**script)) elif script["type"] == ScriptType.POWERSHELL: template_scripts.append(ImageTemplatePowerShellCustomizer(**script)) else: # Should never happen logger.debug("Script %s has type %s", script["script"], script["type"]) raise CLIError("Script {} has an invalid type.".format(script["script"])) # create image template distribution / destination settings for dest_type, rid, loc_info in destinations_lists: parsed = parse_resource_id(rid) if dest_type == _DestType.MANAGED_IMAGE: template_destinations.append(ImageTemplateManagedImageDistributor( image_id=rid, location=loc_info, run_output_name=parsed['name'])) elif dest_type == _DestType.SHARED_IMAGE_GALLERY: template_destinations.append(ImageTemplateSharedImageDistributor( gallery_image_id=rid, replication_regions=loc_info, run_output_name=parsed['child_name_1'])) else: logger.info("No applicable destination found for destination %s", str(tuple([dest_type, rid, loc_info]))) image_template = ImageTemplate(source=template_source, customize=template_scripts, distribute=template_destinations, location=location, build_timeout_in_minutes=build_timeout, tags=(tags or {})) return cached_put(cmd, client.virtual_machine_image_templates.create_or_update, parameters=image_template, resource_group_name=resource_group_name, image_template_name=image_template_name)
def add_template_customizer(cmd, client, resource_group_name, image_template_name, customizer_name, customizer_type, script_url=None, inline_script=None, valid_exit_codes=None, restart_command=None, restart_check_command=None, restart_timeout=None, file_source=None, dest_path=None, search_criteria=None, filters=None, update_limit=None): _require_defer(cmd) from azure.mgmt.imagebuilder.models import (ImageTemplateShellCustomizer, ImageTemplatePowerShellCustomizer, ImageTemplateRestartCustomizer, ImageTemplateFileCustomizer, ImageTemplateWindowsUpdateCustomizer) existing_image_template = cached_get(cmd, client.virtual_machine_image_templates.get, resource_group_name=resource_group_name, image_template_name=image_template_name) if existing_image_template.customize is None: existing_image_template.customize = [] else: for existing_customizer in existing_image_template.customize: if existing_customizer.name == customizer_name: raise CLIError("Output with output name {} already exists in image template {}." .format(customizer_name, image_template_name)) new_customizer = None if customizer_type.lower() == ScriptType.SHELL.value.lower(): # pylint:disable=no-member new_customizer = ImageTemplateShellCustomizer(name=customizer_name, script_uri=script_url, inline=inline_script) elif customizer_type.lower() == ScriptType.POWERSHELL.value.lower(): # pylint:disable=no-member new_customizer = ImageTemplatePowerShellCustomizer(name=customizer_name, script_uri=script_url, inline=inline_script, valid_exit_codes=valid_exit_codes) elif customizer_type.lower() == ScriptType.WINDOWS_RESTART.value.lower(): # pylint:disable=no-member new_customizer = ImageTemplateRestartCustomizer(name=customizer_name, restart_command=restart_command, restart_check_command=restart_check_command, restart_timeout=restart_timeout) elif customizer_type.lower() == ScriptType.FILE.value.lower(): # pylint:disable=no-member new_customizer = ImageTemplateFileCustomizer(name=customizer_name, source_uri=file_source, destination=dest_path) elif customizer_type.lower() == ScriptType.WINDOWS_UPDATE.value.lower(): new_customizer = ImageTemplateWindowsUpdateCustomizer(name=customizer_name, search_criteria=search_criteria, filters=filters, update_limit=update_limit) if not new_customizer: raise CLIError("Cannot determine customizer from type {}.".format(customizer_type)) existing_image_template.customize.append(new_customizer) return cached_put(cmd, client.virtual_machine_image_templates.create_or_update, parameters=existing_image_template, resource_group_name=resource_group_name, image_template_name=image_template_name)
def create_image_template( # pylint: disable=too-many-locals, too-many-branches, too-many-statements, unused-argument cmd, client, resource_group_name, image_template_name, location=None, source_dict=None, scripts_list=None, destinations_lists=None, build_timeout=None, tags=None, source=None, scripts=None, checksum=None, managed_image_destinations=None, shared_image_destinations=None, no_wait=False, image_template=None, identity=None, vm_size=None, os_disk_size=None, vnet=None, subnet=None, proxy_vm_size=None, build_vm_identities=None): from azure.mgmt.imagebuilder.models import ( ImageTemplate, ImageTemplateSharedImageVersionSource, ImageTemplatePlatformImageSource, ImageTemplateManagedImageSource, ImageTemplateShellCustomizer, ImageTemplatePowerShellCustomizer, ImageTemplateManagedImageDistributor, ImageTemplateSharedImageDistributor, ImageTemplateIdentity, ComponentsVrq145SchemasImagetemplateidentityPropertiesUserassignedidentitiesAdditionalproperties, # pylint: disable=line-too-long ImageTemplateVmProfile, VirtualNetworkConfig) if image_template is not None: logger.warning( 'You are using --image-template. All other parameters will be ignored.' ) if os.path.exists(image_template): # Local file with open(image_template) as f: content = f.read() else: # It should be an URL msg = '\nusage error: --image-template is not a correct local path or URL' try: r = requests.get(image_template) except Exception: raise CLIError(traceback.format_exc() + msg) if r.status_code != 200: raise CLIError(traceback.format_exc() + msg) content = r.content try: obj = json.loads(content) except json.JSONDecodeError: raise CLIError( traceback.format_exc() + '\nusage error: Content of --image-template is not a valid JSON string' ) content = {} if 'properties' in obj: content = obj['properties'] if 'location' in obj: content['location'] = obj['location'] if 'tags' in obj: content['tags'] = obj['tags'] if 'identity' in obj: content['identity'] = obj['identity'] return client.virtual_machine_image_templates.begin_create_or_update( parameters=content, resource_group_name=resource_group_name, image_template_name=image_template_name) template_source, template_scripts, template_destinations = None, [], [] # create image template source settings if source_dict['type'] == _SourceType.PLATFORM_IMAGE: template_source = ImageTemplatePlatformImageSource(**source_dict) elif source_dict['type'] == _SourceType.ISO_URI: # It was supported before but is removed in the current service version. raise CLIError('usage error: Source type ISO URI is not supported.') elif source_dict['type'] == _SourceType.MANAGED_IMAGE: template_source = ImageTemplateManagedImageSource(**source_dict) elif source_dict['type'] == _SourceType.SIG_VERSION: template_source = ImageTemplateSharedImageVersionSource(**source_dict) # create image template customizer settings # Script structure can be found in _parse_script's function definition for script in scripts_list: script.pop("is_url") script["script_uri"] = script.pop("script") if script["type"] == ScriptType.SHELL: template_scripts.append(ImageTemplateShellCustomizer(**script)) elif script["type"] == ScriptType.POWERSHELL: template_scripts.append( ImageTemplatePowerShellCustomizer(**script)) else: # Should never happen logger.debug("Script %s has type %s", script["script"], script["type"]) raise CLIError("Script {} has an invalid type.".format( script["script"])) # create image template distribution / destination settings for dest_type, rid, loc_info in destinations_lists: parsed = parse_resource_id(rid) if dest_type == _DestType.MANAGED_IMAGE: template_destinations.append( ImageTemplateManagedImageDistributor( image_id=rid, location=loc_info, run_output_name=parsed['name'])) elif dest_type == _DestType.SHARED_IMAGE_GALLERY: template_destinations.append( ImageTemplateSharedImageDistributor( gallery_image_id=rid, replication_regions=loc_info, run_output_name=parsed['child_name_1'])) else: logger.info("No applicable destination found for destination %s", str(tuple([dest_type, rid, loc_info]))) # Identity identity_body = None if identity is not None: subscription_id = get_subscription_id(cmd.cli_ctx) user_assigned_identities = {} for ide in identity: if not is_valid_resource_id(ide): ide = resource_id(subscription=subscription_id, resource_group=resource_group_name, namespace='Microsoft.ManagedIdentity', type='userAssignedIdentities', name=ide) user_assigned_identities[ide] = ComponentsVrq145SchemasImagetemplateidentityPropertiesUserassignedidentitiesAdditionalproperties() # pylint: disable=line-too-long identity_body = ImageTemplateIdentity( type='UserAssigned', user_assigned_identities=user_assigned_identities) # VM profile vnet_config = None if vnet or subnet: if not is_valid_resource_id(subnet): subscription_id = get_subscription_id(cmd.cli_ctx) subnet = resource_id(subscription=subscription_id, resource_group=resource_group_name, namespace='Microsoft.Network', type='virtualNetworks', name=vnet, child_type_1='subnets', child_name_1=subnet) vnet_config = VirtualNetworkConfig(subnet_id=subnet) if proxy_vm_size is not None: if subnet is not None: vnet_config = VirtualNetworkConfig(subnet_id=subnet, proxy_vm_size=proxy_vm_size) else: raise RequiredArgumentMissingError( 'Usage error: --proxy-vm-size is only configurable when --subnet is specified.' ) vm_profile = ImageTemplateVmProfile(vm_size=vm_size, os_disk_size_gb=os_disk_size, user_assigned_identities=build_vm_identities, vnet_config=vnet_config) # pylint: disable=line-too-long image_template = ImageTemplate(source=template_source, customize=template_scripts, distribute=template_destinations, location=location, build_timeout_in_minutes=build_timeout, tags=(tags or {}), identity=identity_body, vm_profile=vm_profile) return cached_put( cmd, client.virtual_machine_image_templates.begin_create_or_update, parameters=image_template, resource_group_name=resource_group_name, image_template_name=image_template_name)
def create_image_template( # pylint: disable=too-many-locals, too-many-branches, too-many-statements cmd, client, resource_group_name, image_template_name, location=None, source_dict=None, scripts_list=None, destinations_lists=None, build_timeout=None, tags=None, source=None, scripts=None, checksum=None, managed_image_destinations=None, # pylint: disable=unused-argument shared_image_destinations=None, no_wait=False, image_template=None): # pylint: disable=unused-argument, too-many-locals from azure.mgmt.imagebuilder.models import (ImageTemplate, ImageTemplateSharedImageVersionSource, ImageTemplatePlatformImageSource, ImageTemplateIsoSource, ImageTemplateManagedImageSource, # pylint: disable=line-too-long ImageTemplateShellCustomizer, ImageTemplatePowerShellCustomizer, ImageTemplateManagedImageDistributor, ImageTemplateSharedImageDistributor) # pylint: disable=line-too-long if image_template is not None: if os.path.exists(image_template): # Local file with open(image_template) as f: content = f.read() else: # It should be an URL msg = '\nusage error: --image-template is not a correct local path or URL' try: r = requests.get(image_template) except Exception: raise CLIError(traceback.format_exc() + msg) if r.status_code != 200: raise CLIError(traceback.format_exc() + msg) content = r.content try: obj = json.loads(content) except json.JSONDecodeError: raise CLIError( traceback.format_exc() + '\nusage error: Content of --image-template is not a valid JSON string' ) content = {} if 'properties' in obj: content = obj['properties'] if 'location' in obj: content['location'] = obj['location'] if 'tags' in obj: content['tags'] = obj['tags'] return client.virtual_machine_image_templates.create_or_update( parameters=content, resource_group_name=resource_group_name, image_template_name=image_template_name) template_source, template_scripts, template_destinations = None, [], [] # create image template source settings if source_dict['type'] == _SourceType.PLATFORM_IMAGE: template_source = ImageTemplatePlatformImageSource(**source_dict) elif source_dict['type'] == _SourceType.ISO_URI: template_source = ImageTemplateIsoSource(**source_dict) elif source_dict['type'] == _SourceType.MANAGED_IMAGE: template_source = ImageTemplateManagedImageSource(**source_dict) elif source_dict['type'] == _SourceType.SIG_VERSION: template_source = ImageTemplateSharedImageVersionSource(**source_dict) # create image template customizer settings # Script structure can be found in _parse_script's function definition for script in scripts_list: script.pop("is_url") script["script_uri"] = script.pop("script") if script["type"] == ScriptType.SHELL: template_scripts.append(ImageTemplateShellCustomizer(**script)) elif script["type"] == ScriptType.POWERSHELL: template_scripts.append( ImageTemplatePowerShellCustomizer(**script)) else: # Should never happen logger.debug("Script %s has type %s", script["script"], script["type"]) raise CLIError("Script {} has an invalid type.".format( script["script"])) # create image template distribution / destination settings for dest_type, rid, loc_info in destinations_lists: parsed = parse_resource_id(rid) if dest_type == _DestType.MANAGED_IMAGE: template_destinations.append( ImageTemplateManagedImageDistributor( image_id=rid, location=loc_info, run_output_name=parsed['name'])) elif dest_type == _DestType.SHARED_IMAGE_GALLERY: template_destinations.append( ImageTemplateSharedImageDistributor( gallery_image_id=rid, replication_regions=loc_info, run_output_name=parsed['child_name_1'])) else: logger.info("No applicable destination found for destination %s", str(tuple([dest_type, rid, loc_info]))) image_template = ImageTemplate(source=template_source, customize=template_scripts, distribute=template_destinations, location=location, build_timeout_in_minutes=build_timeout, tags=(tags or {})) return cached_put(cmd, client.virtual_machine_image_templates.create_or_update, parameters=image_template, resource_group_name=resource_group_name, image_template_name=image_template_name)