setup.py

#!/usr/bin/env python

import sys
import os
import numpy as np
import platform
from Cython.Build import cythonize
try:
    from setuptools import setup
    from setuptools import Extension
except ImportError:
    from distutils.core import setup
    from distutils.extension import Extension
try:
    from urllib import request as req
except ImportError:
    import urllib as req


includeDirs = [ "build/include", np.get_include() ]
libraryDirs = [ "build/lib" ]

def download_klb_files( includeDirs, libraryDirs ):
    # version (by commit id) of main library to use
    klbCommitId = "5edcaecc858911c7b3855579bde5cb3116cb4680"
    
    
    klbUrl = "https://bitbucket.org/fernandoamat/keller-lab-block-filetype/raw/%s" % klbCommitId
    
    downloadFiles = [
        # collect downloads, in format (sourceFileUrl, targetDir)
        ("%s/src/common.h" % klbUrl, includeDirs[0]),
        ("%s/src/klb_Cwrapper.h" % klbUrl, includeDirs[0])
        ]
    
    
    # download main library dependency
    errorMsg = """
    
        *************************************************************************************
        * No precompiled binary of main KLB library available.                              *
        * Please download main KLB library source code from the link below and build it.    *
        * https://bitbucket.org/fernandoamat/keller-lab-block-filetype/get/%s.zip *
        *************************************************************************************
    
        """ % klbCommitId
    platformName = platform.uname()[0].lower()
    if platform.architecture()[0].startswith("64"):
        if "linux" in platformName:
            downloadFiles.append(( "%s/bin/libklb.so" % klbUrl, libraryDirs[0] ))
        elif "darwin" in platformName:
            downloadFiles.append(( "%s/bin/libklb.dylib" % klbUrl, libraryDirs[0] ))
        elif "win" in platformName:
            downloadFiles.append(( "%s/bin/klb.dll" % klbUrl, libraryDirs[0] ))
            downloadFiles.append(( "%s/bin/klb.lib" % klbUrl, libraryDirs[0] ))
            # fix windows build with msvc
            #downloadFiles.append(( "http://msinttypes.googlecode.com/svn/trunk/stdint.h", includeDirs[0] ))
        else:
            print(errorMsg)
    else:
        print(errorMsg)
    

    # download
    for (source, targetDir) in downloadFiles:
        if not os.path.exists(targetDir):
            os.makedirs(targetDir)
        target = os.path.join(targetDir, os.path.split(source)[1])
        req.urlretrieve( source, target )


if '--skip-klb-download' in sys.argv:
    i = sys.argv.index('--skip-klb-download')
    sys.argv.pop(i)
else:
    # download required KLB headers,
    download_klb_files(includeDirs, libraryDirs)

setup(
    name = "pyklb",
    version = "0.0.3.dev0",
    description = "Python wrapper of the KLB file format, a high-performance file format for up to 5-dimensional arrays.",
    long_description = "See https://bitbucket.org/fernandoamat/keller-lab-block-filetype",
    url = "https://github.com/bhoeckendorf/pyklb",
    ext_modules = cythonize([
        Extension("pyklb", ["src/pyklb.pyx"], include_dirs=includeDirs, library_dirs=libraryDirs, libraries=["klb"])
        ]),
    setup_requires = ["numpy"],
    install_requires = ["numpy"]
)