Modify flask_control for vmware_host test suites (#26175)

Fix adds additional helper methods to get information about
host system from vcsim

Signed-off-by: Abhijeet Kasurde <akasurde@redhat.com>
This commit is contained in:
Abhijeet Kasurde 2017-06-30 04:41:28 +05:30 committed by jctanner
parent a396c18a61
commit bc4ff8aed9

View file

@ -190,30 +190,75 @@ def get_govc_vm_info():
return jsonify(vm_output) return jsonify(vm_output)
@app.route('/govc_host_info')
def get_govc_host_info():
""" Run govc host.info """
host_name = request.args.get("host_name") or None
host_output = {}
if host_name:
all_hosts = [host_name]
else:
all_hosts = _get_all_objs(ofilter='H')
for host_system in all_hosts:
host_info = _get_host_info(host_name=host_system)
name = host_info.get('Name', host_system)
host_output[name] = host_info
return jsonify(host_output)
def _get_host_info(host_name=None):
"""
Get all information of host from vcsim
:param vm_name: Name of host
:return: Dictionary containing information about VM,
where KEY represent attributes and VALUE represent attribute's value
"""
cmd = '%s host.info -host=%s 2>&1' % (GOVCPATH, host_name)
host_info = {}
if host_name is None:
return host_info
host_info = parse_govc_info(cmd)
return host_info
def _get_vm_info(vm_name=None): def _get_vm_info(vm_name=None):
""" """
Get all information of VM from vcsim Get all information of VM from vcsim
:param vm_name: Name of VM :param vm_name: Name of VM
:return: Dictionary containing inforamtion about VM, :return: Dictionary containing information about VM,
where KEY represent attributes and VALUE represent attribute's value where KEY represent attributes and VALUE represent attribute's value
""" """
cmd = '%s vm.info %s 2>&1' % (GOVCPATH, vm_name) cmd = '%s vm.info %s 2>&1' % (GOVCPATH, vm_name)
so, se = run_cmd(cmd)
stdout_lines = so.splitlines()
vm_info = {} vm_info = {}
if vm_name is None: if vm_name is None:
return vm_info return vm_info
vm_info = parse_govc_info(cmd)
for line in stdout_lines:
if ":" in line:
key, value = line.split(":")
key = key.lstrip()
vm_info[key] = value.strip()
return vm_info return vm_info
def parse_govc_info(cmd):
"""
Helper function to parse output of govc info commands
:param cmd: command variable to run and parse output for
:return: Dictionary containing information about object
"""
so, se = run_cmd(cmd)
stdout_lines = so.splitlines()
info = {}
for line in stdout_lines:
if ":" in line:
key, value = line.split(":", 1)
key = key.lstrip()
info[key] = value.strip()
return info
def _get_all_objs(ofilter=None): def _get_all_objs(ofilter=None):
""" """
Get all VM Objects from vcsim Get all VM Objects from vcsim