aboutsummaryrefslogtreecommitdiffstats
path: root/resources/tools/scripts/topo_container_copy.py
diff options
context:
space:
mode:
Diffstat (limited to 'resources/tools/scripts/topo_container_copy.py')
-rw-r--r--resources/tools/scripts/topo_container_copy.py132
1 files changed, 132 insertions, 0 deletions
diff --git a/resources/tools/scripts/topo_container_copy.py b/resources/tools/scripts/topo_container_copy.py
new file mode 100644
index 0000000000..d243182236
--- /dev/null
+++ b/resources/tools/scripts/topo_container_copy.py
@@ -0,0 +1,132 @@
+#!/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 zcat $f | "\
+ "sudo docker load; done".format(directory=directory)
+ stdout = ssh_no_error(ssh, cmd)
+ print("###TI {}".format(stdout))
+
+
+if __name__ == "__main__":
+ sys.exit(main())