From 9d11cd311b64969c9597a0955064506c35e63dc5 Mon Sep 17 00:00:00 2001 From: Zim Kalinowski Date: Thu, 21 Feb 2019 05:50:46 +0800 Subject: [PATCH] adding vmss instance module (#51461) * adding vmss instance * fix indent * fixing doc --- ...azure_rm_virtualmachinescalesetinstance.py | 270 ++++++++++++++++++ .../tasks/main.yml | 37 +++ 2 files changed, 307 insertions(+) create mode 100644 lib/ansible/modules/cloud/azure/azure_rm_virtualmachinescalesetinstance.py diff --git a/lib/ansible/modules/cloud/azure/azure_rm_virtualmachinescalesetinstance.py b/lib/ansible/modules/cloud/azure/azure_rm_virtualmachinescalesetinstance.py new file mode 100644 index 00000000000..d54f73eb5ab --- /dev/null +++ b/lib/ansible/modules/cloud/azure/azure_rm_virtualmachinescalesetinstance.py @@ -0,0 +1,270 @@ +#!/usr/bin/python +# +# Copyright (c) 2019 Zim Kalinowski, (@zikalino) +# +# 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_virtualmachinescalesetinstance +version_added: "2.8" +short_description: Get Azure Virtual Machine Scale Set Instance facts. +description: + - Get facts of Azure Virtual Machine Scale Set VMs. + +options: + resource_group: + description: + - The name of the resource group. + required: True + vmss_name: + description: + - The name of the VM scale set. + required: True + instance_id: + description: + - The instance ID of the virtual machine. + required: True + latest_model: + type: bool + description: + - Set to C(yes) to upgrade to the latest model. + power_state: + description: + - Use this option to change power state of the instance. + required: True + choices: + - 'running' + - 'stopped' + - 'deallocated' + state: + description: + - Assert the state of the VMSS instance. Use 'present' to update a instance and + 'absent' to delete instance. + default: present + choices: + - absent + - present + +extends_documentation_fragment: + - azure + +author: + - "Zim Kalinowski (@zikalino)" + +''' + +EXAMPLES = ''' + - name: Upgrade instance to the latest image + azure_rm_computevirtualmachinescalesetinstance: + resource_group: myResourceGroup + vmss_name: myVMSS + instance_id: "2" + latest_model: yes +''' + +RETURN = ''' +instances: + description: A list of instances. + returned: always + type: complex + contains: + id: + description: + - Instance resource ID + returned: always + type: str + sample: /subscriptions/xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx/resourceGroups/TestGroup/providers/Microsoft.Compute/scalesets/myscaleset/vms/myvm +''' + +from ansible.module_utils.azure_rm_common import AzureRMModuleBase + +try: + from msrestazure.azure_exceptions import CloudError + from azure.mgmt.compute import ComputeManagementClient + from msrest.serialization import Model +except ImportError: + # This is handled in azure_rm_common + pass + + +class AzureRMVirtualMachineScaleSetInstance(AzureRMModuleBase): + def __init__(self): + # define user inputs into argument + self.module_arg_spec = dict( + resource_group=dict( + type='str', + required=True + ), + vmss_name=dict( + type='str', + required=True + ), + instance_id=dict( + type='str' + ), + latest_model=dict( + type='bool' + ), + power_state=dict( + type='str', + choices=['running', 'stopped', 'deallocated'] + ), + state=dict( + type='str', + default='present', + choices=['present', 'absent'] + ) + ) + # store the results of the module operation + self.results = dict( + changed=False + ) + self.mgmt_client = None + self.resource_group = None + self.vmss_name = None + self.instance_id = None + self.latest_model = None + self.power_state = None + self.state = None + super(AzureRMVirtualMachineScaleSetInstance, self).__init__(self.module_arg_spec, supports_tags=False) + + def exec_module(self, **kwargs): + for key in self.module_arg_spec: + setattr(self, key, kwargs[key]) + self.mgmt_client = self.get_mgmt_svc_client(ComputeManagementClient, + base_url=self._cloud_environment.endpoints.resource_manager) + + instances = self.get() + + if self.state == 'absent': + for item in instances: + if not self.check_mode: + self.delete(item['instance_id']) + self.results['changed'] = True + self.results['instances'] = [] + else: + if self.latest_model is not None: + for item in instances: + if not item.get('latest_model', None): + if not self.check_mode: + self.apply_latest_model(item['instance_id']) + item['latest_model'] = True + self.results['changed'] = True + + if self.power_state is not None: + for item in instances: + if self.power_state == 'stopped' and item['power_state'] not in ['stopped', 'stopping']: + if not self.check_mode: + self.stop(item['instance_id']) + self.results['changed'] = True + elif self.power_state == 'deallocated' and item['power_state'] not in ['deallocated']: + if not self.check_mode: + self.deallocate(item['instance_id']) + self.results['changed'] = True + elif self.power_state == 'running' and item['power_state'] not in ['running']: + if not self.check_mode: + self.start(item['instance_id']) + self.results['changed'] = True + + self.results['instances'] = [{'id': item['id']} for item in instances] + return self.results + + def get(self): + response = None + results = [] + try: + response = self.mgmt_client.virtual_machine_scale_set_vms.get(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_id=self.instance_id) + self.log("Response : {0}".format(response)) + except CloudError as e: + self.log('Could not get facts for Virtual Machine Scale Set VM.') + + if response: + results.append(self.format_response(response)) + + return results + + def apply_latest_model(self, instance_id): + try: + poller = self.compute_client.virtual_machine_scale_sets.update_instances(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_ids=[instance_id]) + self.get_poller_result(poller) + except CloudError as exc: + self.log("Error applying latest model {0} - {1}".format(self.name, str(exc))) + self.fail("Error applying latest model {0} - {1}".format(self.name, str(exc))) + + def delete(self, instance_id): + try: + self.mgmt_client.virtual_machine_scale_set_vms.delete(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_id=instance_id) + except CloudError as e: + self.log('Could not delete instance of Virtual Machine Scale Set VM.') + self.fail('Could not delete instance of Virtual Machine Scale Set VM.') + + def start(self, instance_id): + try: + self.mgmt_client.virtual_machine_scale_set_vms.start(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_id=instance_id) + except CloudError as e: + self.log('Could not start instance of Virtual Machine Scale Set VM.') + self.fail('Could not start instance of Virtual Machine Scale Set VM.') + + def stop(self, instance_id): + try: + self.mgmt_client.virtual_machine_scale_set_vms.power_off(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_id=instance_id) + except CloudError as e: + self.log('Could not stop instance of Virtual Machine Scale Set VM.') + self.fail('Could not stop instance of Virtual Machine Scale Set VM.') + + def deallocate(self, instance_id): + try: + self.mgmt_client.virtual_machine_scale_set_vms.deallocate(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_id=instance_id) + except CloudError as e: + self.log('Could not deallocate instance of Virtual Machine Scale Set VM.') + self.fail('Could not deallocate instance of Virtual Machine Scale Set VM.') + + def format_response(self, item): + d = item.as_dict() + iv = self.mgmt_client.virtual_machine_scale_set_vms.get_instance_view(resource_group_name=self.resource_group, + vm_scale_set_name=self.vmss_name, + instance_id=d.get('instance_id', None)).as_dict() + power_state = "" + for index in range(len(iv['statuses'])): + code = iv['statuses'][index]['code'].split('/') + if code[0] == 'PowerState': + power_state = code[1] + break + d = { + 'id': d.get('id'), + 'tags': d.get('tags'), + 'instance_id': d.get('instance_id'), + 'latest_model': d.get('latest_model_applied'), + 'power_state': power_state + } + return d + + +def main(): + AzureRMVirtualMachineScaleSetInstance() + + +if __name__ == '__main__': + main() diff --git a/test/integration/targets/azure_rm_virtualmachine_scaleset/tasks/main.yml b/test/integration/targets/azure_rm_virtualmachine_scaleset/tasks/main.yml index 053785a196a..894d953a7ac 100644 --- a/test/integration/targets/azure_rm_virtualmachine_scaleset/tasks/main.yml +++ b/test/integration/targets/azure_rm_virtualmachine_scaleset/tasks/main.yml @@ -344,6 +344,43 @@ assert: that: results.changed +- name: Upgrade instance to the latest image + azure_rm_virtualmachinescalesetinstance: + resource_group: "{{ resource_group }}" + vmss_name: testVMSS{{ rpfx }} + instance_id: "{{ instances.instances[0].instance_id }}" + latest_model: yes + register: results + +- name: Assert that something has changed + assert: + that: results.changed + +- name: Stop virtual machine + azure_rm_virtualmachinescalesetinstance: + resource_group: "{{ resource_group }}" + vmss_name: testVMSS{{ rpfx }} + instance_id: "{{ instances.instances[0].instance_id }}" + power_state: stopped + register: results + +- name: Assert that something has changed + assert: + that: results.changed + +- name: Delete instance + azure_rm_virtualmachinescalesetinstance: + resource_group: "{{ resource_group }}" + vmss_name: testVMSS{{ rpfx }} + instance_id: "{{ instances.instances[0].instance_id }}" + state: absent + register: results + +- name: Assert that something has changed + assert: + that: results.changed + + - name: Delete VMSS azure_rm_virtualmachine_scaleset: resource_group: "{{ resource_group }}"