forpy/setup.py at master · classner/forpy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""The setup script for the forpy project."""
from __future__ import print_function

import platform
import subprocess
import sys
import unittest

try:
    from skbuild import setup
except:  # pylint: disable=bare-except
    print(
        "This package requires `scikit-build` for setup. Trying to install it..."
    )
    try:
        subprocess.check_call(
            [sys.executable, '-m', 'pip', 'install', 'scikit-build'])
    except:  # pylint: disable=bare-except
        print(
            "Automatic installation failed. Please install `scikit-build` and "
        )
        print("rerun the package setup!")
from skbuild import setup  # isort:skip

# Parse the version string.
with open("CMakeLists.txt", 'r') as inf:
    for line in inf:
        if 'forpy_VERSION_MAJOR' in line:
            ver_major = line[line.find("forpy_VERSION_MAJOR") + 20:line.find(
                ")")]
        if 'forpy_VERSION_MINOR' in line:
            ver_minor = line[line.find("forpy_VERSION_MINOR") + 20:line.find(
                ")")]
        if 'forpy_VERSION_PATCH' in line:
            ver_patch = line[line.find("forpy_VERSION_PATCH") + 20:line.find(
                ")")]
            break
VERSION = ver_major + '.' + ver_minor + '.' + ver_patch

try:
    subprocess.check_call(["cmake", "--version"])
except:  # pylint: disable=bare-except
    try:
        subprocess.check_call(
            [sys.executable, '-m', 'pip', 'install', 'cmake'])
    except:  # pylint: disable=bare-except
        print("Automatic installation failed. Please install `cmake` and ")
        print("rerun the package setup!")


def python_test_suite():
    """Discover all python tests."""
    test_loader = unittest.TestLoader()
    test_suite = test_loader.discover('test/python', pattern='*.py')
    return test_suite


if platform.system().lower() == 'linux':
    CORE_LIB_FE = '.so'
    FORPY_LIB_FE = '.so'
elif platform.system().lower() == 'darwin':
    CORE_LIB_FE = '.dylib'
    FORPY_LIB_FE = '.so'
else:
    CORE_LIB_FE = '.dll'
    FORPY_LIB_FE = '.dll'

setup(
    name='forpy',
    author='Christoph Lassner',
    author_email='mail@christophlassner.de',
    url='https://github.com/classner/forpy',
    download_url='https://github.com/classner/forpy/tarball/v{}'.format(
        VERSION),
    keywords='random forests decision machine learning',
    classifiers=[],
    data_files=[('lib/python{}.{}/site-packages'.format(
        sys.version_info[0], sys.version_info[1]), [
            'libforpy_core{}'.format(CORE_LIB_FE),
            'forpy{}'.format(FORPY_LIB_FE)
        ])],
    cmake_source_dir='.',
    cmake_install_dir='',
    cmake_args=[
        '-DWITH_PYTHON=On',
        '-DPYTHON_EXECUTABLE=' + sys.executable,
    ],
    test_suite='setup.python_test_suite',
    setup_requires=['numpy', 'scikit-build'],
    install_requires=['numpy', 'scipy', 'matplotlib', 'sklearn'],
    version=VERSION,
    license='BSD 2-clause')