1
0
mirror of https://github.com/vmware/vsphere-automation-sdk-python.git synced 2024-11-27 03:30:00 -05:00
vsphere-automation-sdk-python/vsphere/samples/vcenter/vm/hardware/boot_device.py
2017-01-16 21:24:30 -08:00

137 lines
4.7 KiB
Python

#!/usr/bin/env python
"""
* *******************************************************
* Copyright (c) VMware, Inc. 2016. All Rights Reserved.
* *******************************************************
*
* DISCLAIMER. THIS PROGRAM IS PROVIDED TO YOU "AS IS" WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, WHETHER ORAL OR WRITTEN,
* EXPRESS OR IMPLIED. THE AUTHOR SPECIFICALLY DISCLAIMS ANY IMPLIED
* WARRANTIES OR CONDITIONS OF MERCHANTABILITY, SATISFACTORY QUALITY,
* NON-INFRINGEMENT AND FITNESS FOR A PARTICULAR PURPOSE.
"""
from com.vmware.vcenter.vm.hardware.boot_client import Device as BootDevice
from com.vmware.vcenter.vm.hardware_client import (Disk, Ethernet)
from vsphere.samples.common import vapiconnect
from vsphere.samples.common.sample_util import parse_cli_args_vm
from vsphere.samples.common.sample_util import pp
from vsphere.samples.vcenter.helper.vm_helper import get_vm
from vsphere.samples.vcenter.setup import testbed
"""
Demonstrates how to modify the boot devices used by a virtual machine, and in
what order they are tried.
Sample Prerequisites:
The sample needs an existing VM with the following configuration:
1 Ethernet adapter
1 Cdrom
1 Floppy drive
3 Disks
"""
vm = None
vm_name = None
stub_config = None
boot_device_svc = None
cleardata = False
orig_boot_device_entries = None
def setup(context=None):
global vm, vm_name, stub_config, cleardata
if context:
# Run sample suite via setup script
stub_config = context.stub_config
vm_name = testbed.config['VM_NAME_EXHAUSTIVE']
else:
# Run sample in standalone mode
server, username, password, cleardata, skip_verification, vm_name = \
parse_cli_args_vm(testbed.config['VM_NAME_EXHAUSTIVE'])
stub_config = vapiconnect.connect(server,
username,
password,
skip_verification)
def run():
global vm
vm = get_vm(stub_config, vm_name)
if not vm:
exit('Sample requires an existing vm with name ({}). '
'Please create the vm first.'.format(vm_name))
print("Using VM '{}' ({}) for BootDevice Sample".format(vm_name, vm))
# Create BootDevice stub used for making requests
global boot_device_svc
boot_device_svc = BootDevice(stub_config)
print('\n# Example: Get current BootDevice configuration')
boot_device_entries = boot_device_svc.get(vm)
print('vm.hardware.boot.Device.get({}) -> {}'.
format(vm, pp(boot_device_entries)))
# Save current BootDevice info to verify that we have cleaned up properly
global orig_boot_device_entries
orig_boot_device_entries = boot_device_entries
# Get device identifiers for Disks
disk_svc = Disk(stub_config)
disk_summaries = disk_svc.list(vm)
print('vm.hardware.Disk.list({}) -> {}'.format(vm, pp(disk_summaries)))
disks = [disk_summary.disk for disk_summary in disk_summaries]
# Get device identifiers for Ethernet nics
ethernet_svc = Ethernet(stub_config)
nic_summaries = ethernet_svc.list(vm)
print('vm.hardware.Ethernet.list({}) -> {}'.format(vm, pp(nic_summaries)))
nics = [nic_summary.nic for nic_summary in nic_summaries]
print('\n# Example: Set Boot Order to be Floppy, '
'Disk1, Disk2, Disk3, Cdrom,')
print('# Network (nic0), Network (nic1).')
boot_device_entries = [
BootDevice.Entry(BootDevice.Type.FLOPPY),
BootDevice.Entry(BootDevice.Type.DISK, disks=disks),
BootDevice.Entry(BootDevice.Type.CDROM)]
for nic in nics:
boot_device_entries.append(
BootDevice.Entry(BootDevice.Type.ETHERNET, nic=nic))
print('vm.hardware.boot.Device.set({}, {})'.format(vm, boot_device_entries))
boot_device_svc.set(vm, boot_device_entries)
boot_device_entries = boot_device_svc.get(vm)
print('vm.hardware.boot.Device.get({}) -> {}'.
format(vm, pp(boot_device_entries)))
def cleanup():
print('\n# Cleanup: Revert BootDevice configuration')
boot_device_entries = orig_boot_device_entries
print('vm.hardware.boot.Device.set({}, {})'.format(vm, boot_device_entries))
boot_device_svc.set(vm, boot_device_entries)
boot_device_entries = boot_device_svc.get(vm)
print('vm.hardware.boot.Device.get({}) -> {}'.
format(vm, pp(boot_device_entries)))
if boot_device_entries != orig_boot_device_entries:
print('vm.hardware.boot.Device WARNING: '
'Final BootDevice info does not match original')
def main():
try:
setup()
run()
if cleardata:
cleanup()
finally:
if stub_config:
vapiconnect.logout(stub_config)
if __name__ == '__main__':
main()