setup.py

from setuptools import setup, find_packages
from setuptools.extension import Extension
import subprocess
import re
import numpy as np


def get_cython_version():
    """
    Returns:
        Version as a pair of ints (major, minor)

    Raises:
        ImportError: Can't load cython or find version
    """
    import Cython.Compiler.Main
    match = re.search('^([0-9]+)\.([0-9]+)',
                      Cython.Compiler.Main.Version.version)
    try:
        return map(int, match.groups())
    except AttributeError:
        raise ImportError

# Only use Cython if it is available, else just use the pre-generated files
try:
    cython_version = get_cython_version()
    # Requires Cython version 0.13 and up
    if cython_version[0] == 0 and cython_version[1] < 13:
        raise ImportError
    from Cython.Distutils import build_ext
    source_ext = '.pyx'
    cmdclass = {'build_ext': build_ext}
except ImportError:
    source_ext = '.c'
    cmdclass = {}

# TODO This is a nasty hack, eventually we want to wrap everything with Cython
def compile_shared():
    subprocess.call('g++ -o classipy/lib/liblinear.so -shared thirdparty/liblinear-1.6/linear.os thirdparty/liblinear-1.6/tron.os thirdparty/liblinear-1.6/blas/dnrm2.os thirdparty/liblinear-1.6/blas/daxpy.os thirdparty/liblinear-1.6/blas/ddot.os thirdparty/liblinear-1.6/blas/dscal.os'.split())
    subprocess.check_call('g++ -o classipy/lib/libsvm.so -shared thirdparty/libsvm-3.1/svm.os'.split())

try:
    compile_shared()
except OSError:
    print("Wasn't able to compile necessary shared libraries")
    raise
ext_modules = [Extension("_classipy_rand_forest", ["classipy/classifiers/rand_forest/rand_forest" + source_ext,
                                                   'classipy/classifiers/rand_forest/fast_hist.c'],
                         extra_compile_args=['-I', np.get_include()]),
               Extension("_classipy_kernels", ["classipy/classifiers/kernels/kernels" + source_ext,
                                               'classipy/classifiers/kernels/kernels_aux.c'],
                         extra_compile_args=['-I', np.get_include()])]
setup(name='classipy',
      cmdclass=cmdclass,
      version='0.0.3',
      packages=find_packages(),
      package_data={'classipy' : ['lib/*.so']},
      author='Brandyn A. White',
      author_email='bwhite@dappervision.com',
      license='GPL',
      url='https://github.com/bwhite/classipy',
      ext_modules=ext_modules)