mirror of
https://github.com/vmware/vsphere-automation-sdk-python.git
synced 2024-11-23 01:49:59 -05:00
214 lines
7.2 KiB
Python
214 lines
7.2 KiB
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.
|
|
"""
|
|
|
|
__author__ = 'VMware, Inc.'
|
|
__copyright__ = 'Copyright 2016 VMware, Inc. All rights reserved.'
|
|
|
|
|
|
import pyVim.task
|
|
from com.vmware.vcenter_client import (Folder, Host)
|
|
from pyVmomi import vim
|
|
|
|
|
|
def detect_host(context, host_name):
|
|
"""Find host based on host name"""
|
|
host_svc = Host(context.stub_config)
|
|
names = set([host_name])
|
|
|
|
host_summaries = host_svc.list(Host.FilterSpec(names=names))
|
|
if len(host_summaries) > 0:
|
|
host = host_summaries[0].host
|
|
print("Detected Host '{}' as {}".format(host_name, host))
|
|
context.testbed.entities['HOST_IDS'][host_name] = host
|
|
return True
|
|
else:
|
|
print("Host '{}' missing".format(host_name))
|
|
return False
|
|
|
|
|
|
def detect_hosts(context):
|
|
"""Find host used to run vcenter samples"""
|
|
context.testbed.entities['HOST_IDS'] = {}
|
|
host1_name = context.testbed.config['ESX_HOST1']
|
|
host2_name = context.testbed.config['ESX_HOST2']
|
|
return (detect_host(context, host1_name) and
|
|
detect_host(context, host2_name))
|
|
|
|
|
|
def cleanup_hosts(context):
|
|
"""Delete hosts after sample run"""
|
|
host_svc = Host(context.stub_config)
|
|
|
|
host1_name = context.testbed.config['ESX_HOST1']
|
|
host2_name = context.testbed.config['ESX_HOST2']
|
|
names = set([host1_name, host2_name])
|
|
|
|
host_summaries = host_svc.list(Host.FilterSpec(names=names))
|
|
print('Found {} Hosts matching names {}'.
|
|
format(len(host_summaries), ', '.
|
|
join(["'{}'".format(n) for n in names])))
|
|
|
|
for host_summary in host_summaries:
|
|
host = host_summary.host
|
|
print("Deleting Host '{}' ({})".format(host_summary.name, host))
|
|
host_svc.delete(host)
|
|
|
|
|
|
def create_host_vapi(context, host_name, datacenter_name):
|
|
"""
|
|
Adds a single Host to the vCenter inventory under the named Datacenter
|
|
using vAPI.
|
|
"""
|
|
host_svc = Host(context.stub_config)
|
|
|
|
user = context.testbed.config['ESX_USER']
|
|
pwd = context.testbed.config['ESX_PASS']
|
|
|
|
# Get the host folder for the Datacenter1 using the folder query
|
|
datacenter = context.testbed.entities['DATACENTER_IDS'][datacenter_name]
|
|
|
|
folder_svc = Folder(context.stub_config)
|
|
folder_summaries = folder_svc.list(
|
|
Folder.FilterSpec(type=Folder.Type.HOST, datacenters=set([datacenter])))
|
|
folder = folder_summaries[0].folder
|
|
|
|
create_spec = Host.CreateSpec(
|
|
hostname=host_name,
|
|
user_name=user,
|
|
password=pwd,
|
|
folder=folder,
|
|
thumbprint_verification=Host.CreateSpec.ThumbprintVerification.NONE)
|
|
host = host_svc.create(create_spec)
|
|
print("Created Host '{}' ({})".format(host, host_name))
|
|
|
|
return host
|
|
|
|
|
|
def create_host_vim(context, host_name, datacenter_name):
|
|
"""
|
|
Adds a single Host to the vCenter inventory under the named Datacenter
|
|
using the VIM API.
|
|
"""
|
|
user = context.testbed.config['ESX_USER']
|
|
pwd = context.testbed.config['ESX_PASS']
|
|
|
|
# Get the host folder for the Datacenter1 using the folder query
|
|
datacenter = context.testbed.entities['DATACENTER_IDS'][datacenter_name]
|
|
|
|
for entity in context.service_instance.content.rootFolder.childEntity:
|
|
if isinstance(entity, vim.Datacenter) and\
|
|
entity.name == datacenter_name:
|
|
datacenter_mo = entity
|
|
|
|
folder_mo = datacenter_mo.hostFolder
|
|
connect_spec = vim.host.ConnectSpec(hostName=host_name,
|
|
userName=user,
|
|
password=pwd,
|
|
force=False)
|
|
print("Creating Host ({})".format(host_name))
|
|
task = folder_mo.AddStandaloneHost(connect_spec,
|
|
vim.ComputeResource.ConfigSpec(),
|
|
True)
|
|
pyVim.task.WaitForTask(task)
|
|
|
|
# Get host from task result
|
|
host_mo = task.info.result.host[0]
|
|
print("Created Host '{}' ({})".format(host_mo._moId, host_name))
|
|
|
|
return host_mo._moId
|
|
|
|
|
|
def move_host_into_cluster_vim(context, host_name, cluster_name):
|
|
"""Use vim api to move host to another cluster"""
|
|
TIMEOUT = 30 # sec
|
|
|
|
host = context.testbed.entities['HOST_IDS'][host_name]
|
|
host_mo = vim.HostSystem(host, context.soap_stub)
|
|
|
|
# Move the host into the cluster
|
|
if not host_mo.runtime.inMaintenanceMode:
|
|
task = host_mo.EnterMaintenanceMode(TIMEOUT)
|
|
pyVim.task.WaitForTask(task)
|
|
print("Host '{}' ({}) in maintenance mode".format(host, host_name))
|
|
|
|
cluster = context.testbed.entities['CLUSTER_IDS'][cluster_name]
|
|
cluster_mo = vim.ClusterComputeResource(cluster, context.soap_stub)
|
|
|
|
task = cluster_mo.MoveInto([host_mo])
|
|
pyVim.task.WaitForTask(task)
|
|
print("Host '{}' ({}) moved into Cluster {} ({})".
|
|
format(host, host_name, cluster, cluster_name))
|
|
|
|
task = host_mo.ExitMaintenanceMode(TIMEOUT)
|
|
pyVim.task.WaitForTask(task)
|
|
print("Host '{}' ({}) out of maintenance mode".format(host, host_name))
|
|
|
|
|
|
def setup_hosts_vapi(context):
|
|
"""Use vsphere automation API to setup host for sample run"""
|
|
# Create Host1 as a standalone host in Datacenter1
|
|
host1_name = context.testbed.config['ESX_HOST1']
|
|
datacenter1_name = context.testbed.config['DATACENTER1_NAME']
|
|
host1 = create_host_vapi(context, host1_name, datacenter1_name)
|
|
|
|
# Create Host2 in a Cluster2
|
|
host2_name = context.testbed.config['ESX_HOST2']
|
|
datacenter2_name = context.testbed.config['DATACENTER2_NAME']
|
|
host2 = create_host_vapi(context, host2_name, datacenter2_name)
|
|
|
|
context.testbed.entities['HOST_IDS'] = {
|
|
host1_name: host1,
|
|
host2_name: host2
|
|
}
|
|
|
|
# Move Host2 into Cluster2
|
|
cluster_name = context.testbed.config['CLUSTER1_NAME']
|
|
move_host_into_cluster_vim(context, host2_name, cluster_name)
|
|
|
|
|
|
def setup_hosts_vim(context):
|
|
"""Use vim API to setup host for sample run"""
|
|
# Create Host1 as a standalone host in Datacenter1
|
|
host1_name = context.testbed.config['ESX_HOST1']
|
|
datacenter1_name = context.testbed.config['DATACENTER1_NAME']
|
|
host1 = create_host_vim(context, host1_name, datacenter1_name)
|
|
|
|
# Create Host2 in a Cluster2
|
|
host2_name = context.testbed.config['ESX_HOST2']
|
|
datacenter2_name = context.testbed.config['DATACENTER2_NAME']
|
|
host2 = create_host_vim(context, host2_name, datacenter2_name)
|
|
|
|
context.testbed.entities['HOST_IDS'] = {
|
|
host1_name: host1,
|
|
host2_name: host2
|
|
}
|
|
|
|
# Move Host2 into Cluster2
|
|
cluster_name = context.testbed.config['CLUSTER1_NAME']
|
|
move_host_into_cluster_vim(context, host2_name, cluster_name)
|
|
|
|
|
|
def setup_hosts(context):
|
|
setup_hosts_vapi(context)
|
|
|
|
|
|
def setup(context):
|
|
setup_hosts(context)
|
|
|
|
|
|
def cleanup(context):
|
|
cleanup_hosts(context)
|
|
|
|
|
|
def validate(context):
|
|
return detect_hosts(context)
|