mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
Add azure image module support create and delete (#32589)
* Add azure image module support create and delete * Add os_type description * fix lint * fix lint * fix lint * fix line ending * add enum_modules for serialize * update for python3 * Fix mirror * fix lint * fix as pr comments * fix name * fix syntax * fix * none check * fix test * fix syntax * fix example * return only id in the top level * doc * doc * fix * fix test
This commit is contained in:
parent
0c7c30b47a
commit
37ce7ef0db
5 changed files with 485 additions and 0 deletions
|
@ -111,6 +111,7 @@ except ImportError as exc:
|
|||
try:
|
||||
from enum import Enum
|
||||
from msrestazure.azure_exceptions import CloudError
|
||||
from msrestazure.tools import resource_id, is_valid_resource_id
|
||||
from msrestazure import azure_cloud
|
||||
from azure.mgmt.network.models import PublicIPAddress, NetworkSecurityGroup, SecurityRule, NetworkInterface, \
|
||||
NetworkInterfaceIPConfiguration, Subnet
|
||||
|
@ -152,6 +153,13 @@ def azure_id_to_dict(id):
|
|||
return result
|
||||
|
||||
|
||||
def format_resource_id(val, subscription_id, namespace, types, resource_group):
|
||||
return resource_id(name=val,
|
||||
resource_group=resource_group,
|
||||
namespace=namespace,
|
||||
type=types,
|
||||
subscription=subscription_id) if not is_valid_resource_id(val) else val
|
||||
|
||||
AZURE_PKG_VERSIONS = {
|
||||
StorageManagementClient.__name__: {
|
||||
'package_name': 'storage',
|
||||
|
|
318
lib/ansible/modules/cloud/azure/azure_rm_image.py
Normal file
318
lib/ansible/modules/cloud/azure/azure_rm_image.py
Normal file
|
@ -0,0 +1,318 @@
|
|||
#!/usr/bin/python
|
||||
#
|
||||
# Copyright (c) 2017 Yuwei Zhou, <yuwzho@microsoft.com>
|
||||
#
|
||||
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||||
|
||||
from __future__ import absolute_import, division, print_function
|
||||
__metaclass__ = type
|
||||
|
||||
|
||||
ANSIBLE_METADATA = {'metadata_version': '1.1',
|
||||
'status': ['preview'],
|
||||
'supported_by': 'community'}
|
||||
|
||||
|
||||
DOCUMENTATION = '''
|
||||
---
|
||||
module: azure_rm_image
|
||||
version_added: "2.5"
|
||||
short_description: Manage Azure image.
|
||||
description:
|
||||
- Create, delete an image from virtual machine, blob uri, managed disk or snapshot.
|
||||
options:
|
||||
resource_group:
|
||||
description:
|
||||
- Name of resource group.
|
||||
required: true
|
||||
name:
|
||||
description:
|
||||
- Name of the image.
|
||||
required: true
|
||||
source:
|
||||
description:
|
||||
- OS disk source from the same region, including a virtual machine id or name,
|
||||
OS disk blob uri, managed OS disk id or name, or OS snapshot id or name.
|
||||
required: true
|
||||
data_disk_sources:
|
||||
description:
|
||||
- List of data disk sources, including unmanaged blob uri, managed disk id or name, or snapshot id or name.
|
||||
location:
|
||||
description:
|
||||
- Location of the image. Derived from I(resource_group) if not specified.
|
||||
os_type:
|
||||
description: The OS type of image.
|
||||
choices:
|
||||
- Windows
|
||||
- Linux
|
||||
state:
|
||||
description:
|
||||
- Assert the state of the image. Use C(present) to create or update a image and C(absent) to delete an image.
|
||||
default: present
|
||||
choices:
|
||||
- absent
|
||||
- present
|
||||
|
||||
extends_documentation_fragment:
|
||||
- azure
|
||||
- azure_tags
|
||||
|
||||
author:
|
||||
- "Yuwei Zhou (@yuwzho)"
|
||||
|
||||
'''
|
||||
|
||||
EXAMPLES = '''
|
||||
- name: Create an image from a virtual machine
|
||||
azure_rm_image:
|
||||
resource_group: Testing
|
||||
name: foobar
|
||||
source: testvm001
|
||||
|
||||
- name: Create an image from os disk
|
||||
azure_rm_image:
|
||||
resource_group: Testing
|
||||
name: foobar
|
||||
source: /subscriptions/XXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXX/resourceGroups/Testing/providers/Microsoft.Compute/disks/disk001
|
||||
data_disk_sources:
|
||||
- datadisk001
|
||||
- datadisk002
|
||||
os_type: Linux
|
||||
|
||||
- name: Delete an image
|
||||
azure_rm_image:
|
||||
state: absent
|
||||
resource_group: Testing
|
||||
name: foobar
|
||||
source: testvm001
|
||||
'''
|
||||
|
||||
RETURN = '''
|
||||
id:
|
||||
description: Image resource path.
|
||||
type: str
|
||||
returned: success
|
||||
example: "/subscriptions/XXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXX/resourceGroups/Testing/providers/Microsoft.Compute/images/foobar"
|
||||
''' # NOQA
|
||||
|
||||
from ansible.module_utils.azure_rm_common import AzureRMModuleBase, format_resource_id
|
||||
|
||||
try:
|
||||
from msrestazure.tools import parse_resource_id
|
||||
from msrestazure.azure_exceptions import CloudError
|
||||
from azure.mgmt.compute.models import SubResource, OperatingSystemStateTypes, ImageStorageProfile, Image, ImageOSDisk, ImageDataDisk
|
||||
except ImportError:
|
||||
# This is handled in azure_rm_common
|
||||
pass
|
||||
|
||||
|
||||
class AzureRMImage(AzureRMModuleBase):
|
||||
|
||||
def __init__(self):
|
||||
|
||||
self.module_arg_spec = dict(
|
||||
resource_group=dict(type='str', required=True),
|
||||
name=dict(type='str', required=True),
|
||||
state=dict(type='str', default='present', choices=['present', 'absent']),
|
||||
location=dict(type='str'),
|
||||
source=dict(type='str'),
|
||||
data_disk_sources=dict(type='list', default=[]),
|
||||
os_type=dict(type='str', choices=['Windows', 'Linux'])
|
||||
)
|
||||
|
||||
self.results = dict(
|
||||
changed=False,
|
||||
id=None
|
||||
)
|
||||
|
||||
required_if = [
|
||||
('state', 'present', ['source'])
|
||||
]
|
||||
|
||||
self.resource_group = None
|
||||
self.name = None
|
||||
self.state = None
|
||||
self.location = None
|
||||
self.source = None
|
||||
self.data_disk_sources = None
|
||||
self.os_type = None
|
||||
self.tags = None
|
||||
|
||||
super(AzureRMImage, self).__init__(self.module_arg_spec, supports_check_mode=True, required_if=required_if)
|
||||
|
||||
def exec_module(self, **kwargs):
|
||||
|
||||
for key in self.module_arg_spec:
|
||||
setattr(self, key, kwargs[key])
|
||||
|
||||
results = None
|
||||
changed = False
|
||||
image = None
|
||||
|
||||
if not self.location:
|
||||
# Set default location
|
||||
resource_group = self.get_resource_group(self.resource_group)
|
||||
self.location = resource_group.location
|
||||
|
||||
self.log('Fetching image {0}'.format(self.name))
|
||||
image = self.get_image()
|
||||
if image:
|
||||
self.check_provisioning_state(image, self.state)
|
||||
results = image.id
|
||||
# update is not supported
|
||||
if self.state == 'absent':
|
||||
changed = True
|
||||
# the image does not exist and create a new one
|
||||
elif self.state == 'present':
|
||||
changed = True
|
||||
|
||||
self.results['changed'] = changed
|
||||
self.results['id'] = results
|
||||
|
||||
if changed:
|
||||
if self.state == 'present':
|
||||
image_instance = None
|
||||
# create from virtual machine
|
||||
vm = self.get_source_vm()
|
||||
if vm:
|
||||
if self.data_disk_sources:
|
||||
self.fail('data_disk_sources is not allowed when capturing image from vm')
|
||||
image_instance = Image(self.location, source_virtual_machine=SubResource(vm.id))
|
||||
else:
|
||||
if not self.os_type:
|
||||
self.fail('os_type is required to create the image')
|
||||
os_disk = self.create_os_disk()
|
||||
data_disks = self.create_data_disks()
|
||||
storage_profile = ImageStorageProfile(os_disk=os_disk, data_disks=data_disks)
|
||||
image_instance = Image(self.location, storage_profile=storage_profile, tags=self.tags)
|
||||
|
||||
# finally make the change if not check mode
|
||||
if not self.check_mode and image_instance:
|
||||
new_image = self.create_image(image_instance)
|
||||
self.results['id'] = new_image.id
|
||||
|
||||
elif self.state == 'absent':
|
||||
if not self.check_mode:
|
||||
# delete image
|
||||
self.delete_image()
|
||||
# the delete does not actually return anything. if no exception, then we'll assume it worked.
|
||||
self.results['id'] = None
|
||||
|
||||
return self.results
|
||||
|
||||
def resolve_storage_source(self, source):
|
||||
blob_uri = None
|
||||
disk = None
|
||||
snapshot = None
|
||||
if source.lower().endswith('.vhd'):
|
||||
blob_uri = source
|
||||
return (blob_uri, disk, snapshot)
|
||||
|
||||
tokenize = parse_resource_id(source)
|
||||
if tokenize.get('type') == 'disks':
|
||||
disk = source
|
||||
return (blob_uri, disk, snapshot)
|
||||
|
||||
if tokenize.get('type') == 'snapshots':
|
||||
snapshot = source
|
||||
return (blob_uri, disk, snapshot)
|
||||
|
||||
# not a disk or snapshots
|
||||
if 'type' in tokenize:
|
||||
return (blob_uri, disk, snapshot)
|
||||
|
||||
# source can be name of snapshot or disk
|
||||
snapshot_instance = self.get_snapshot(source)
|
||||
if snapshot_instance:
|
||||
snapshot = snapshot_instance.id
|
||||
return (blob_uri, disk, snapshot)
|
||||
|
||||
disk_instance = self.get_disk(source)
|
||||
if disk_instance:
|
||||
disk = disk_instance.id
|
||||
return (blob_uri, disk, snapshot)
|
||||
|
||||
def create_os_disk(self):
|
||||
blob_uri, disk, snapshot = self.resolve_storage_source(self.source)
|
||||
snapshot_resource = SubResource(snapshot) if snapshot else None
|
||||
managed_disk = SubResource(disk) if disk else None
|
||||
return ImageOSDisk(os_type=self.os_type,
|
||||
os_state=OperatingSystemStateTypes.generalized,
|
||||
snapshot=snapshot_resource,
|
||||
managed_disk=managed_disk,
|
||||
blob_uri=blob_uri)
|
||||
|
||||
def create_data_disk(self, lun, source):
|
||||
blob_uri, disk, snapshot = self.resolve_storage_source(source)
|
||||
if blob_uri or disk or snapshot:
|
||||
snapshot_resource = SubResource(snapshot) if snapshot else None
|
||||
managed_disk = SubResource(disk) if disk else None
|
||||
return ImageDataDisk(lun,
|
||||
blob_uri=blob_uri,
|
||||
snapshot=snapshot_resource,
|
||||
managed_disk=managed_disk)
|
||||
|
||||
def create_data_disks(self):
|
||||
return list(filter(None, [self.create_data_disk(lun, source) for lun, source in enumerate(self.data_disk_sources)]))
|
||||
|
||||
def get_source_vm(self):
|
||||
vm_resource_id = format_resource_id(self.source,
|
||||
self.subscription_id,
|
||||
'Microsoft.Compute',
|
||||
'virtualMachines',
|
||||
self.resource_group)
|
||||
resource = parse_resource_id(vm_resource_id)
|
||||
return self.get_vm(resource['resource_group'], resource['name']) if resource['type'] == 'virtualMachines' else None
|
||||
|
||||
def get_snapshot(self, snapshot_name):
|
||||
return self._get_resource(self.compute_client.snapshots.get, self.resource_group, snapshot_name)
|
||||
|
||||
def get_disk(self, disk_name):
|
||||
return self._get_resource(self.compute_client.disks.get, self.resource_group, disk_name)
|
||||
|
||||
def get_vm(self, resource_group, vm_name):
|
||||
return self._get_resource(self.compute_client.virtual_machines.get, resource_group, vm_name, 'instanceview')
|
||||
|
||||
def get_image(self):
|
||||
return self._get_resource(self.compute_client.images.get, self.resource_group, self.name)
|
||||
|
||||
def _get_resource(self, get_method, resource_group, name, expand=None):
|
||||
try:
|
||||
if expand:
|
||||
return get_method(resource_group, name, expand=expand)
|
||||
else:
|
||||
return get_method(resource_group, name)
|
||||
except CloudError as cloud_err:
|
||||
# Return None iff the resource is not found
|
||||
if cloud_err.status_code == 404:
|
||||
self.log('{0}'.format(str(cloud_err)))
|
||||
return None
|
||||
self.fail('Error: failed to get resource {0} - {1}'.format(name, str(cloud_err)))
|
||||
except Exception as exc:
|
||||
self.fail('Error: failed to get resource {0} - {1}'.format(name, str(exc)))
|
||||
|
||||
def create_image(self, image):
|
||||
try:
|
||||
poller = self.compute_client.images.create_or_update(self.resource_group, self.name, image)
|
||||
new_image = self.get_poller_result(poller)
|
||||
except Exception as exc:
|
||||
self.fail("Error creating image {0} - {1}".format(self.name, str(exc)))
|
||||
self.check_provisioning_state(new_image)
|
||||
return new_image
|
||||
|
||||
def delete_image(self):
|
||||
self.log('Deleting image {0}'.format(self.name))
|
||||
try:
|
||||
poller = self.compute_client.images.delete(self.resource_group, self.name)
|
||||
result = self.get_poller_result(poller)
|
||||
except Exception as exc:
|
||||
self.fail("Error deleting image {0} - {1}".format(self.name, str(exc)))
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def main():
|
||||
AzureRMImage()
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
3
test/integration/targets/azure_rm_image/aliases
Normal file
3
test/integration/targets/azure_rm_image/aliases
Normal file
|
@ -0,0 +1,3 @@
|
|||
cloud/azure
|
||||
posix/ci/cloud/group2/azure
|
||||
destructive
|
2
test/integration/targets/azure_rm_image/meta/main.yml
Normal file
2
test/integration/targets/azure_rm_image/meta/main.yml
Normal file
|
@ -0,0 +1,2 @@
|
|||
dependencies:
|
||||
- setup_azure
|
154
test/integration/targets/azure_rm_image/tasks/main.yml
Normal file
154
test/integration/targets/azure_rm_image/tasks/main.yml
Normal file
|
@ -0,0 +1,154 @@
|
|||
- name: Create storage account name
|
||||
set_fact:
|
||||
storage_account: "{{ resource_group | hash('md5') | truncate(24, True, '') }}"
|
||||
|
||||
- name: Create storage account
|
||||
azure_rm_storageaccount:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: "{{ storage_account }}"
|
||||
account_type: Standard_LRS
|
||||
|
||||
- name: Create virtual network
|
||||
azure_rm_virtualnetwork:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm001
|
||||
address_prefixes: "10.10.0.0/16"
|
||||
|
||||
- name: Add subnet
|
||||
azure_rm_subnet:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm001
|
||||
address_prefix: "10.10.0.0/24"
|
||||
virtual_network: testvm001
|
||||
|
||||
- name: Create public ip
|
||||
azure_rm_publicipaddress:
|
||||
resource_group: "{{ resource_group }}"
|
||||
allocation_method: Static
|
||||
name: testvm001
|
||||
|
||||
- name: Create security group
|
||||
azure_rm_securitygroup:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm001
|
||||
|
||||
- name: Create NIC
|
||||
azure_rm_networkinterface:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm001
|
||||
virtual_network: testvm001
|
||||
subnet: testvm001
|
||||
public_ip_name: testvm001
|
||||
security_group: testvm001
|
||||
|
||||
- name: Create virtual machine
|
||||
azure_rm_virtualmachine:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm002
|
||||
vm_size: Standard_A0
|
||||
storage_account: "{{ storage_account }}"
|
||||
storage_container: testvm001
|
||||
storage_blob: testvm001.vhd
|
||||
admin_username: adminuser
|
||||
admin_password: Password123!
|
||||
os_type: Linux
|
||||
network_interfaces: testvm001
|
||||
image:
|
||||
offer: UbuntuServer
|
||||
publisher: Canonical
|
||||
sku: 16.04-LTS
|
||||
version: latest
|
||||
|
||||
- name: Deallocate the virtual machine
|
||||
azure_rm_virtualmachine:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm002
|
||||
allocated: no
|
||||
vm_size: Standard_A0
|
||||
register: output
|
||||
|
||||
- name: Start the virtual machine
|
||||
azure_rm_virtualmachine:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm002
|
||||
vm_size: Standard_A0
|
||||
|
||||
- name: Create an image from VM (check mode)
|
||||
azure_rm_image:
|
||||
resource_group: "{{ resource_group }}"
|
||||
source: "https://{{ storage_account }}.blob.core.windows.net/testvm001/testvm001.vhd"
|
||||
name: testimage001
|
||||
os_type: Linux
|
||||
check_mode: yes
|
||||
register: output
|
||||
|
||||
- assert:
|
||||
that: output.changed
|
||||
|
||||
- name: Create an image from VM
|
||||
azure_rm_image:
|
||||
resource_group: "{{ resource_group }}"
|
||||
source: "https://{{ storage_account }}.blob.core.windows.net/testvm001/testvm001.vhd"
|
||||
name: testimage001
|
||||
os_type: Linux
|
||||
register: output
|
||||
|
||||
- assert:
|
||||
that:
|
||||
- output.changed
|
||||
- output.id
|
||||
|
||||
- name: Create an image from VM (idempotent)
|
||||
azure_rm_image:
|
||||
resource_group: "{{ resource_group }}"
|
||||
source: "https://{{ storage_account }}.blob.core.windows.net/testvm001/testvm001.vhd"
|
||||
name: testimage001
|
||||
os_type: Linux
|
||||
register: output
|
||||
|
||||
- assert:
|
||||
that:
|
||||
- not output.changed
|
||||
- output.id
|
||||
|
||||
- name: Delete image (check mode)
|
||||
azure_rm_image:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testimage001
|
||||
state: absent
|
||||
register: output
|
||||
check_mode: yes
|
||||
|
||||
- assert:
|
||||
that:
|
||||
- output.changed
|
||||
|
||||
- name: Delete image
|
||||
azure_rm_image:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testimage001
|
||||
state: absent
|
||||
register: output
|
||||
|
||||
- assert:
|
||||
that:
|
||||
- output.changed
|
||||
|
||||
- name: Delete image (idempotent)
|
||||
azure_rm_image:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testimage001
|
||||
state: absent
|
||||
register: output
|
||||
|
||||
- assert:
|
||||
that:
|
||||
- not output.changed
|
||||
|
||||
- name: Delete VM
|
||||
azure_rm_virtualmachine:
|
||||
resource_group: "{{ resource_group }}"
|
||||
name: testvm002
|
||||
state: absent
|
||||
vm_size: Standard_A0
|
||||
register: output
|
Loading…
Reference in a new issue