valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-08 01:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor vm_tools_upgrade()

Browse Source
This commit is contained in:
Nitin Madhok 2015-04-16 17:54:43 -04:00
parent d024374d2e
commit 72c5152ac8
1 changed files with 38 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
salt/cloud/clouds/vmware.py
View File

@ -1568,11 +1568,15 @@ def create_cluster(kwargs=None, call=None):
return False return False
def vm_tools_upgrade(vm_name, reboot=False, call=None): def vm_tools_upgrade(name, reboot=False, call=None):
''' '''
To check status and upgrade VMware Tools on a VM using vm name. To upgrade VMware Tools on a specified virtual machine.
Default is ``reboot=False`` which attempts to suppress the reboot
on Windows VMs. Use ``reboot=True`` to override. .. note::
If the virtual machine is running Windows OS, use ``reboot=True``
to reboot the virtual machine after VMware tools upgrade. Default
is ``reboot=False``
CLI Example: CLI Example:
@ -1583,56 +1587,38 @@ def vm_tools_upgrade(vm_name, reboot=False, call=None):
''' '''
if call != 'action': if call != 'action':
raise SaltCloudSystemExit( raise SaltCloudSystemExit(
'The start action must be called with -a or --action.' 'The vm_tools_upgrade action must be called with -a or --action.'
) )
vm = _get_mor_by_property(vim.VirtualMachine, vm_name) vm = _get_mor_by_property(vim.VirtualMachine, name)
# Check if VM is powered on and vmware tools is up to date. tools_status = vm.guest.toolsStatus
if vm.summary.runtime.powerState == "poweredOn":
if vm.guest.toolsStatus == "toolsOk":
ret = 'VMware tools is up to date.'
log.info('VM {0} {1}'.format(vm_name, ret))
return ret
# If vmware tools is out of date, check major OS family. # Exit if VMware tools is already up to date
# Upgrade tools on Linux guests. if tools_status == "toolsOk":
elif vm.guest.toolsStatus == "toolsOld": return 'VMware tools is already up to date'
if vm.guest.guestFamily == "linuxGuest":
# Exit if VM is not powered on
if vm.summary.runtime.powerState != "poweredOn":
return 'Tools cannot be upgraded if the VM is not powered on'
# If vmware tools is out of date, check major OS family
# Upgrade tools on Linux and Windows guests
if tools_status == "toolsOld":
log.info('Upgrading VMware tools on {0}'.format(name))
os_family = vm.guest.guestFamily
try: try:
log.info('Upgrading VMware tools on {0}'.format(vm_name)) if os_family == "windowsGuest" and not reboot:
task = vm.UpgradeTools() log.info('Reboot suppressed on {0}'.format(name))
_wait_for_task(task, vm_name, "Upgrade tools", 5, "info")
except Exception as exc:
log.error('Could not upgrade VMware tools on VM {0}: {1}'.format(vm_name, exc))
return 'failed to upgrade VMware tools'
# If Windows, check if reboot is true.
elif vm.guest.guestFamily == "windowsGuest":
if reboot:
try:
log.info('Upgrading VMware tools on {0}'.format(vm_name))
task = vm.UpgradeTools()
_wait_for_task(task, vm_name, "Upgrade tools", 5, "info")
except Exception as exc:
log.error('Could not upgrade VMware tools on VM {0}: {1}'.format(vm_name, exc))
return 'failed to upgrade VMware tools'
# If reboot is false, upgrade tools while supressing the reboot.
else:
try:
log.info('Upgrading VMware tools on {0}'.format(vm_name))
task = vm.UpgradeTools('/S /v"/qn REBOOT=R"') task = vm.UpgradeTools('/S /v"/qn REBOOT=R"')
_wait_for_task(task, vm_name, "Upgrade tools", 5, "info") elif os_family in ["linuxGuest", "windowsGuest"]:
task = vm.UpgradeTools()
else:
return 'VMware tools upgrade is currently supported only on Linux or Windows guests.'
_wait_for_task(task, name, "tools upgrade", 5, "info")
except Exception as exc: except Exception as exc:
log.error('Could not upgrade VMware tools on VM {0}: {1}'.format(vm_name, exc)) log.error('Could not upgrade VMware tools on VM {0}: {1}'.format(name, exc))
return 'failed to upgrade VMware tools' return 'failed to upgrade VMware tools'
return 'VMware tools upgraded' return 'VMware tools upgraded'
# If vm is powered off, exit. return 'VMware tools is not installed'
elif vm.summary.runtime.powerState == "poweredOff":
ret = 'Tools cannot be upgraded on powered off VMs.'
log.info('VM {0} {1}'.format(vm_name, ret))
return ret