cloud-init/setup.py
Joshua Harlow d043d304f4 Add the ability to have setup.py have a CLI option that specifies the daemon type
which then later affects the installation of certain config files, which then
can be extracted during package creation as needed.
2012-07-03 16:04:58 -07:00

152 lines
5.3 KiB
Python
Executable File

# vi: ts=4 expandtab
#
# Distutils magic for ec2-init
#
# Copyright (C) 2009 Canonical Ltd.
# Copyright (C) 2012 Yahoo! Inc.
#
# Author: Soren Hansen <soren@canonical.com>
# Author: Joshua Harlow <harlowja@yahoo-inc.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from glob import glob
import os
import re
import setuptools
from setuptools.command.install import install
from distutils.command.install_data import install_data
from distutils.errors import DistutilsArgError
import subprocess
def is_f(p):
return os.path.isfile(p)
DAEMON_FILES = {
'initd': filter((lambda x: is_f(x)
and x.find('local') == -1), glob('initd/*')),
'initd-local': filter((lambda x: is_f(x)
and not x.endswith('cloud-init')), glob('initd/*')),
'systemd': filter((lambda x: is_f(x)), glob('systemd/*')),
'upstart': filter((lambda x: is_f(x)
and x.find('local') == -1
and x.find('nonet') == -1), glob('upstart/*')),
'upstart-nonet': filter((lambda x: is_f(x)
and x.find('local') == -1
and not x.endswith('cloud-init.conf')), glob('upstart/*')),
'upstart-local': filter((lambda x: is_f(x)
and x.find('nonet') == -1
and not x.endswith('cloud-init.conf')), glob('upstart/*')),
}
DAEMON_ROOTS = {
'initd': '/etc/rc.d/init.d',
'initd-local': '/etc/rc.d/init.d',
'systemd': '/etc/systemd/system/',
'upstart': '/etc/init/',
'upstart-nonet': '/etc/init/',
'upstart-local': '/etc/init/',
}
DAEMON_TYPES = sorted(list(DAEMON_ROOTS.keys()))
def tiny_p(cmd, capture=True):
# Darn python 2.6 doesn't have check_output (argggg)
stdout = subprocess.PIPE
stderr = subprocess.PIPE
if not capture:
stdout = None
stderr = None
sp = subprocess.Popen(cmd, stdout=stdout,
stderr=stderr, stdin=None)
(out, err) = sp.communicate()
if sp.returncode not in [0]:
raise RuntimeError("Failed running %s [rc=%s] (%s, %s)"
% (cmd, sp.returncode, out, err))
return (out, err)
def get_version():
cmd = ['tools/read-version']
(ver, _e) = tiny_p(cmd)
return ver.strip()
def read_requires():
cmd = ['tools/read-dependencies']
(deps, _e) = tiny_p(cmd)
return deps.splitlines()
# TODO: Is there a better way to do this??
class DaemonInstallData(install):
user_options = install.user_options + [
# This will magically show up in member variable 'daemon_type'
('daemon-type=', None,
('daemon type to configure (%s) [default: None]') %
(", ".join(DAEMON_TYPES))
),
]
def initialize_options(self):
install.initialize_options(self)
self.daemon_type = None
def finalize_options(self):
install.finalize_options(self)
if self.daemon_type and self.daemon_type not in DAEMON_TYPES:
raise DistutilsArgError(
("You must specify one of (%s) when"
" specifying a daemon type!") % (", ".join(DAEMON_TYPES))
)
elif self.daemon_type:
self.distribution.data_files.append((DAEMON_ROOTS[self.daemon_type],
DAEMON_FILES[self.daemon_type]))
# Force that command to reinitalize (with new file list)
self.distribution.reinitialize_command('install_data', True)
setuptools.setup(name='cloud-init',
version=get_version(),
description='EC2 initialisation magic',
author='Scott Moser',
author_email='scott.moser@canonical.com',
url='http://launchpad.net/cloud-init/',
packages=setuptools.find_packages(exclude=['tests']),
scripts=['bin/cloud-init',
'tools/cloud-init-per',
],
license='GPLv3',
data_files=[('/etc/cloud', glob('config/*.cfg')),
('/etc/cloud/cloud.cfg.d', glob('config/cloud.cfg.d/*')),
('/etc/cloud/templates', glob('templates/*')),
('/usr/share/cloud-init', []),
('/usr/lib/cloud-init',
['tools/uncloud-init', 'tools/write-ssh-key-fingerprints']),
('/usr/share/doc/cloud-init', filter(is_f, glob('doc/*'))),
('/usr/share/doc/cloud-init/examples', filter(is_f, glob('doc/examples/*'))),
('/usr/share/doc/cloud-init/examples/seed', filter(is_f, glob('doc/examples/seed/*'))),
],
install_requires=read_requires(),
cmdclass = {
# Use a subclass for install that handles
# adding on the right daemon configuration files
'install': DaemonInstallData,
},
)