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
95
96
97
98
99
100
101
|
#!/usr/bin/env python
import os
import platform
import re
import sys
from pathlib import Path
from setuptools import Extension
from setuptools import find_namespace_packages
from setuptools import setup
from setuptools.command.build_ext import build_ext
from setuptools.dist import Distribution
# Enable code coverage for C code: we cannot use CFLAGS=-coverage in tox.ini, since that may mess with compiling
# dependencies (e.g. numpy). Therefore, we set SETUPPY_CFLAGS=-coverage in tox.ini and copy it to CFLAGS here (after
# deps have been safely installed).
if 'TOX_ENV_NAME' in os.environ and os.environ.get('SETUPPY_EXT_COVERAGE') == 'yes' and platform.system() == 'Linux':
CFLAGS = os.environ['CFLAGS'] = '-fprofile-arcs -ftest-coverage'
LFLAGS = os.environ['LFLAGS'] = '-lgcov'
else:
CFLAGS = ''
LFLAGS = ''
allow_extensions = True
if sys.implementation.name in ('pypy', 'graalpy'):
print('NOTICE: C extensions disabled on PyPy/GraalPy (would be broken)!')
allow_extensions = False
if os.environ.get('SETUPPY_FORCE_PURE'):
print('NOTICE: C extensions disabled (SETUPPY_FORCE_PURE)!')
allow_extensions = False
class OptionalBuildExt(build_ext):
"""
Allow the building of C extensions to fail.
"""
def run(self):
try:
super().run()
except Exception as e:
self._unavailable(e)
self.extensions = [] # avoid copying missing files (it would fail).
def _unavailable(self, e):
print('*' * 80)
print(
"""WARNING:
An optional code optimization (C extension) could not be compiled.
Optimizations for this package will not be available!
"""
)
print('CAUSE:')
print('')
print(' ' + repr(e))
print('*' * 80)
class BinaryDistribution(Distribution):
"""
Distribution which almost always forces a binary package with platform name
"""
def has_ext_modules(self):
return super().has_ext_modules() or not os.environ.get('SETUPPY_ALLOW_PURE')
def read(*names, **kwargs):
with Path(__file__).parent.joinpath(*names).open(encoding=kwargs.get('encoding', 'utf8')) as fh:
return fh.read()
setup(
long_description='{}\n{}'.format(
re.compile('^.. start-badges.*^.. end-badges', re.M | re.S).sub('', read('README.rst')),
re.sub(':[a-z]+:`~?(.*?)`', r'``\1``', read('CHANGELOG.rst')),
),
long_description_content_type='text/x-rst',
packages=find_namespace_packages('src'),
package_dir={'': 'src'},
py_modules=[path.stem for path in Path('src').glob('*.py')],
include_package_data=True,
zip_safe=False,
cmdclass={'build_ext': OptionalBuildExt},
ext_modules=[
Extension(
str(path.relative_to('src').with_suffix('')).replace(os.sep, '.'),
sources=[str(path)],
extra_compile_args=CFLAGS.split(),
extra_link_args=LFLAGS.split(),
include_dirs=[str(path.parent)],
)
for path in Path('src').glob('**/*.c')
]
if allow_extensions
else [],
distclass=BinaryDistribution if allow_extensions else None,
)
|
