setup.py

# $Header: //prod/main/ap/shrapnel/setup.py#17 $
#!/usr/bin/env python

import sys
import glob
import os

from distribute_setup import use_setuptools
use_setuptools()
from setuptools import setup

try:
    from Cython.Distutils import build_ext
    from Cython.Distutils.extension import Extension
except ImportError:
    sys.stderr.write (
        '\nThe Cython compiler is required to build Shrapnel.\n'
        '  Try "pip install cython"\n'
        '  *or* "easy_install cython"\n'
        )
    sys.exit (-1)


include_dir = os.getcwd()

def newer(x, y):
    x_mtime = os.path.getmtime(x)
    try:
        y_mtime = os.path.getmtime(y)
    except OSError:
        return True
    return x_mtime > y_mtime

def exit_ok(status):
    return os.WIFEXITED(status) and os.WEXITSTATUS(status) == 0

def check_lio():
    if os.getenv('BUILDING') is not None:
        # Doing this in a build environment.
        # This is a terrible hack.  However, there is no easy way to add
        # arbitrary options to distutils.
        return True
    if newer('test/build/test_lio.c', 'test/build/test_lio'):
        status = os.system('gcc -o test/build/test_lio test/build/test_lio.c')
        if not exit_ok(status):
            return False
    status = os.system('test/build/test_lio')
    return exit_ok(status)

setup (
    name='coro',
    version='1.0.2-000',
    description='IronPort Coroutine/Threading Library',
    author='Sam Rushing, Eric Huss, IronPort Engineering',
    author_email='sam-coro@rushing.nightmare.com',
    license = "MIT",
    url = "http://github.com/ironport/shrapnel",
    ext_modules = [
        Extension(
            'coro.event_queue',
            ['coro/event_queue.pyx'],
            language='c++',
            depends=[os.path.join(include_dir, 'pyrex', 'python.pxi'),],
            pyrex_include_dirs=[
                os.path.join(include_dir, '.'),
                os.path.join(include_dir, 'pyrex'),
                ],),
        Extension (
            'coro._coro',
            ['coro/_coro.pyx', 'coro/swap.c'],
            extra_compile_args = ['-Wno-unused-function'],
            depends=(glob.glob('coro/*.pyx') +
                     glob.glob('coro/*.pxi') +
                     glob.glob('coro/*.pxd') +
                     [os.path.join(include_dir, 'pyrex', 'python.pxi'),
                      os.path.join(include_dir, 'pyrex', 'pyrex_helpers.pyx'),
                      os.path.join(include_dir, 'include', 'pyrex_helpers.h'),
                      os.path.join(include_dir, 'pyrex',
                                   'tsc_time_include.pyx'),
                      os.path.join(include_dir, 'include', 'tsc_time.h'),
                      os.path.join(include_dir, 'pyrex', 'libc.pxd'),
                     ]
                    ),
            pyrex_include_dirs=[
                os.path.join(include_dir, '.'),
                os.path.join(include_dir, 'pyrex'),
                ],
            #include_dirs=[os.path.join(include_dir, 'pyrex')],
            include_dirs=[
                os.path.join(include_dir, '.'),
                os.path.join(include_dir, 'include'),
                ],
            #pyrex_compile_time_env={'COMPILE_LIO': check_lio(),
            #                        'CORO_DEBUG': True,
            #                       },
            # to enable LZO|LZ4 for stack compression, set COMPILE_LZO|COMPILE_LZ4 in coro/_coro.pyx
            #   and uncomment one of the following:
            #libraries=['lzo2', 'z']
            #libraries=['lz4', 'z'],
            libraries=['z']
            ),
        Extension(
            'coro.oserrors',
            ['coro/oserrors.pyx', ],
            ),
        Extension ('coro.dns.packet', ['coro/dns/packet.pyx', ],),
        Extension (
            'coro.clocks.tsc_time',
            ['coro/clocks/tsc_time.pyx', ],
            pyrex_include_dirs=[os.path.join(include_dir, 'pyrex')],
            include_dirs=[
                os.path.join(include_dir, '.'),
                os.path.join(include_dir, 'include'),
                ],
            ),
        ],
    packages=['coro', 'coro.clocks', 'coro.http', 'coro.dns', 'coro.emulation'],
    package_dir = {
#        '': 'coroutine',
        'coro': 'coro',
        'coro.clocks': 'coro/clocks',
        'coro.dns': 'coro/dns',
        'coro.emulation': 'coro/emulation',
    },
    py_modules = ['backdoor', 'coro.read_stream', 'coro_process', 'coro_unittest',],
    download_url = 'http://github.com/ironport/shrapnel/tarball/master#egg=coro-1.0.2',
    install_requires = ['Cython>=0.12.1', 'distribute>=0.6.16'],
    cmdclass={'build_ext': build_ext},
)