adding vmss instance module (#51461)

* adding vmss instance

* fix indent

* fixing doc
pull/52688/head
Zim Kalinowski 6 years ago committed by Matt Davis
parent a6752f002b
commit 9d11cd311b

@ -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()

@ -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 }}"

Loading…
Cancel
Save