mirror of https://github.com/ansible/ansible.git
adding vmss instance module (#51461)
* adding vmss instance * fix indent * fixing docpull/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()
|
||||
Loading…
Reference in New Issue