def folders(cls, session, datacenter):
"""
Method will prompt user to select a folder from a datacenter
Args:
session (obj): Auth session object
datacenter (str): Name of datacenter
Returns:
folder (str): Name of selected folder
"""
datacenters = Query.create_container(
session, session.content.rootFolder,
[vim.Datacenter], True
)
folders = Query.list_vm_folders(
datacenters.view, datacenter
)
folders.sort()
for num, opt in enumerate(folders, start=1):
print('%s: %s' % (num, opt))
while True:
val = int(raw_input('\nPlease select number: ').strip())
if int(val) <= len(folders):
# need to substract 1 since we start enumeration at 1.
val = int(val) - 1
selected_folder = folders[val]
break
else:
print('Invalid number.')
continue
cls.logger.info(selected_folder)
return selected_folder
def folders(cls, session, datacenter):
"""
Method will prompt user to select a folder from a datacenter
Args:
session (obj): Auth session object
datacenter (str): Name of datacenter
Returns:
folder (str): Name of selected folder
"""
datacenters = Query.create_container(session,
session.content.rootFolder,
[vim.Datacenter], True)
folders = Query.list_vm_folders(datacenters.view, datacenter)
folders.sort()
for num, opt in enumerate(folders, start=1):
print('%s: %s' % (num, opt))
while True:
val = int(raw_input('\nPlease select number: ').strip())
if int(val) <= len(folders):
# need to substract 1 since we start enumeration at 1.
val = int(val) - 1
selected_folder = folders[val]
break
else:
print('Invalid number.')
continue
cls.logger.info(selected_folder)
return selected_folder
def main(self):
"""
This is the main method, which parses all the argparse options and runs
the necessary code blocks if True.
"""
try:
call_count = 0
self.auth = Auth(self.opts.host)
self.auth.login(
self.opts.user, self.opts.passwd, self.opts.domain, self.opts.passwd_file
)
self.opts.passwd = None
self.logger.debug(self.opts)
virtual_machines_container = Query.create_container(
self.auth.session, self.auth.session.content.rootFolder,
[vim.VirtualMachine], True
)
self.vmcfg = VMConfigHelper(self.auth, self.opts, argparser.dotrc)
self.clustercfg = ClusterConfig(self.auth, self.opts, argparser.dotrc)
call_count = self.auth.session.content.sessionManager.currentSession.callCount
if not self.opts.datacenter:
self.opts.datacenter = Prompts.datacenters(self.auth.session)
if self.opts.cmd == 'create':
if self.opts.config:
for cfg in self.opts.config:
spec = self.vmcfg.dict_merge(
argparser.dotrc, yaml.load(cfg, Loader=yaml.FullLoader)
)
cfgcheck_update = CfgCheck.cfg_checker(spec, self.auth, self.opts)
spec['vmconfig'].update(
self.vmcfg.dict_merge(spec['vmconfig'], cfgcheck_update)
)
spec = self.vmcfg.pre_create_hooks(**spec)
spec = self.vmcfg.create_wrapper(**spec)
self.vmcfg.post_create_hooks(**spec)
filename = spec['vmconfig']['name'] + '.yaml'
server_cfg = {}
server_cfg['vmconfig'] = {}
server_cfg['vmconfig'].update(spec['vmconfig'])
if spec.get('mkbootiso', None):
server_cfg['mkbootiso'] = {}
server_cfg['mkbootiso'].update(spec['mkbootiso'])
print(
yaml.dump(server_cfg, default_flow_style=False),
file=open(os.path.join(os.environ['OLDPWD'], filename), 'w')
)
if self.opts.cmd == 'mount':
self.vmcfg.mount_wrapper(self.opts.datastore, self.opts.path, *self.opts.name)
if self.opts.cmd == 'power':
self.vmcfg.power_wrapper(self.opts.power, *self.opts.name)
if self.opts.cmd == 'umount':
self.vmcfg.umount_wrapper(*self.opts.name)
if self.opts.cmd == 'upload':
self.vmcfg.upload_wrapper(
self.opts.datastore, self.opts.dest,
self.opts.verify_ssl, *self.opts.iso
)
if self.opts.cmd == 'add':
hostname = Query.get_obj(virtual_machines_container.view, self.opts.name)
# nics
if self.opts.device == 'nic':
self.vmcfg.add_nic_recfg(hostname)
if self.opts.cmd == 'reconfig':
host = Query.get_obj(virtual_machines_container.view, self.opts.name)
if self.opts.cfgs:
self.logger.info(
'reconfig: %s cfgs: %s', host.name,
' '.join('%s=%s' % (k, v) for k, v in self.opts.cfgs.items())
)
self.vmcfg.reconfig(host, **self.opts.cfgs)
if self.opts.folder:
self.vmcfg.folder_recfg()
if self.opts.device == 'disk':
self.vmcfg.disk_recfg()
if self.opts.device == 'nic':
self.vmcfg.nic_recfg()
if self.opts.cmd == 'drs':
if not self.opts.cluster:
self.opts.cluster = Prompts.clusters(self.auth.session)
self.clustercfg.drs_rule()
if self.opts.cmd == 'query':
datacenters_container = Query.create_container(
self.auth.session, self.auth.session.content.rootFolder,
[vim.Datacenter], True
)
clusters_container = Query.create_container(
self.auth.session, self.auth.session.content.rootFolder,
[vim.ClusterComputeResource], True
)
if self.opts.anti_affinity_rules:
if self.opts.cluster:
anti_affinity_rules = Query.return_anti_affinity_rules(
clusters_container.view, self.opts.cluster
)
else:
cluster = Prompts.clusters(self.auth.session)
anti_affinity_rules = Query.return_anti_affinity_rules(
clusters_container.view, cluster
)
if not anti_affinity_rules:
print('No antiaffinity rules defined.')
else:
print('Antiaffinity rules:')
for key, val in sorted(anti_affinity_rules.items()):
print('{0}: {1}'.format(key, ' '.join(sorted(val))))
if self.opts.datastores:
if self.opts.cluster:
datastores = Query.return_datastores(
clusters_container.view, self.opts.cluster
)
else:
cluster = Prompts.clusters(self.auth.session)
datastores = Query.return_datastores(clusters_container.view, cluster)
for row in datastores:
print('{0:30}\t{1:10}\t{2:10}\t{3:6}\t{4:10}\t{5:6}'.format(*row))
if self.opts.folders:
if self.opts.datacenter:
folders = Query.list_vm_folders(
datacenters_container.view, self.opts.datacenter
)
folders.sort()
for folder in folders:
print(folder)
else:
datacenter = Prompts.datacenters(self.auth.session)
folders = Query.list_vm_folders(datacenters_container.view, datacenter)
folders.sort()
for folder in folders:
print(folder)
if self.opts.clusters:
clusters = Query.list_obj_attrs(clusters_container, 'name')
clusters.sort()
for cluster in clusters:
print(cluster)
if self.opts.networks:
if self.opts.cluster:
cluster = Query.get_obj(clusters_container.view, self.opts.cluster)
networks = Query.list_obj_attrs(cluster.network, 'name', view=False)
networks.sort()
for net in networks:
print(net)
else:
cluster_name = Prompts.clusters(self.auth.session)
cluster = Query.get_obj(clusters_container.view, cluster_name)
networks = Query.list_obj_attrs(cluster.network, 'name', view=False)
networks.sort()
for net in networks:
print(net)
if self.opts.vms:
vms = Query.list_vm_info(datacenters_container.view, self.opts.datacenter)
for key, value in vms.items():
print(key, value)
if self.opts.vmconfig:
for name in self.opts.vmconfig:
virtmachine = Query.get_obj(virtual_machines_container.view, name)
self.logger.debug(virtmachine.config)
if self.opts.createcfg:
print(
yaml.dump(
Query.vm_config(
virtual_machines_container.view, name, self.opts.createcfg
),
default_flow_style=False
)
)
else:
print(
yaml.dump(
Query.vm_config(virtual_machines_container.view, name),
default_flow_style=False
)
)
if self.opts.vm_by_datastore:
if self.opts.cluster and self.opts.datastore:
vms = Query.vm_by_datastore(
clusters_container.view, self.opts.cluster, self.opts.datastore
)
for vm_name in vms:
print(vm_name)
else:
if not self.opts.cluster:
cluster = Prompts.clusters(self.auth.session)
if not self.opts.datastore:
datastore = Prompts.datastores(self.auth.session, cluster)
print()
vms = Query.vm_by_datastore(clusters_container.view, cluster, datastore)
for vm_name in vms:
print(vm_name)
if self.opts.vm_guest_ids:
for guest_id in Query.list_guestids():
print(guest_id)
self.auth.logout()
self.logger.debug('Call count: {0}'.format(call_count))
except ValueError as err:
self.logger.error(err, exc_info=False)
self.auth.logout()
self.logger.debug('Call count: {0}'.format(call_count))
sys.exit(3)
except vim.fault.InvalidLogin as loginerr:
self.logger.error(loginerr.msg, exc_info=False)
sys.exit(2)
except ssl.CertificateError as err:
self.logger.error(err, exc_info=False)
sys.exit(2)
except KeyboardInterrupt as err:
self.logger.error(err, exc_info=False)
self.auth.logout()
self.logger.debug('Call count: {0}'.format(call_count))
sys.exit(1)
def main(self):
"""
This is the main method, which parses all the argparse options and runs
the necessary code blocks if True.
"""
try:
self.auth = Auth(self.opts.host)
self.auth.login(self.opts.user, self.opts.passwd, self.opts.domain,
self.opts.passwd_file)
self.opts.passwd = None
self.logger.debug(self.opts)
virtual_machines_container = Query.create_container(
self.auth.session, self.auth.session.content.rootFolder,
[vim.VirtualMachine], True)
self.vmcfg = VMConfigHelper(self.auth, self.opts, argparser.dotrc)
self.clustercfg = ClusterConfig(self.auth, self.opts,
argparser.dotrc)
call_count = self.auth.session.content.sessionManager.currentSession.callCount
if self.opts.cmd == 'create':
if self.opts.config:
for cfg in self.opts.config:
spec = self.vmcfg.dict_merge(argparser.dotrc,
yaml.load(cfg))
cfgcheck_update = CfgCheck.cfg_checker(
spec, self.auth, self.opts)
spec['vmconfig'].update(
self.vmcfg.dict_merge(spec['vmconfig'],
cfgcheck_update))
spec = self.vmcfg.pre_create_hooks(**spec)
spec = self.vmcfg.create_wrapper(**spec)
self.vmcfg.post_create_hooks(**spec)
filename = spec['vmconfig']['name'] + '.yaml'
server_cfg = {}
server_cfg['vmconfig'] = {}
server_cfg['vmconfig'].update(spec['vmconfig'])
if spec.get('mkbootiso', None):
server_cfg['mkbootiso'] = {}
server_cfg['mkbootiso'].update(spec['mkbootiso'])
print(yaml.dump(server_cfg, default_flow_style=False),
file=open(filename, 'w'))
if self.opts.cmd == 'mount':
self.vmcfg.mount_wrapper(self.opts.datastore, self.opts.path,
*self.opts.name)
if self.opts.cmd == 'power':
self.vmcfg.power_wrapper(self.opts.power, *self.opts.name)
if self.opts.cmd == 'umount':
self.vmcfg.umount_wrapper(*self.opts.name)
if self.opts.cmd == 'upload':
self.vmcfg.upload_wrapper(self.opts.datastore, self.opts.dest,
self.opts.verify_ssl, *self.opts.iso)
if self.opts.cmd == 'add':
hostname = Query.get_obj(virtual_machines_container.view,
self.opts.name)
# nics
if self.opts.device == 'nic':
self.vmcfg.add_nic_recfg(hostname)
if self.opts.cmd == 'reconfig':
host = Query.get_obj(virtual_machines_container.view,
self.opts.name)
if self.opts.cfgs:
self.logger.info(
'reconfig: %s cfgs: %s', host.name,
' '.join('%s=%s' % (k, v)
for k, v in self.opts.cfgs.iteritems()))
self.vmcfg.reconfig(host, **self.opts.cfgs)
if self.opts.folder:
self.vmcfg.folder_recfg()
if self.opts.device == 'disk':
self.vmcfg.disk_recfg()
if self.opts.device == 'nic':
self.vmcfg.nic_recfg()
if self.opts.cmd == 'drs':
if not self.opts.cluster:
self.opts.cluster = Prompts.clusters(self.auth.session)
self.clustercfg.drs_rule()
if self.opts.cmd == 'query':
datacenters_container = Query.create_container(
self.auth.session, self.auth.session.content.rootFolder,
[vim.Datacenter], True)
clusters_container = Query.create_container(
self.auth.session, self.auth.session.content.rootFolder,
[vim.ClusterComputeResource], True)
if self.opts.anti_affinity_rules:
if self.opts.cluster:
anti_affinity_rules = Query.return_anti_affinity_rules(
clusters_container.view, self.opts.cluster)
else:
cluster = Prompts.clusters(self.auth.session)
anti_affinity_rules = Query.return_anti_affinity_rules(
clusters_container.view, cluster)
if not anti_affinity_rules:
print('No antiaffinity rules defined.')
else:
print('Antiaffinity rules:')
for key, val in sorted(
anti_affinity_rules.iteritems()):
print('{0}: {1}'.format(key,
' '.join(sorted(val))))
if self.opts.datastores:
if self.opts.cluster:
datastores = Query.return_datastores(
clusters_container.view, self.opts.cluster)
else:
cluster = Prompts.clusters(self.auth.session)
datastores = Query.return_datastores(
clusters_container.view, cluster)
for row in datastores:
print('{0:30}\t{1:10}\t{2:10}\t{3:6}\t{4:10}\t{5:6}'.
format(*row))
if self.opts.folders:
if self.opts.datacenter:
folders = Query.list_vm_folders(
datacenters_container.view, self.opts.datacenter)
folders.sort()
for folder in folders:
print(folder)
else:
datacenter = Prompts.datacenters(self.auth.session)
folders = Query.list_vm_folders(
datacenters_container.view, datacenter)
folders.sort()
for folder in folders:
print(folder)
if self.opts.clusters:
clusters = Query.list_obj_attrs(clusters_container, 'name')
clusters.sort()
for cluster in clusters:
print(cluster)
if self.opts.networks:
if self.opts.cluster:
cluster = Query.get_obj(clusters_container.view,
self.opts.cluster)
networks = Query.list_obj_attrs(cluster.network,
'name',
view=False)
networks.sort()
for net in networks:
print(net)
else:
cluster_name = Prompts.clusters(self.auth.session)
cluster = Query.get_obj(clusters_container.view,
cluster_name)
networks = Query.list_obj_attrs(cluster.network,
'name',
view=False)
networks.sort()
for net in networks:
print(net)
if self.opts.vms:
vms = Query.list_vm_info(datacenters_container.view,
self.opts.datacenter)
for key, value in vms.iteritems():
print(key, value)
if self.opts.vmconfig:
for name in self.opts.vmconfig:
virtmachine = Query.get_obj(
virtual_machines_container.view, name)
self.logger.debug(virtmachine.config)
if self.opts.createcfg:
print(
yaml.dump(Query.vm_config(
virtual_machines_container.view, name,
self.opts.createcfg),
default_flow_style=False))
else:
print(
yaml.dump(Query.vm_config(
virtual_machines_container.view, name),
default_flow_style=False))
if self.opts.vm_by_datastore:
if self.opts.cluster and self.opts.datastore:
vms = Query.vm_by_datastore(clusters_container.view,
self.opts.cluster,
self.opts.datastore)
for vm_name in vms:
print(vm_name)
else:
if not self.opts.cluster:
cluster = Prompts.clusters(self.auth.session)
if not self.opts.datastore:
datastore = Prompts.datastores(
self.auth.session, cluster)
print()
vms = Query.vm_by_datastore(clusters_container.view,
cluster, datastore)
for vm_name in vms:
print(vm_name)
if self.opts.vm_guest_ids:
for guest_id in Query.list_guestids():
print(guest_id)
self.auth.logout()
self.logger.debug('Call count: {0}'.format(call_count))
except ValueError as err:
self.logger.error(err, exc_info=False)
self.auth.logout()
self.logger.debug('Call count: {0}'.format(call_count))
sys.exit(3)
except vim.fault.InvalidLogin as loginerr:
self.logger.error(loginerr.msg, exc_info=False)
sys.exit(2)
except KeyboardInterrupt as err:
self.logger.error(err, exc_info=False)
self.auth.logout()
self.logger.debug('Call count: {0}'.format(call_count))
sys.exit(1)