setup.py

from __future__ import print_function
import distutils.cmd
import setuptools
import os
import sys
import shutil
import subprocess
import platform

with open("README.md", "r") as fh:
    long_description = fh.read()


def check_rust_bitness():
    toolchain = subprocess.Popen(['rustup', 'show', 'active-toolchain'], stdout=subprocess.PIPE).communicate()[0]

    toolchain = str(toolchain)
    python_bitness = platform.architecture()[0]

    if python_bitness == "32bit":
        if not "i686" in toolchain:
            print("WARNING: Python interpreter is 32-bit, while default Rust target toolchain is not 32-bit.\n"
                            "\tUse \"rustup target list\" to list all the available Rust target toolchains.\n"
                            "\tUse \"rustup target add {toolchain_name}\" to include a compatible Rust target toolchain.\n"
                            "\tUse \"rustup default {toolchain_name}\" to set a default target toolchain.\n",
                   file=sys.stderr)

    if python_bitness == "64bit":
        if not "64" in toolchain:
            print("WARNING: Python interpreter is 64-bit, while default Rust target toolchain is not 64-bit.\n"
                            "\tUse \"rustup target list\" to list all the available Rust target toolchains.\n"
                            "\tUse \"rustup target add {toolchain_name}\" to include a compatible Rust target toolchain.\n"
                            "\tUse \"rustup default {toolchain_name}\" to set a default target toolchain.\n",
                   file=sys.stderr)

def build_native(spec):
    check_rust_bitness()

    build = spec.add_external_build(
        cmd=['cargo', 'build', '--release'],
        path='./rust'
    )

    spec.add_cffi_module(
        module_path='kde_ocl._ffi',
        dylib=lambda: build.find_dylib('kde_ocl_sys', in_path='target/release'),
        header_filename=lambda: build.find_header('kde-ocl-sys.h', in_path='target'),
        rtld_flags=['NOW', 'NODELETE']
    )

class CleanCommand(distutils.cmd.Command):
    """
    Our custom command to clean out junk files.
    """
    description = "Cleans out all generated files and folders."
    user_options = []

    def initialize_options(self):
        pass

    def finalize_options(self):
        pass

    def run(self):

        files_to_delete = ["kde_ocl/_ffi.py", "kde_ocl/_ffi__ffi.py", "kde_ocl/_ffi__lib.so", "kde_ocl/_ffi__lib.pyd",
                           "rust/src/open_cl_code.rs"]

        for file in files_to_delete:
            if os.path.exists(file):
                os.remove(file)

        current_dir = os.getcwd()

        os.chdir('rust')
        os.system("cargo clean")
        os.chdir(current_dir)

        folders_to_delete = [".eggs", "kde_ocl.egg-info", "build"]

        for folder in folders_to_delete:
            if os.path.exists(folder):
                shutil.rmtree(folder)

setuptools.setup(
    name="kde_ocl",
    version="0.1.0",
    author="David Atienza",
    author_email="datienza@fi.upm.es",
    description="An OpenCL implementation of Kernel Density Estimation",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/davenza/kde_ocl",
    packages=["kde_ocl"],
    classifiers=[
        "Programming Language :: Python :: 2.7",
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
    ],
    zip_safe=False,
    platforms="any",
    setup_requires=["milksnake"],
    install_requires=["milksnake", "numpy", "six"],
    milksnake_tasks=[
        build_native
    ],
    cmdclass={
        "clean": CleanCommand,
    },
)