def _destroy_host(self, hostname):
req = VI.Destroy_TaskRequestMsg()
mor = (key for key, value in self.api.get_hosts().items() if value == hostname).next()
sys = VIMor(mor, 'HostSystem')
_this = req.new__this(sys)
_this.set_attribute_type(sys.get_attribute_type())
req.set_element__this(_this)
task_mor = self.api._proxy.Destroy_Task(req)._returnval
t = VITask(task_mor, self.api)
wait_for(lambda: 'success' in t.get_state())
def test_property_types(self):
hosts = self.server.get_hosts()
for hmor, hname in hosts.items():
p = VIProperty(self.server, hmor)
assert p.name == hname
#string
assert isinstance(p.name, str)
#property mor
assert VIMor.is_mor(p._obj)
#nested properties
assert isinstance(p.configManager, VIProperty)
#bool
assert isinstance(p.runtime.inMaintenanceMode, bool)
#list
assert isinstance(p.vm, list)
#mor without traversing
assert VIMor.is_mor(p.vm[0]._obj)
#traversing
assert isinstance(p.vm[0].name, str)
#enum
assert p.runtime.powerState in [
'poweredOff', 'poweredOn', 'standBy', 'unknown'
]
#int
assert isinstance(p.summary.config.port, int)
#long
assert isinstance(p.hardware.memorySize, int)
#short as int
assert isinstance(p.hardware.cpuInfo.numCpuCores, int)
#date as tuple
assert isinstance(p.runtime.bootTime, tuple)
#check property cache
assert "runtime" in list(p.__dict__.keys())
assert "memorySize" in list(p.hardware.__dict__.keys())
assert "numCpuCores" in list(p.hardware.cpuInfo.__dict__.keys())
assert "name" in list(p.vm[0].__dict__.keys())
#check cache flush
p._flush_cache()
assert "runtime" not in list(p.__dict__.keys())
assert "memorySize" not in list(p.hardware.__dict__.keys())
assert "numCpuCores" not in list(
p.hardware.cpuInfo.__dict__.keys())
assert "name" not in list(p.vm[0].__dict__.keys())
#check unexistent property
try:
p.hoochiemama
except AttributeError:
pass
except:
raise AssertionError("Attribute Error expected")
else:
raise AssertionError("Attribute Error expected")
def test_property_types(self):
hosts = self.server.get_hosts()
for hmor, hname in hosts.iteritems():
p = VIProperty(self.server, hmor)
assert p.name == hname
#string
assert isinstance(p.name, basestring)
#property mor
assert VIMor.is_mor(p._obj)
#nested properties
assert isinstance(p.configManager, VIProperty)
#bool
assert isinstance(p.runtime.inMaintenanceMode, bool)
#list
assert isinstance(p.vm, list)
#mor without traversing
assert VIMor.is_mor(p.vm[0]._obj)
#traversing
assert isinstance(p.vm[0].name, basestring)
#enum
assert p.runtime.powerState in ['poweredOff', 'poweredOn',
'standBy', 'unknown']
#int
assert isinstance(p.summary.config.port, int)
#long
assert isinstance(p.hardware.memorySize, long)
#short as int
assert isinstance(p.hardware.cpuInfo.numCpuCores, int)
#date as tuple
assert isinstance(p.runtime.bootTime, tuple)
#check property cache
assert "runtime" in p.__dict__.keys()
assert "memorySize" in p.hardware.__dict__.keys()
assert "numCpuCores" in p.hardware.cpuInfo.__dict__.keys()
assert "name" in p.vm[0].__dict__.keys()
#check cache flush
p._flush_cache()
assert "runtime" not in p.__dict__.keys()
assert "memorySize" not in p.hardware.__dict__.keys()
assert "numCpuCores" not in p.hardware.cpuInfo.__dict__.keys()
assert "name" not in p.vm[0].__dict__.keys()
#check unexistent property
try:
p.hoochiemama
except AttributeError:
pass
except:
raise AssertionError("Attribute Error expected")
else:
raise AssertionError("Attribute Error expected")
def __init__(self, server, entity=None, recursion=None, types=None, chain_id=None):
"""Creates a Event History Collector that gathers Event objects.
based on the provides filters.
* server: the connected VIServer instance
* entity: Entity MOR, if provided filters events related to this entity
* recursion: If 'entity' is provided then recursion is mandatory.
specification of related managed entities in the inventory hierarchy
should be either: 'all', 'children', or 'self'
* types: if provided, limits the set of collected events by their
types.
* chain_id: if provided, retrieves events by chain ID
"""
super(VIEventHistoryCollector, self).__init__(server)
if entity and not VIMor.is_mor(entity):
raise VIException("Entity should be a MOR object",
FaultTypes.PARAMETER_ERROR)
if entity and not recursion in [Recursion.ALL, Recursion.CHILDREN,
Recursion.SELF]:
raise VIException("Recursion should be either: "
"'all', 'children', or 'self'",
FaultTypes.PARAMETER_ERROR)
try:
event_manager = server._do_service_content.EventManager
request = VI.CreateCollectorForEventsRequestMsg()
_this = request.new__this(event_manager)
_this.set_attribute_type(event_manager.get_attribute_type())
request.set_element__this(_this)
_filter = request.new_filter()
if types and not isinstance(types, list):
types = [types]
if types:
_filter.set_element_eventTypeId(types)
if chain_id is not None:
_filter.set_element_eventChainId(chain_id)
if entity:
entity_filter = _filter.new_entity()
mor_entity = entity_filter.new_entity(entity)
mor_entity.set_attribute_type(entity.get_attribute_type())
entity_filter.set_element_entity(mor_entity)
entity_filter.set_element_recursion(recursion)
_filter.set_element_entity(entity_filter)
request.set_element_filter(_filter)
resp = server._proxy.CreateCollectorForEvents(request)._returnval
except (VI.ZSI.FaultException) as e:
raise VIApiException(e)
self._mor = resp
self._props = VIProperty(self._server, self._mor)
def test_if_all_hosts_arent_in_maintenance(self):
"""Are all the hosts out of the maintenance?"""
hosts = self.server.get_hosts()
for host in hosts:
host_mor = VIMor(host, 'HostSystem')
host_props = VIProperty(self.server, host_mor)
self.assertFalse(host_props.runtime.inMaintenanceMode,
u"Host Not Activated:{}".format(
host))
def __init__(self, server, entity=None, recursion=None, states=None):
"""Creates a Task History Collector that gathers Task info objects.
based on the provides filters.
* server: the connected VIServer instance
* entity: Entity MOR, if provided filters tasks related to this entity
* recursion: If 'entity' is provided then recursion is mandatory.
specification of related managed entities in the inventory hierarchy
should be either: 'all', 'children', or 'self'
* states: if provided, limits the set of collected tasks by their
states. Should be a list or one of 'queued', 'running', 'error', or
'success'
"""
self._server = server
self._mor = None
if entity and not VIMor.is_mor(entity):
raise VIException("Entity should be a MOR object",
FaultTypes.PARAMETER_ERROR)
if entity and not recursion in [
Recursion.ALL, Recursion.CHILDREN, Recursion.SELF
]:
raise VIException(
"Recursion should be either: "
"'all', 'children', or 'self'", FaultTypes.PARAMETER_ERROR)
try:
task_manager = server._do_service_content.TaskManager
request = VI.CreateCollectorForTasksRequestMsg()
_this = request.new__this(task_manager)
_this.set_attribute_type(task_manager.get_attribute_type())
request.set_element__this(_this)
_filter = request.new_filter()
if states and not isinstance(states, list):
states = [states]
if states:
_filter.set_element_state(states)
if entity:
entity_filter = _filter.new_entity()
mor_entity = entity_filter.new_entity(entity)
mor_entity.set_attribute_type(entity.get_attribute_type())
entity_filter.set_element_entity(mor_entity)
entity_filter.set_element_recursion(recursion)
_filter.set_element_entity(entity_filter)
request.set_element_filter(_filter)
resp = server._proxy.CreateCollectorForTasks(request)._returnval
except (VI.ZSI.FaultException) as e:
raise VIApiException(e)
self._mor = resp
self._props = VIProperty(self._server, self._mor)
def test_if_all_hosts_are_in_maintenance(self):
"""Todos os hosts estão em modo de manutenção?"""
server = vcenter_connect()
hosts = server.get_hosts()
for host in hosts:
host_mor = VIMor(host, 'HostSystem')
host_props = VIProperty(server, host_mor)
self.assertTrue(host_props.runtime.inMaintenanceMode,
u"Nem todos os hosts estão desativados")
def __init__(self, server, entity=None, recursion=None, states=None):
"""Creates a Task History Collector that gathers Task info objects.
based on the provides filters.
* server: the connected VIServer instance
* entity: Entity MOR, if provided filters tasks related to this entity
* recursion: If 'entity' is provided then recursion is mandatory.
specification of related managed entities in the inventory hierarchy
should be either: 'all', 'children', or 'self'
* states: if provided, limits the set of collected tasks by their
states. Should be a list or one of 'queued', 'running', 'error', or
'success'
"""
self._server = server
self._mor = None
if entity and not VIMor.is_mor(entity):
raise VIException("Entity should be a MOR object",
FaultTypes.PARAMETER_ERROR)
if entity and not recursion in [Recursion.ALL, Recursion.CHILDREN,
Recursion.SELF]:
raise VIException("Recursion should be either: "
"'all', 'children', or 'self'",
FaultTypes.PARAMETER_ERROR)
try:
task_manager = server._do_service_content.TaskManager
request = VI.CreateCollectorForTasksRequestMsg()
_this = request.new__this(task_manager)
_this.set_attribute_type(task_manager.get_attribute_type())
request.set_element__this(_this)
_filter = request.new_filter()
if states and not isinstance(states, list):
states = [states]
if states:
_filter.set_element_state(states)
if entity:
entity_filter = _filter.new_entity()
mor_entity = entity_filter.new_entity(entity)
mor_entity.set_attribute_type(entity.get_attribute_type())
entity_filter.set_element_entity(mor_entity)
entity_filter.set_element_recursion(recursion)
_filter.set_element_entity(entity_filter)
request.set_element_filter(_filter)
resp = server._proxy.CreateCollectorForTasks(request)._returnval
except (VI.ZSI.FaultException), e:
raise VIApiException(e)
def deploy_vm(args, server):
u"""Deploy VM(import)
@param args Commandline argument
@param server Instance of VIServer
"""
# Import
import os
import shutil
from pysphere import VIMor
import lib
download_temp = './temp/'
extract_temp = None
try:
# Download
if args.file_url:
if args.file_url.find('.ova') > 0:
args.filepath = lib.download_file(args.file_url, download_temp)
else:
raise Exception('File format not supported.')
filepath = args.filepath
if os.path.exists(filepath):
raise Exception("File not found.")
if filepath.find('.ova') > 0:
# If file format is ova
file_name = filepath.split('/')[-1]
extract_temp = r"./%s/" % file_name.replace('.ova', '')
filepath = lib.ova_extract(filepath, extract_temp)
elif filepath.find('.ovf') < 0:
# If file format isn't ova and ovf
raise Exception("File format not supported.")
# ovf_filepath path
ovf_filepath = filepath
lib.check_ovf(ovf_filepath)
# Remove mf File
MF_FILE = ovf_filepath.replace('.ovf', '.mf')
if os.path.exists(MF_FILE):
os.remove(MF_FILE)
# Host
if not args.datacenter:
host_name = server.get_hosts().values()[0]
else:
host_name = VIMor(args.datacenter,
MORTypes.Datacenter)
# Datastore
if not args.datastore:
datastore_name = server.get_datastores().values()[0]
else:
datastore_name = VIMor(args.datastore,
MORTypes.Datastore)
# Resource pool
if not args.resource_pool:
resource_pools = server.get_resource_pools()
resource_pool_name = [k for k, v in resource_pools.items()
if v == resource_pools.values()[0]][0]
else:
resource_pool_name = VIMor(args.resource_pool,
MORTypes.ResourcePool)
# Network
if not args.network:
network_name = 'VM Network'
else:
network_name = VIMor(args.network, MORTypes.Network)
ovf = lib.get_descriptor(ovf_filepath)
descriptor_info = lib.parse_descriptor(ovf, server)
support_info = lib.validate_host(host_name, ovf, server)
vm_names = args.vm_names
for vm_name in vm_names:
if lib.get_vm_index(server, vm_name) != -1:
print 'Already Exists VM.'
else:
# New VM name
vapp_name = vm_name
# Create spec
import_spec = lib.create_import_spec(resource_pool_name,
datastore_name,
ovf, vapp_name,
host=host_name,
network=network_name,
server=server)
if hasattr(import_spec, "Warning"):
print "Warning", import_spec.Warning[0].LocalizedMessage
http_nfc_lease = lib.import_vapp(resource_pool_name,
import_spec,
host=host_name,
server=server)
# Http request
lib.lease(http_nfc_lease, ovf_filepath, server)
print '%s Done.' % vm_name
finally:
# Remove
if os.path.exists(download_temp):
print 'Remove download temporary files...'
shutil.rmtree(download_temp)
if os.path.exists(extract_temp):
shutil.rmtree(extract_temp)
def clone(self,
name,
sync_run=True,
folder=None,
resourcepool=None,
datastore=None,
host=None,
power_on=True,
template=False,
snapshot=None,
linked=False,
customize=None,
data=None):
"""Clones this Virtual Machine
@name: name of the new virtual machine
@sync_run: if True (default) waits for the task to finish, and returns
a VIVirtualMachine instance with the new VM (raises an exception if
the task didn't succeed). If sync_run is set to False the task is
started and a VITask instance is returned
@folder: name of the folder that will contain the new VM, if not set
the vm will be added to the folder the original VM belongs to
@resourcepool: MOR of the resourcepool to be used for the new vm.
If not set, it uses the same resourcepool than the original vm.
@datastore: MOR of the datastore where the virtual machine
should be located. If not specified, the current datastore is used.
@host: MOR of the host where the virtual machine should be registered.
IF not specified:
* if resourcepool is not specified, current host is used.
* if resourcepool is specified, and the target pool represents a
stand-alone host, the host is used.
* if resourcepool is specified, and the target pool represents a
DRS-enabled cluster, a host selected by DRS is used.
* if resource pool is specified and the target pool represents a
cluster without DRS enabled, an InvalidArgument exception be
thrown.
@power_on: If the new VM will be powered on after being created. If
template is set to True, this parameter is ignored.
@template: Specifies whether or not the new virtual machine should be
marked as a template.
@snapshot: Snaphot MOR, or VISnaphost object, or snapshot name (if a
name is given, then the first matching occurrence will be used).
Is the snapshot reference from which to base the clone. If this
parameter is set, the clone is based off of the snapshot point. This
means that the newly created virtual machine will have the same
configuration as the virtual machine at the time the snapshot was
taken. If this parameter is not set then the clone is based off of
the virtual machine's current configuration.
@linked: If True (requires @snapshot to be set) creates a new child disk
backing on the destination datastore. None of the virtual disk's
existing files should be moved from their current locations.
Note that in the case of a clone operation, this means that the
original virtual machine's disks are now all being shared. This is
only safe if the clone was taken from a snapshot point, because
snapshot points are always read-only. Thus for a clone this option
is only valid when cloning from a snapshot
"""
try:
#get the folder to create the VM
folders = self._server._retrieve_properties_traversal(
property_names=['name', 'childEntity'],
obj_type=MORTypes.Folder)
folder_mor = None
for f in folders:
fname = ""
children = []
for prop in f.PropSet:
if prop.Name == "name":
fname = prop.Val
elif prop.Name == "childEntity":
children = prop.Val.ManagedObjectReference
if folder == fname or (not folder and self._mor in children):
folder_mor = f.Obj
break
if not folder_mor and folder:
raise VIException("Couldn't find folder %s" % folder,
FaultTypes.OBJECT_NOT_FOUND)
elif not folder_mor:
raise VIException("Error locating current VM folder",
FaultTypes.OBJECT_NOT_FOUND)
request = VI.CloneVM_TaskRequestMsg()
_this = request.new__this(self._mor)
_this.set_attribute_type(self._mor.get_attribute_type())
request.set_element__this(_this)
request.set_element_folder(folder_mor)
request.set_element_name(name)
spec = request.new_spec()
if template:
spec.set_element_powerOn(False)
else:
spec.set_element_powerOn(power_on)
location = spec.new_location()
if resourcepool:
if not VIMor.is_mor(resourcepool):
resourcepool = VIMor(resourcepool, MORTypes.ResourcePool)
pool = location.new_pool(resourcepool)
pool.set_attribute_type(resourcepool.get_attribute_type())
location.set_element_pool(pool)
if datastore:
if not VIMor.is_mor(datastore):
datastore = VIMor(datastore, MORTypes.Datastore)
ds = location.new_datastore(datastore)
ds.set_attribute_type(datastore.get_attribute_type())
location.set_element_datastore(ds)
if host:
if not VIMor.is_mor(host):
host = VIMor(host, MORTypes.HostSystem)
hs = location.new_host(host)
hs.set_attribute_type(host.get_attribute_type())
location.set_element_host(hs)
if snapshot:
sn_mor = None
if VIMor.is_mor(snapshot):
sn_mor = snapshot
elif isinstance(snapshot, VISnapshot):
sn_mor = snapshot._mor
elif isinstance(snapshot, basestring):
for sn in self.get_snapshots():
if sn.get_name() == snapshot:
sn_mor = sn._mor
break
if not sn_mor:
raise VIException(
"Could not find snapshot '%s'" % snapshot,
FaultTypes.OBJECT_NOT_FOUND)
snapshot = spec.new_snapshot(sn_mor)
snapshot.set_attribute_type(sn_mor.get_attribute_type())
spec.set_element_snapshot(snapshot)
if linked and snapshot:
location.set_element_diskMoveType("createNewChildDiskBacking")
if not template and customize:
if data is None:
raise VIApiException(
"Cannot use Customization without data")
customization = spec.new_customization()
spec.set_element_customization(customization)
globalIPSettings = customization.new_globalIPSettings()
customization.set_element_globalIPSettings(globalIPSettings)
# nicSettingMap
nicSetting = customization.new_nicSettingMap()
adapter = nicSetting.new_adapter()
nicSetting.set_element_adapter(adapter)
ipAddress = data.get('ip')
netmask = data.get('netmask')
gateway = data.get('gateway')
fixedip = VI.ns0.CustomizationFixedIp_Def(
"ipAddress").pyclass()
fixedip.set_element_ipAddress(ipAddress)
#dhcp = VI.ns0.CustomizationDhcpIpGenerator_Def("ip").pyclass()
adapter.set_element_ip(fixedip)
adapter.set_element_subnetMask(netmask)
#help(adapter.set_element_gateway([gateway,]))
adapter.set_element_gateway([
gateway,
])
nicSetting.set_element_adapter(adapter)
customization.set_element_nicSettingMap([
nicSetting,
])
if customize == "SYSPREP":
# here starts windows
identity = VI.ns0.CustomizationSysprep_Def(
"identity").pyclass()
customization.set_element_identity(identity)
guiUnattended = identity.new_guiUnattended()
guiUnattended.set_element_autoLogon(True)
guiUnattended.set_element_autoLogonCount(1)
passw = guiUnattended.new_password()
guiUnattended.set_element_password(passw)
passw.set_element_value(data["adminpw"])
passw.set_element_plainText(True)
# http://msdn.microsoft.com/en-us/library/ms912391(v=winembedded.11).aspx
# 85 is GMT Standard Time
timeZone = data.get("timezone", 85)
guiUnattended.set_element_timeZone(timeZone)
identity.set_element_guiUnattended(guiUnattended)
userData = identity.new_userData()
userData.set_element_fullName(
data.get("fullName", "PyShere"))
userData.set_element_orgName(
data.get("orgName", "PySphere"))
userData.set_element_productId("")
computerName = VI.ns0.CustomizationFixedName_Def(
"computerName").pyclass()
computerName.set_element_name(name.replace("_", ""))
userData.set_element_computerName(computerName)
identity.set_element_userData(userData)
identification = identity.new_identification()
if data.get("joinDomain", False):
# join the domain
identification.set_element_domainAdmin(
data["domainAdmin"])
domainAdminPassword = identification.new_domainAdminPassword(
)
domainAdminPassword.set_element_plainText(True)
domainAdminPassword.set_element_value(
data["domainAdminPassword"])
identification.set_element_domainAdminPassword(
domainAdminPassword)
identification.set_element_joinDomain(
data["joinDomain"])
identity.set_element_identification(identification)
elif customize == "SYSPREPTEXT":
identity = VI.ns0.CustomizationSysprepText_Def(
"identity").pyclass()
customization.set_element_identity(identity)
identity.set_element_value(data["value"])
elif customize == "LINUX":
identity = VI.ns0.CustomizationLinuxPrep_Def(
"identity").pyclass()
customization.set_element_identity(identity)
identity.set_element_domain(data["domain"])
hostName = VI.ns0.CustomizationFixedName_Def(
"hostName").pyclass()
hostName.set_element_name(name.replace("_", ""))
identity.set_element_hostName(hostName)
spec.set_element_location(location)
spec.set_element_template(template)
request.set_element_spec(spec)
task = self._server._proxy.CloneVM_Task(request)._returnval
vi_task = VITask(task, self._server)
if sync_run:
status = vi_task.wait_for_state(
[vi_task.STATE_SUCCESS, vi_task.STATE_ERROR])
if status == vi_task.STATE_ERROR:
raise VIException(vi_task.get_error_message(),
FaultTypes.TASK_ERROR)
return mianbao(self._server, vi_task.get_result()._obj)
return vi_task
except (VI.ZSI.FaultException), e:
raise VIApiException(e)
def host_mor(mor):
if not VIMor.is_mor(mor):
return VIMor(mor, "HostSystem")
return mor