1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#!/usr/bin/env python
#
# jose.lausuch@ericsson.com
# valentin.boucher@orange.com
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
# pylint: disable=missing-docstring
from __future__ import print_function
import logging
import subprocess
import sys
import yaml
from functest.utils import env
LOGGER = logging.getLogger(__name__)
def execute_command_raise(cmd, info=False, error_msg="",
verbose=True, output_file=None):
ret = execute_command(cmd, info, error_msg, verbose, output_file)
if ret != 0:
raise Exception(error_msg)
def execute_command(cmd, info=False, error_msg="",
verbose=True, output_file=None):
if not error_msg:
error_msg = ("The command '%s' failed." % cmd)
msg_exec = ("Executing command: '%s'" % cmd)
if verbose:
if info:
LOGGER.info(msg_exec)
else:
LOGGER.debug(msg_exec)
popen = subprocess.Popen(
cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
if output_file:
ofd = open(output_file, "w")
for line in iter(popen.stdout.readline, b''):
if output_file:
ofd.write(line)
else:
line = line.replace('\n', '')
print (line)
sys.stdout.flush()
if output_file:
ofd.close()
popen.stdout.close()
returncode = popen.wait()
if returncode != 0:
if verbose:
LOGGER.error(error_msg)
return returncode
def get_parameter_from_yaml(parameter, yfile):
"""
Returns the value of a given parameter in file.yaml
parameter must be given in string format with dots
Example: general.openstack.image_name
"""
with open(yfile) as yfd:
file_yaml = yaml.safe_load(yfd)
value = file_yaml
for element in parameter.split("."):
value = value.get(element)
if value is None:
raise ValueError("The parameter %s is not defined in"
" %s" % (parameter, yfile))
return value
def get_external_network(cloud):
"""
Returns the configured external network name or
the first retrieved external network name
"""
assert cloud
if env.get("EXTERNAL_NETWORK"):
network = cloud.get_network(
env.get("EXTERNAL_NETWORK"), {"router:external": True})
if network:
return network
networks = cloud.list_networks({"router:external": True})
if networks:
return networks[0]
return None
|