aboutsummaryrefslogtreecommitdiffstats
path: root/resources/tools/scripts/topo_container_copy.py
blob: d2431822360761aa63122f0a41e6e172fe32aa69 (plain)
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
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())