summaryrefslogtreecommitdiffstats
path: root/scripts/container.py
blob: 3223c9931f7e559c1d4939368827190d99c6fb91 (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
#!/usr/bin/env python
#
# grakiss.wanglei@huawei.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
#

import utils.dovetail_logger as dt_logger
import utils.dovetail_utils as dt_utils
from conf.dovetail_config import *

logger = dt_logger.Logger('container.py').getLogger()

class Container:

    container_list = {}
    has_pull_latest_image = {'yardstick':False, 'functest':False}
    has_build_images = {'yardstick':False, 'functest':False}

    def __init__(cls):
        pass

    def __str__(cls):
        pass

    @classmethod
    def get(cls, type):
        return cls.container_list[type]

    @classmethod
    def get_docker_image(cls, type):
        return '%s:%s' % (dovetail_config[type]['image_name'], dovetail_config[type]['docker_tag'])

    @classmethod
    def create(cls, type):
        #sshkey="-v /root/.ssh/id_rsa:/root/.ssh/id_rsa "
        docker_image = cls.get_docker_image(type)
        envs = dovetail_config[type]['envs']
        opts = dovetail_config[type]['opts']
        sshkey = ''
        result_volume = ' -v %s:%s ' % (dovetail_config['result_dir'],dovetail_config[type]['result']['dir'])
        cmd = 'sudo docker run %s %s %s %s %s /bin/bash' % (opts, envs, sshkey, result_volume, docker_image)
        dt_utils.exec_cmd(cmd,logger)
        ret, container_id=dt_utils.exec_cmd("sudo docker ps | grep "+ docker_image + " | awk '{print $1}' | head -1",logger)
        cls.container_list[type] = container_id
        return container_id

    @classmethod
    def pull_image(cls, type):
        docker_image = cls.get_docker_image(type)
        if cls.has_pull_latest_image[type] == True:
            logger.debug('%s is already the newest version.' % (docker_image))
        else:
            cmd = 'sudo docker pull %s' % (docker_image)
            dt_utils.exec_cmd(cmd,logger)
            cls.has_pull_latest_image[type] = True

    @classmethod
    def clean(cls, container_id):
        cmd1 = 'sudo docker stop %s' % (container_id)
        dt_utils.exec_cmd(cmd1,logger)
        cmd2 = 'sudo docker rm %s' % (container_id)
        dt_utils.exec_cmd(cmd2,logger)

    @classmethod
    def exec_cmd(cls, container_id, sub_cmd, exit_on_error=False):
        cmd = 'sudo docker exec %s /bin/bash -c "%s"' % (container_id, sub_cmd)
        dt_utils.exec_cmd(cmd,logger,exit_on_error)

    @classmethod
    def copy_file(cls, file_dir, container_id, container_dir):
        for root, dirs, files in os.walk(file_dir):
            for file_name in files:
                cmd = 'sudo docker cp %s %s:%s' % (os.path.join(file_dir,file_name), container_id, container_dir)
                dt_utils.exec_cmd(cmd, logger, exit_on_error = False)