mirror of
https://github.com/vmware/vsphere-automation-sdk-python.git
synced 2024-11-27 03:30:00 -05:00
114 lines
4.0 KiB
Python
114 lines
4.0 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.
|
|
"""
|
|
|
|
from com.vmware.vcenter_client import (Datacenter, Folder)
|
|
|
|
|
|
def folder_list_datacenter_folder(context):
|
|
folder_svc = Folder(context.stub_config)
|
|
return folder_svc.list(Folder.FilterSpec(type=Folder.Type.DATACENTER))
|
|
|
|
|
|
def detect_datacenter(context, datacenter_name):
|
|
"""Find the datacenter with the given name"""
|
|
datacenter_svc = Datacenter(context.stub_config)
|
|
|
|
names = set([datacenter_name])
|
|
datacenter_summaries = datacenter_svc.list(
|
|
Datacenter.FilterSpec(names=names))
|
|
if len(datacenter_summaries) > 0:
|
|
datacenter = datacenter_summaries[0].datacenter
|
|
print("Detected Datacenter '{}' as {}".
|
|
format(datacenter_name, datacenter))
|
|
context.testbed.entities['DATACENTER_IDS'][datacenter_name] = datacenter
|
|
return True
|
|
else:
|
|
print("Datacenter '{}' missing".format(datacenter_name))
|
|
return False
|
|
|
|
|
|
def detect_datacenters(context):
|
|
"""Find datacenters to run the vcenter samples"""
|
|
context.testbed.entities['DATACENTER_IDS'] = {}
|
|
|
|
# Look for the two datacenters
|
|
datacenter1_name = context.testbed.config['DATACENTER1_NAME']
|
|
datacenter2_name = context.testbed.config['DATACENTER2_NAME']
|
|
|
|
return (detect_datacenter(context, datacenter1_name) and
|
|
detect_datacenter(context, datacenter2_name))
|
|
|
|
|
|
def cleanup_datacenters(context):
|
|
"""Cleanup datacenters after sample run"""
|
|
datacenter_svc = Datacenter(context.stub_config)
|
|
|
|
# Look for the two datacenters
|
|
datacenter1_name = context.testbed.config['DATACENTER1_NAME']
|
|
datacenter2_name = context.testbed.config['DATACENTER2_NAME']
|
|
names = set([datacenter1_name, datacenter2_name])
|
|
|
|
datacenter_summaries = datacenter_svc.list(
|
|
Datacenter.FilterSpec(names=names))
|
|
print("Found {} Datacenters matching names {}".
|
|
format(len(datacenter_summaries),", ".
|
|
join(["'{}'".format(n) for n in names])))
|
|
|
|
for datacenter_summary in datacenter_summaries:
|
|
datacenter = datacenter_summary.datacenter
|
|
print("Deleting Datacenter '{}' ({})".
|
|
format(datacenter, datacenter_summary.name))
|
|
datacenter_svc.delete(datacenter, force=True)
|
|
|
|
|
|
def setup_datacenters(context):
|
|
"""Create datacenters for running vcenter samples"""
|
|
# Find a Folder in which to put the Datacenters
|
|
folder_summaries = folder_list_datacenter_folder(context)
|
|
folder = folder_summaries[0].folder
|
|
print("Creating datacenters in Folder '{}' ({})".
|
|
format(folder, folder_summaries[0].name))
|
|
|
|
datacenter_svc = Datacenter(context.stub_config)
|
|
|
|
# Create first datacenter
|
|
datacenter1_name = context.testbed.config['DATACENTER1_NAME']
|
|
datacenter1 = datacenter_svc.create(
|
|
Datacenter.CreateSpec(name=datacenter1_name, folder=folder)
|
|
)
|
|
print("Created Datacenter '{}' ({})".format(datacenter1, datacenter1_name))
|
|
|
|
# Create second datacenter
|
|
datacenter2_name = context.testbed.config['DATACENTER2_NAME']
|
|
datacenter2 = datacenter_svc.create(
|
|
Datacenter.CreateSpec(name=datacenter2_name, folder=folder)
|
|
)
|
|
print("Created Datacenter '{}' ({})".format(datacenter2, datacenter2_name))
|
|
|
|
# Save datacenter name to identifier mappings for later use
|
|
context.testbed.entities['DATACENTER_IDS'] = {
|
|
datacenter1_name: datacenter1,
|
|
datacenter2_name: datacenter2
|
|
}
|
|
|
|
|
|
def cleanup(context):
|
|
cleanup_datacenters(context)
|
|
|
|
|
|
def setup(context):
|
|
setup_datacenters(context)
|
|
|
|
|
|
def validate(context):
|
|
return detect_datacenters(context)
|