07e47e1593
Change-Id: I24810421f82da3d9d0ba6600b98955f28569a6a8 Closes-bug: 1261518
171 lines
5.9 KiB
Python
171 lines
5.9 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2013 OpenStack Foundation
|
|
# Copyright 2013 IBM Corp.
|
|
#
|
|
# 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.
|
|
|
|
"""Provides methods needed by installation script for OpenStack development
|
|
virtual environments.
|
|
|
|
Synced in from openstack-common
|
|
"""
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
class InstallVenv(object):
|
|
|
|
def __init__(self, root, venv, pip_requires, test_requires, py_version,
|
|
project):
|
|
self.root = root
|
|
self.venv = venv
|
|
self.pip_requires = pip_requires
|
|
self.test_requires = test_requires
|
|
self.py_version = py_version
|
|
self.project = project
|
|
|
|
def die(self, message, *args):
|
|
print >> sys.stderr, message % args
|
|
sys.exit(1)
|
|
|
|
def check_python_version(self):
|
|
if sys.version_info < (2, 6):
|
|
self.die("Need Python Version >= 2.6")
|
|
|
|
def run_command_with_code(self, cmd, redirect_output=True,
|
|
check_exit_code=True):
|
|
"""Runs a command in an out-of-process shell.
|
|
|
|
Returns the output of that command. Working directory is self.root.
|
|
"""
|
|
if redirect_output:
|
|
stdout = subprocess.PIPE
|
|
else:
|
|
stdout = None
|
|
|
|
proc = subprocess.Popen(cmd, cwd=self.root, stdout=stdout)
|
|
output = proc.communicate()[0]
|
|
if check_exit_code and proc.returncode != 0:
|
|
self.die('Command "%s" failed.\n%s', ' '.join(cmd), output)
|
|
return (output, proc.returncode)
|
|
|
|
def run_command(self, cmd, redirect_output=True, check_exit_code=True):
|
|
return self.run_command_with_code(cmd, redirect_output,
|
|
check_exit_code)[0]
|
|
|
|
def get_distro(self):
|
|
if (os.path.exists('/etc/fedora-release') or
|
|
os.path.exists('/etc/redhat-release')):
|
|
return Fedora(self.root, self.venv, self.pip_requires,
|
|
self.test_requires, self.py_version, self.project)
|
|
else:
|
|
return Distro(self.root, self.venv, self.pip_requires,
|
|
self.test_requires, self.py_version, self.project)
|
|
|
|
def check_dependencies(self):
|
|
self.get_distro().install_virtualenv()
|
|
|
|
def create_virtualenv(self, no_site_packages=True):
|
|
"""Creates the virtual environment and installs PIP.
|
|
|
|
Creates the virtual environment and installs PIP only into the
|
|
virtual environment.
|
|
"""
|
|
if not os.path.isdir(self.venv):
|
|
print 'Creating venv...',
|
|
if no_site_packages:
|
|
self.run_command(['virtualenv', '-q', '--no-site-packages',
|
|
self.venv])
|
|
else:
|
|
self.run_command(['virtualenv', '-q', self.venv])
|
|
print 'done.'
|
|
print 'Installing pip in venv...',
|
|
if not self.run_command(['tools/with_venv.sh', 'easy_install',
|
|
'pip>1.0']).strip():
|
|
self.die("Failed to install pip.")
|
|
print 'done.'
|
|
else:
|
|
print "venv already exists..."
|
|
pass
|
|
|
|
def pip_install(self, *args):
|
|
self.run_command(['tools/with_venv.sh',
|
|
'pip', 'install', '--upgrade'] + list(args),
|
|
redirect_output=False)
|
|
|
|
def install_dependencies(self):
|
|
print 'Installing dependencies with pip (this can take a while)...'
|
|
self.pip_install('pip>=1.3')
|
|
self.pip_install('setuptools')
|
|
self.pip_install('-r', self.pip_requires, '-r', self.test_requires)
|
|
|
|
def parse_args(self, argv):
|
|
"""Parses command-line arguments."""
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-n', '--no-site-packages',
|
|
action='store_true',
|
|
help="Do not inherit packages from global Python "
|
|
"install")
|
|
return parser.parse_args(argv[1:])
|
|
|
|
|
|
class Distro(InstallVenv):
|
|
|
|
def check_cmd(self, cmd):
|
|
return bool(self.run_command(['which', cmd],
|
|
check_exit_code=False).strip())
|
|
|
|
def install_virtualenv(self):
|
|
if self.check_cmd('virtualenv'):
|
|
return
|
|
|
|
if self.check_cmd('easy_install'):
|
|
print 'Installing virtualenv via easy_install...',
|
|
if self.run_command(['easy_install', 'virtualenv']):
|
|
print 'Succeeded'
|
|
return
|
|
else:
|
|
print 'Failed'
|
|
|
|
self.die('ERROR: virtualenv not found.\n\n%s development'
|
|
' requires virtualenv, please install it using your'
|
|
' favorite package management tool' % self.project)
|
|
|
|
|
|
class Fedora(Distro):
|
|
"""This covers all Fedora-based distributions.
|
|
|
|
Includes: Fedora, RHEL, CentOS, Scientific Linux
|
|
"""
|
|
|
|
def check_pkg(self, pkg):
|
|
return self.run_command_with_code(['rpm', '-q', pkg],
|
|
check_exit_code=False)[1] == 0
|
|
|
|
def yum_install(self, pkg, **kwargs):
|
|
print "Attempting to install '%s' via yum" % pkg
|
|
self.run_command(['sudo', 'yum', 'install', '-y', pkg], **kwargs)
|
|
|
|
def install_virtualenv(self):
|
|
if self.check_cmd('virtualenv'):
|
|
return
|
|
|
|
if not self.check_pkg('python-virtualenv'):
|
|
self.yum_install('python-virtualenv', check_exit_code=False)
|
|
|
|
super(Fedora, self).install_virtualenv()
|