-
Notifications
You must be signed in to change notification settings - Fork 36
Expand file tree
/
Copy pathsetup.py
More file actions
93 lines (73 loc) · 2.62 KB
/
Copy pathsetup.py
File metadata and controls
93 lines (73 loc) · 2.62 KB
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
from pathlib import Path
from setuptools import setup, find_packages
from setuptools.command.develop import develop
from setuptools.command.install import install
import cffi
import tomli
from typing import Dict, Any
from setuptools.dist import Distribution
project_dir = Path(__file__).parent
def load_configuration(file_path: Path) -> Dict[str, Any]:
"""Load and parse the configuration from a given TOML file."""
return tomli.loads(file_path.read_text())
def build_cffi():
"""
Build the CFFI modules as per the configuration in `pyproject.toml`.
Reads the configuration from `tool.cffi_builder` section.
"""
config = load_configuration(project_dir / "pyproject.toml")
cffi_builder_config = config.get("tool", {}).get("cffi_builder", {})
sources = [Path(s) for s in cffi_builder_config.get("sources", [])]
headers = [Path(s) for s in cffi_builder_config.get("headers", [])]
include_dir = cffi_builder_config.get("include_dir", ".")
flags = cffi_builder_config.get("flags", [])
source_content = '\n'.join(s.read_text() for s in sources)
header_content = '\n'.join(h.read_text() for h in headers)
ffibuilder = cffi.FFI()
ffibuilder.cdef(header_content)
ffibuilder.set_source(
cffi_builder_config.get("module_name", "tmu.tmulib"),
source_content,
include_dirs=[Path(include_dir).absolute()],
extra_compile_args=flags
)
ffibuilder.compile(verbose=True)
return ffibuilder
class TMUInstall(install):
"""
Custom install command that builds the CFFI modules
before proceeding with the standard installation process.
"""
def run(self):
build_cffi()
super().run()
class TMUDevelop(develop):
"""
Custom develop command that builds the CFFI modules
before proceeding with the standard develop process.
"""
def run(self):
build_cffi()
super().run()
class BinaryDistribution(Distribution):
"""Distribution which always forces a binary package with platform name"""
def has_ext_modules(foo):
return True
setup(
include_package_data=True,
packages=find_packages(),
ext_modules=[build_cffi().distutils_extension()],
classifiers=[
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Development Status :: 5 - Production/Stable",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
]
#cmdclass={
# "install": TMUInstall,
# "develop": TMUDevelop,
#},
#distclass=BinaryDistribution
)