def instalar(self,name,template):
with open("Templates/%s.json"%template,"r") as f:
comandos = json.loads(f.read())
d = Docker()
s = SSH()
for c in comandos.get("comandos"):
print s.exec_command(d.execCommand(name,c))
def contratar(self,name,template):
d = Docker()
s = SSH()
s.exec_command(d.createContainer(name))
#metodo da propria classe
self.instalar(name,template)
endereco = s.exec_command(d.getContainerAddress(name))
endereco = json.loads(endereco)
print endereco[0].get("NetworkSettings")\
.get("Networks")\
.get("bridge")\
.get("IPAddress")
def execute_script(self, local_path, node, timeout=10, json_out=True):
"""Copy local_path script to node, execute it and return result.
Returns (rc, stdout, stderr tuple).
"""
ssh = SSH()
ssh.connect(node)
local_basename = os.path.basename(local_path)
remote_file_path = self.__TMP_DIR + local_basename
remote_file_out = remote_file_path + ".out"
ssh.scp(local_path, remote_file_path)
cmd = "sudo -S {vat} {json} < {input}".format(
vat=self.__VAT_BIN,
json="json" if json_out == True else "",
input=remote_file_path)
(ret_code, stdout, stderr) = ssh.exec_command(cmd, timeout)
self._ret_code = ret_code
self._stdout = stdout
self._stderr = stderr
logger.trace("Command '{0}' returned {1}'".format(cmd, self._ret_code))
logger.trace("stdout: '{0}'".format(self._stdout))
logger.trace("stderr: '{0}'".format(self._stderr))
#TODO: download vpp_api_test output file
self._delete_files(node, remote_file_path, remote_file_out)
def execute_script(self, local_path, node, timeout=10, json_out=True):
"""Copy local_path script to node, execute it and return result.
Returns (rc, stdout, stderr tuple).
"""
ssh = SSH()
ssh.connect(node)
local_basename = os.path.basename(local_path)
remote_file_path = self.__TMP_DIR + local_basename
remote_file_out = remote_file_path + ".out"
ssh.scp(local_path, remote_file_path)
cmd = "sudo -S {vat} {json} < {input}".format(vat=self.__VAT_BIN,
json="json" if json_out == True else "",
input=remote_file_path)
(ret_code, stdout, stderr) = ssh.exec_command(cmd, timeout)
self._ret_code = ret_code
self._stdout = stdout
self._stderr = stderr
logger.trace("Command '{0}' returned {1}'".format(cmd, self._ret_code))
logger.trace("stdout: '{0}'".format(self._stdout))
logger.trace("stderr: '{0}'".format(self._stderr))
#TODO: download vpp_api_test output file
self._delete_files(node, remote_file_path, remote_file_out)
def __extract_tarball_at_node(self, tarball, node):
logger.console('Extracting tarball to {0} on {1}'.format(
con.REMOTE_FW_DIR, node['host']))
ssh = SSH()
ssh.connect(node)
cmd = 'rm -rf {1}; mkdir {1} ; sudo -Sn tar -zxf {0} -C {1};'.format(
tarball, con.REMOTE_FW_DIR)
(ret_code, stdout, stderr) = ssh.exec_command(cmd, timeout=30)
if 0 != ret_code:
logger.error('Unpack error: {0}'.format(stderr))
raise Exception('Failed to unpack {0} at node {1}'.format(
tarball, node['host']))
def setup_dut(self, node):
ssh = SSH()
ssh.connect(node)
ssh.scp('resources/libraries/bash/dut_setup.sh', '/tmp/dut_setup.sh')
(ret_code, stdout, stderr) = \
ssh.exec_command('sudo -Sn bash /tmp/dut_setup.sh')
logger.trace(stdout)
if 0 != int(ret_code):
logger.error('DUT {0} setup script failed: "{1}"'.
format(node['host'], stdout + stderr))
raise Exception('DUT test setup script failed at node {}'.
format(node['host']))
def __extract_tarball_at_node(self, tarball, node):
logger.console('Extracting tarball to {0} on {1}'.format(
con.REMOTE_FW_DIR, node['host']))
ssh = SSH()
ssh.connect(node)
cmd = 'rm -rf {1}; mkdir {1} ; sudo -Sn tar -zxf {0} -C {1};'.format(
tarball, con.REMOTE_FW_DIR)
(ret_code, stdout, stderr) = ssh.exec_command(cmd, timeout=30)
if 0 != ret_code:
logger.error('Unpack error: {0}'.format(stderr))
raise Exception('Failed to unpack {0} at node {1}'.format(
tarball, node['host']))
def listar_servicos(self): d = Docker() s = SSH() print s.exec_command(d.listContainer())
def cancelar(self,name): d = Docker() s = SSH() print s.exec_command(d.removeContainer(name))
def _delete_files(self, node, *files):
ssh = SSH()
ssh.connect(node)
files = " ".join([str(x) for x in files])
ssh.exec_command("rm {0}".format(files))
def _delete_files(self, node, *files):
ssh = SSH()
ssh.connect(node)
files = " ".join([str(x) for x in files])
ssh.exec_command("rm {0}".format(files))
