diff options
Diffstat (limited to 'resources/tools/scripts/topo_container_copy.py')
-rw-r--r-- | resources/tools/scripts/topo_container_copy.py | 137 |
1 files changed, 0 insertions, 137 deletions
diff --git a/resources/tools/scripts/topo_container_copy.py b/resources/tools/scripts/topo_container_copy.py deleted file mode 100644 index 83599b4444..0000000000 --- a/resources/tools/scripts/topo_container_copy.py +++ /dev/null @@ -1,137 +0,0 @@ -#!/usr/bin/env python - -# Copyright (c) 2017 Cisco and/or its affiliates. -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at: -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -"""This script provides copy and load of Docker container images. - As destinations are used all DUT nodes from the topology file.""" - -import sys -import argparse -from yaml import load - -from resources.libraries.python.ssh import SSH - - -def ssh_no_error(ssh, cmd, sudo=False): - """Execute a command over ssh channel, and log and exit if the command - fails. - - :param ssh: SSH() object connected to a node. - :param cmd: Command line to execute on remote node. - :param sudo: Run command with sudo privileges. - :type ssh: SSH() object - :type cmd: str - :type sudo: bool - :returns: stdout from the SSH command. - :rtype: str - :raises RuntimeError: In case of unexpected ssh command failure - """ - if sudo: - ret, stdo, stde = ssh.exec_command_sudo(cmd, timeout=60) - else: - ret, stdo, stde = ssh.exec_command(cmd, timeout=60) - - if ret != 0: - print('Command execution failed: "{}"'.format(cmd)) - print('stdout: {0}'.format(stdo)) - print('stderr: {0}'.format(stde)) - raise RuntimeError('Unexpected ssh command failure') - - return stdo - - -def ssh_ignore_error(ssh, cmd, sudo=False): - """Execute a command over ssh channel, ignore errors. - - :param ssh: SSH() object connected to a node. - :param cmd: Command line to execute on remote node. - :param sudo: Run command with sudo privileges. - :type ssh: SSH() object - :type cmd: str - :type sudo: bool - :returns: stdout from the SSH command. - :rtype: str - """ - if sudo: - ret, stdo, stde = ssh.exec_command_sudo(cmd) - else: - ret, stdo, stde = ssh.exec_command(cmd) - - if ret != 0: - print('Command execution failed: "{}"'.format(cmd)) - print('stdout: {0}'.format(stdo)) - print('stderr: {0}'.format(stde)) - - return stdo - - -def main(): - """Copy and load of Docker image.""" - parser = argparse.ArgumentParser() - parser.add_argument("-t", "--topo", required=True, - help="Topology file") - parser.add_argument("-d", "--directory", required=True, - help="Destination directory") - parser.add_argument("-i", "--images", required=False, nargs='+', - help="Images paths to copy") - parser.add_argument("-c", "--cancel", help="Cancel all", - action="store_true") - - args = parser.parse_args() - topology_file = args.topo - images = args.images - directory = args.directory - cancel_all = args.cancel - - work_file = open(topology_file) - topology = load(work_file.read())['nodes'] - - ssh = SSH() - for node in topology: - if topology[node]['type'] == "DUT": - print("###TI host: {host}".format(host=topology[node]['host'])) - ssh.connect(topology[node]) - - if cancel_all: - # Remove destination directory on DUT - cmd = "rm -r {directory}".format(directory=directory) - stdout = ssh_ignore_error(ssh, cmd) - print("###TI {stdout}".format(stdout=stdout)) - - else: - # Create installation directory on DUT - cmd = "rm -r {directory}; mkdir {directory}"\ - .format(directory=directory) - stdout = ssh_no_error(ssh, cmd) - print("###TI {stdout}".format(stdout=stdout)) - - # Copy images from local path to destination dir - for image in images: - print("###TI scp: {}".format(image)) - ssh.scp(local_path=image, remote_path=directory) - - # Load image to Docker. - cmd = "for f in {directory}/*.tar.gz; do "\ - "sudo docker load -i $f; done".format(directory=directory) - stdout = ssh_no_error(ssh, cmd) - print("###TI {}".format(stdout)) - - # Remove <none> images from Docker. - cmd = "docker rmi $(sudo docker images -f 'dangling=true' -q)" - stdout = ssh_ignore_error(ssh, cmd, sudo=True) - print("###TI {}".format(stdout)) - - -if __name__ == "__main__": - sys.exit(main()) |