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
|
#!/usr/bin/env python
# Licensed under a 3-clause BSD style license - see LICENSE.rst
# NOTE: The configuration for the package, including the name, version, and
# other information are set in the pyproject.toml file.
import sys
from pathlib import Path
from setuptools import setup
from setuptools.command.editable_wheel import editable_wheel
from setuptools.command.install import install
from extension_helpers import get_extensions
ext_modules = get_extensions()
def install_stubs(build_lib, output_dir):
sys.path.insert(0, build_lib)
try:
from astropy.units.typing_utils import create_stubs # noqa: PLC0415
create_stubs(Path(output_dir))
finally:
# Undo the path modification.
sys.path.pop(0)
class InstallWithStubs(install):
"""Post-installation command for installation mode."""
def run(self):
super().run()
install_stubs(self.build_lib, self.root)
class EditableInstallWithStubs(editable_wheel):
"""Post-installation command for editable_wheel mode."""
def run(self):
super().run()
install_stubs(self.project_dir, self.project_dir)
# Specify the minimum version for the Numpy C-API
for ext in ext_modules:
if ext.include_dirs and "numpy" in ext.include_dirs[0]:
ext.define_macros.append(("NPY_TARGET_VERSION", "NPY_1_24_API_VERSION"))
ext.define_macros.append(("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION"))
setup(
ext_modules=ext_modules,
cmdclass={
"install": InstallWithStubs,
"editable_wheel": EditableInstallWithStubs,
},
)
|
