summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorzshi <zshi@redhat.com>2016-10-20 13:02:38 +0800
committerzshi <zshi@redhat.com>2016-10-26 17:48:41 +0800
commite96b958cb480463f29c237f8aace813e613c4a82 (patch)
tree567d113304bb5bfdf447f03b43d4fb25d2524350
parent9b473f90a40cea5671f52665de694d58f7b2a1bf (diff)
Add more distro support when preparing environment
Change-Id: I3ff4419e38872bc28e59699a11080777ea596dc3 Signed-off-by: zshi <zshi@redhat.com>
-rw-r--r--dovetail/prepare_env.py40
1 files changed, 39 insertions, 1 deletions
diff --git a/dovetail/prepare_env.py b/dovetail/prepare_env.py
index 785d5c3d..3e4d6964 100644
--- a/dovetail/prepare_env.py
+++ b/dovetail/prepare_env.py
@@ -7,13 +7,51 @@
# http://www.apache.org/licenses/LICENSE-2.0
#
+import platform
import utils.dovetail_logger as dt_logger
import utils.dovetail_utils as dt_utils
+def get_os():
+ """Get distro name.
+
+ :returns: return distro name as a string
+ """
+ return platform.dist()[0]
+
+
+def get_install_bin(os):
+ """Get install command binary.
+
+ :returns: return install command according to distro
+ """
+ if os in ['centos', 'redhat']:
+ return 'yum'
+ elif os == 'fedora':
+ return 'dnf'
+ elif os == 'ubuntu':
+ return 'apt-get'
+ else:
+ return None
+
+
+def get_docker_pkgname(os):
+ """Get docker package name.
+
+ :returns: return docker package name according to distro
+ """
+ if os in ['centos', 'fedora', 'redhat']:
+ return 'docker'
+ elif os == 'ubuntu':
+ return 'docker.io'
+ else:
+ return None
+
logger = dt_logger.Logger('prepare_env.py').getLogger()
-cmd = "sudo apt-get -y install docker.io python-pip"
+os_name = get_os()
+cmd = "sudo %s -y install %s python-pip" \
+ % (get_install_bin(os_name), get_docker_pkgname(os_name))
dt_utils.exec_cmd(cmd, logger)
cmd = "sudo pip install click pyyaml jinja2"