root/releases/pkgcore/0.1.1/setup.py @ marienz%2540gentoo.org-20061002235229-d379ff51f9cb294c

#!/usr/bin/env python

import glob
import os
import sys
import errno

from distutils import core, ccompiler, log, sysconfig
from distutils.command import build, sdist, build_py
from stat import ST_MODE

class mysdist(sdist.sdist):
    default_format = dict(sdist.sdist.default_format)
    default_format["posix"] = "bztar"

    def get_file_list(self):
        for key, globs in self.distribution.package_data.iteritems():
            for pattern in globs:
                self.filelist.extend(glob.glob(os.path.join(key, pattern)))
        self.filelist.append("ChangeLog")
        self.filelist.append("AUTHORS")
        self.filelist.append("sandbox/test.py")
        self.filelist.extend(glob.glob("sandbox/test-plugins/*"))
        self.filelist.extend(glob.glob('doc/*.rst'))
        self.filelist.extend(glob.glob('dev-notes/*.rst'))
        self.filelist.extend(glob.glob('dev-notes/reimplementation/*.rst'))
        self.filelist.extend(glob.glob('dev-notes/framework/*.rst'))
        self.filelist.append('build_docs.py')
        # XXX HACK: if you run "setup.py sdist" with python 2.5 this
        # does not get packaged without this.
        self.filelist.append('pkgcore/util/_functoolsmodule.c')
        sdist.sdist.get_file_list(self)

    def run(self):
        print "regenning ChangeLog (may take a while)"
        os.system("bzr log --verbose > ChangeLog")
        sdist.sdist.run(self)


class build_filter_env(core.Command):

    """Build the filter-env utility.

    This rips a bunch of code from the distutils build_clib command.
    """

    user_options = [
        ('debug', 'g', 'compile with debugging information'),
        ('force', 'f', 'compile everything (ignore timestamps)'),
        ('compiler=', 'c', 'specify the compiler type'),
        ]

    boolean_options = ['debug', 'force']

    help_options = [
        ('help-compiler', None,
         'list available compilers', ccompiler.show_compilers),
        ]

    def initialize_options(self):
        """If we had any options we would initialize them here."""
        self.debug = None
        self.force = 0
        self.compiler = None

    def finalize_options(self):
        """If we had any options we would finalize them here."""
        self.set_undefined_options(
            'build',
            ('debug', 'debug'),
            ('force', 'force'),
            ('compiler', 'compiler'))

    def run(self):
        compiler = ccompiler.new_compiler(
            compiler=self.compiler, dry_run=self.dry_run, force=self.force)
        sysconfig.customize_compiler(compiler)
        cc = ' '.join(compiler.compiler)
        
        for x in ("BASECFLAGS", "CCSHARED", "LDFLAGS"):
            f = sysconfig.get_config_var(x)
            if isinstance(f, basestring):
                cc = cc.replace(f, '')
            elif f is None:
                continue
            else:
                cc = cc.replace(" ".join(f), '')

        compiler.set_executables(compiler=cc, compiler_so=cc)
        objects = compiler.compile(list(
                os.path.join('src', 'filter-env', name)
                for name in ('main.c', 'bmh_search.c')), debug=self.debug)
        compiler.link(compiler.EXECUTABLE, objects, os.path.join(
                'pkgcore', 'bin', 'ebuild-env', 'filter-env'))


build.build.sub_commands.append(('build_filter_env', None))

class hacked_build_py(build_py.build_py):

    def run(self):
        build_py.build_py.run(self)

        fp = os.path.join(self.build_lib, "pkgcore", "bin", "ebuild-helpers")
        for f in os.listdir(fp):
            self.set_chmod(os.path.join(fp, f))
        fp = os.path.join(self.build_lib, "pkgcore", "bin", "ebuild-env")
        for f in ("ebuild.sh", "ebuild-daemon.sh"):
            self.set_chmod(os.path.join(fp, f))
        if os.path.exists(os.path.join(fp, "filter-env")):
            self.set_chmod(os.path.join(fp, "filter-env"))

    def set_chmod(self, path):
        if self.dry_run:
            log.info("changing mode of %s", path)
        else:
            mode = ((os.stat(path)[ST_MODE]) | 0555) & 07777
            log.info("changing mode of %s to %o", path, mode)
            os.chmod(path, mode)


packages = []

for root, dirs, files in os.walk('pkgcore'):
    if '__init__.py' in files:
        package = root.replace(os.path.sep, '.')
        print 'adding package %r' % (package,)
        packages.append(package)

try:
    os.unlink("MANIFEST")
except OSError, oe:
    if oe.errno != errno.ENOENT:
        raise
    del oe


extra_flags = ['-Wall']

extensions = []
if sys.version_info < (2, 5):
    # Almost unmodified copy from the python 2.5 source.
    extensions.append(core.Extension(
            'pkgcore.util._functools', ['pkgcore/util/_functoolsmodule.c'],
            extra_compile_args=extra_flags))


core.setup(
    name='pkgcore',
    version='0.1.1',
    description='package managing framework',
    url='http://gentooexperimental.org/~ferringb/bzr/pkgcore/',
    packages=packages,
    package_data={
        'pkgcore': [
            'data/*',
            'bin/ebuild-env/*',
            'bin/ebuild-helpers/*',
            'heapdef.h',
            ],
        'src': [
            'filter-env/*.c',
            'filter-env/*.h',
            'bsd-flags/*',
            'tbz2tool.c'
            ],
        },
    # booo, no glob support in distutils for this one
    scripts=(
        glob.glob('pkgcore/bin/utilities/*.py') +
        glob.glob('bin/*')),
    ext_modules=[
        core.Extension('pkgcore.util._caching', ['pkgcore/util/_caching.c'],
                       extra_compile_args=extra_flags),
        core.Extension('pkgcore.util._lists', ['pkgcore/util/_lists.c'],
                       extra_compile_args=extra_flags),
        core.Extension('pkgcore.ebuild._cpv', ['pkgcore/ebuild/_cpv.c'],
                       extra_compile_args=extra_flags),
        core.Extension('pkgcore.util.osutils._readdir',
                       ['pkgcore/util/osutils/_readdir.c'],
                       extra_compile_args=extra_flags),
        ] + extensions,
    cmdclass={'build_filter_env': build_filter_env,
              'sdist': mysdist,
              'build_py': hacked_build_py},
    )