108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
import os
|
|
import sys
|
|
import platform
|
|
|
|
from setuptools import setup
|
|
from setuptools.command.test import test as TestCommand
|
|
from setuptools.command.sdist import sdist as SdistCommand
|
|
from setuptools_rust import RustExtension
|
|
|
|
|
|
class PyTest(TestCommand):
|
|
user_options = []
|
|
|
|
def run(self):
|
|
self.run_command("test_rust")
|
|
|
|
import subprocess
|
|
|
|
errno = subprocess.call(["pytest", "tests"])
|
|
raise SystemExit(errno)
|
|
|
|
|
|
class CargoModifiedSdist(SdistCommand):
|
|
"""Modifies Cargo.toml to use an absolute rather than a relative path
|
|
|
|
The current implementation of PEP 517 in pip always does builds in an
|
|
isolated temporary directory. This causes problems with the build, because
|
|
Cargo.toml necessarily refers to the current version of pyo3 by a relative
|
|
path.
|
|
|
|
Since these sdists are never meant to be used for anything other than
|
|
tox / pip installs, at sdist build time, we will modify the Cargo.toml
|
|
in the sdist archive to include an *absolute* path to pyo3.
|
|
"""
|
|
|
|
def make_release_tree(self, base_dir, files):
|
|
"""Stages the files to be included in archives"""
|
|
super().make_release_tree(base_dir, files)
|
|
|
|
import toml
|
|
|
|
# Cargo.toml is now staged and ready to be modified
|
|
cargo_loc = os.path.join(base_dir, "Cargo.toml")
|
|
assert os.path.exists(cargo_loc)
|
|
|
|
with open(cargo_loc, "r") as f:
|
|
cargo_toml = toml.load(f)
|
|
|
|
rel_pyo3_path = cargo_toml["dependencies"]["pyo3"]["path"]
|
|
base_path = os.path.dirname(__file__)
|
|
abs_pyo3_path = os.path.abspath(os.path.join(base_path, rel_pyo3_path))
|
|
|
|
cargo_toml["dependencies"]["pyo3"]["path"] = abs_pyo3_path
|
|
|
|
with open(cargo_loc, "w") as f:
|
|
toml.dump(cargo_toml, f)
|
|
|
|
|
|
def get_py_version_cfgs():
|
|
# For now each Cfg Py_3_X flag is interpreted as "at least 3.X"
|
|
version = sys.version_info[0:2]
|
|
py3_min = 5
|
|
out_cfg = []
|
|
for minor in range(py3_min, version[1] + 1):
|
|
out_cfg.append("--cfg=Py_3_%d" % minor)
|
|
|
|
if platform.python_implementation() == "PyPy":
|
|
out_cfg.append("--cfg=PyPy")
|
|
|
|
return out_cfg
|
|
|
|
|
|
def make_rust_extension(module_name):
|
|
return RustExtension(
|
|
module_name, "Cargo.toml", rustc_flags=get_py_version_cfgs(), debug=True
|
|
)
|
|
|
|
|
|
install_requires = []
|
|
tests_require = install_requires + ["pytest", "pytest-benchmark"]
|
|
|
|
setup(
|
|
name="rustapi-module",
|
|
version="0.1.0",
|
|
classifiers=[
|
|
"License :: OSI Approved :: MIT License",
|
|
"Development Status :: 3 - Alpha",
|
|
"Intended Audience :: Developers",
|
|
"Programming Language :: Python",
|
|
"Programming Language :: Rust",
|
|
"Operating System :: POSIX",
|
|
"Operating System :: MacOS :: MacOS X",
|
|
],
|
|
packages=["rustapi_module"],
|
|
rust_extensions=[
|
|
make_rust_extension("rustapi_module.othermod"),
|
|
make_rust_extension("rustapi_module.datetime"),
|
|
make_rust_extension("rustapi_module.subclassing"),
|
|
make_rust_extension("rustapi_module.test_dict"),
|
|
make_rust_extension("rustapi_module.buf_and_str"),
|
|
],
|
|
install_requires=install_requires,
|
|
tests_require=tests_require,
|
|
include_package_data=True,
|
|
zip_safe=False,
|
|
cmdclass={"test": PyTest, "sdist": CargoModifiedSdist},
|
|
)
|