VMware: Refactor vmware_guest_tools_wait (#36098)

Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com>

lib/ansible/modules/cloud/vmware/vmware_guest_tools_wait.py

@@ -1,15 +1,16 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
-
 # Copyright (c) 2017 Philippe Dellaert <philippe@dellaert.org>
 # 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',
+ANSIBLE_METADATA = {
+    'metadata_version': '1.1',
     'status': ['preview'],
-                    'supported_by': 'community'}
+    'supported_by': 'community'
+}
 
 
 DOCUMENTATION = '''
@@ -17,7 +18,7 @@ DOCUMENTATION = '''
 module: vmware_guest_tools_wait
 short_description: Wait for VMware tools to become available
 description:
-    - Wait for VMware tools to become available on the VM and return facts.
+    - This module can be used to wait for VMware tools to become available on the given VM and return facts.
 version_added: '2.4'
 author:
     - Philippe Dellaert (@pdellaert) <philippe@dellaert.org>
@@ -39,7 +40,7 @@ options:
    folder:
      description:
      - Destination folder, absolute or relative path to find an existing guest.
-            - This is required if C(name) is supplied.
+     - This is required only, if multiple VMs with same C(name) is found.
      - The folder should include the datacenter. ESX's datacenter is C(ha-datacenter).
      - 'Examples:'
      - '   folder: /ha-datacenter/vm'
@@ -51,11 +52,10 @@ options:
      - '   folder: /folder1/datacenter1/vm'
      - '   folder: folder1/datacenter1/vm'
      - '   folder: /folder1/datacenter1/vm/folder2'
-        default: /vm
    uuid:
      description:
      - UUID of the VM  for which to wait until the tools become available, if known. This is VMware's unique identifier.
-            - This is required if C(name) is not supplied.
+     - This is required, if C(name) is not supplied.
 extends_documentation_fragment: vmware.documentation
 '''
 
@@ -94,37 +94,19 @@ import time
 
 from ansible.module_utils.basic import AnsibleModule
 from ansible.module_utils._text import to_native
-from ansible.module_utils.vmware import connect_to_api, gather_vm_facts, vmware_argument_spec, find_vm_by_id
+from ansible.module_utils.vmware import PyVmomi, gather_vm_facts, vmware_argument_spec
+
 
-HAS_PYVMOMI = False
 try:
     import pyVmomi
     from pyVmomi import vim
-
-    HAS_PYVMOMI = True
 except ImportError:
     pass
 
 
-class PyVmomiHelper(object):
+class PyVmomiHelper(PyVmomi):
     def __init__(self, module):
-        if not HAS_PYVMOMI:
-            module.fail_json(msg='pyvmomi module required')
-
-        self.module = module
-        self.params = module.params
-        self.content = connect_to_api(self.module)
-
-    def getvm(self, name=None, uuid=None, folder=None):
-        vm = None
-        match_first = False
-        if uuid:
-            vm = find_vm_by_id(self.content, vm_id=uuid, vm_id_type="uuid")
-        elif folder and name:
-            if self.params['name_match'] == 'first':
-                match_first = True
-            vm = find_vm_by_id(self.content, vm_id=name, vm_id_type="inventory_path", folder=folder, match_first=match_first)
-        return vm
+        super(PyVmomiHelper, self).__init__(module)
 
     def gather_facts(self, vm):
         return gather_vm_facts(self.content, vm)
@@ -133,9 +115,8 @@ class PyVmomiHelper(object):
         tools_running = False
         vm_facts = {}
         poll_num = 0
-        vm_uuid = vm.config.uuid
         while not tools_running and poll_num <= poll:
-            newvm = self.getvm(uuid=vm_uuid)
+            newvm = self.get_vm()
             vm_facts = self.gather_facts(newvm)
             if vm_facts['guest_tools_status'] == 'guestToolsRunning':
                 tools_running = True
@@ -157,38 +138,39 @@ def main():
     argument_spec.update(
         name=dict(type='str'),
         name_match=dict(type='str', default='first'),
-        folder=dict(type='str', default='/vm'),
+        folder=dict(type='str'),
         uuid=dict(type='str'),
     )
     module = AnsibleModule(
         argument_spec=argument_spec,
-        required_one_of=[['name', 'uuid']],
-        required_together=['name', 'folder']
-    )
+        required_one_of=[['name', 'uuid']])
 
+    if module.params['folder']:
         # FindByInventoryPath() does not require an absolute path
         # so we should leave the input folder path unmodified
         module.params['folder'] = module.params['folder'].rstrip('/')
 
     pyv = PyVmomiHelper(module)
     # Check if the VM exists before continuing
-    vm = pyv.getvm(name=module.params['name'],
-                   folder=module.params['folder'],
-                   uuid=module.params['uuid'])
+    vm = pyv.get_vm()
 
     if not vm:
-        vm_id = module.params.get('name') or module.params.get('uuid')
-        module.fail_json(msg="Unable to wait for tools for non-existing VM {0:s}".format(vm_id))
+        module.fail_json(msg="Unable to wait for VMware tools for "
+                             "non-existing VM '%s'." % (module.params.get('name') or
+                                                        module.params.get('uuid')))
 
+    result = dict(changed=False)
     try:
         result = pyv.wait_for_tools(vm)
     except Exception as e:
-        module.fail_json(msg="Waiting for tools failed with exception: {0:s}".format(to_native(e)))
+        module.fail_json(msg="Waiting for VMware tools failed with"
+                             " exception: {0:s}".format(to_native(e)))
 
     if result['failed']:
         module.fail_json(**result)
     else:
         module.exit_json(**result)
 
+
 if __name__ == '__main__':
     main()