setup.py

#   Copyright 2011 Red Hat, Inc.
#
#   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.

from distutils.core import setup, Extension
from distutils.command.sdist import sdist as _sdist
import os
import os.path
import subprocess

version_file_name = "version.txt"
try:
    if(not os.path.exists(version_file_name)):
        subprocess.call('/usr/bin/git describe | tr - _ > %s' % (version_file_name, ), shell=True)
    version_file = open(version_file_name, "r")
    VERSION = version_file.read()[0:-1]
    version_file.close()
except Exception, e:
    raise RuntimeError("ERROR: version.txt could not be found.  Run 'git describe > version.txt' to get the correct version info.")

datafiles=[('share/man/man1', ['Documentation/man/imagefactory.1']),
           ('/etc/imagefactory', ['imagefactory.conf']),
           ('/etc/pki/imagefactory', ['cert-ec2.pem']),
           ('/etc/sysconfig', ['imagefactory']),
           ('/etc/logrotate.d', ['imagefactory']),
           ('/etc/rc.d/init.d', ['scripts/imagefactory'])]

class sdist(_sdist):
    """ custom sdist command to prepare imagefactory.spec file """

    def run(self):
        cmd = (""" sed -e "s/@VERSION@/%s/g" < imagefactory.spec.in """ %
               VERSION) + " > imagefactory.spec"
        os.system(cmd)

        _sdist.run(self)

setup(name='imagefactory',
      version=VERSION,
      description='Image Factory system image generation tool',
      author='Ian McLeod',
      author_email='imcleod@redhat.com',
      license='GPLv2',
      url='http://www.aeolusproject.org/imagefactory.html',
      packages=['imgfac', 'imgfac.builders', 'imgfac.rest'],
      scripts=['imagefactory'],
      data_files = datafiles,
      cmdclass = {'sdist': sdist}
      )