utils/infra_setup/runner/docker_usage.py

   1 #!/usr/bin/env python
   2 ##############################################################################
   3 # Copyright (c) 2017 Huawei Technologies Co.,Ltd and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9 ##############################################################################
  10 '''This file contain some function about docker API.
  11 At present, This file contain the following function:
  12 1.Ask Docker service to create a docker(yardstick or ELK).
  13 2.get a docker ip.
  14 3.Remove a docker.'''
  15
  16 import docker
  17
  18 yardstick_info = None
  19 ELK_info = None
  20
  21
  22 def get_client():
  23     client = docker.from_env()
  24     return client
  25
  26
  27 def docker_find(docker_name):
  28     client = get_client()
  29     docker_client = client.containers.get(docker_name)
  30     return docker_client
  31
  32
  33 def env_yardstick(docker_name):
  34     client = get_client()
  35     yardstick_info["name"] = docker_name
  36     try:
  37         env_docker = docker_find(docker_name)
  38         yardstick_info["containner"] = env_docker
  39         yardstick_info["ip"] = get_docker_ip(docker_name)
  40         return env_docker
  41     except docker.errors.NotFound:
  42         pass
  43     env_docker = client.containers.run(image="opnfv/yardstick:latest",
  44                                        privileged=True,
  45                                        detach=True,
  46                                        ports={'8888': '5000'},
  47                                        volumes={'/var/run/docker.sock':
  48                                                 '/var/run/docker.sock'},
  49                                        name=docker_name)
  50     yardstick_info["containner"] = env_docker
  51     yardstick_info["ip"] = get_docker_ip(docker_name)
  52     return env_docker
  53
  54
  55 def env_elk(docker_name):
  56     client = get_client()
  57     ELK_info["name"] = docker_name
  58     try:
  59         env_docker = docker_find(docker_name)
  60         ELK_info["container"] = env_docker
  61         ELK_info["ip"] = get_docker_ip(docker_name)
  62         return env_docker
  63     except docker.errors.NotFound:
  64         pass
  65     env_docker = client.containers.run(image="sebp/elk:es241_l240_k461",
  66                                        privileged=True,
  67                                        detach=True,
  68                                        ports={'5044': '5044',
  69                                               '5601': '5601',
  70                                               '9200': '9200'},
  71                                        name=docker_name)
  72     ELK_info["container"] = env_docker
  73     ELK_info["ip"] = get_docker_ip(docker_name)
  74     return env_docker
  75
  76
  77 def get_docker_ip(docker_name):
  78     env_docker = docker_find(docker_name)
  79     client = docker.APIClient(base_url='unix://var/run/docker.sock')
  80     ip_address = client.inspect_container(env_docker.id)
  81     return ip_address
  82
  83
  84 def docker_exec_cmd(docker, cmd):
  85     return docker.exec_cmd(cmd)
  86
  87
  88 def remove_docker(docker_name):
  89     docker = docker_find(docker_name)
  90     docker.kill()
  91     docker.remove()