95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
import os
|
|
import sys
|
|
|
|
from setuptools import setup
|
|
from setuptools.command.test import test as TestCommand
|
|
from setuptools.command.sdist import sdist as SdistCommand
|
|
|
|
try:
|
|
from setuptools_rust import RustExtension
|
|
except ImportError:
|
|
import subprocess
|
|
|
|
errno = subprocess.call([sys.executable, "-m", "pip", "install", "setuptools-rust"])
|
|
if errno:
|
|
print("Please install setuptools-rust package")
|
|
raise SystemExit(errno)
|
|
else:
|
|
from setuptools_rust import RustExtension
|
|
|
|
|
|
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)
|
|
|
|
|
|
class PyTest(TestCommand):
|
|
user_options = []
|
|
|
|
def run(self):
|
|
self.run_command("test_rust")
|
|
|
|
import subprocess
|
|
|
|
subprocess.check_call(["pytest", "tests"])
|
|
|
|
|
|
setup_requires = ["setuptools-rust>=0.10.1", "wheel"]
|
|
install_requires = []
|
|
tests_require = install_requires + ["pytest", "pytest-benchmark"]
|
|
|
|
setup(
|
|
name="word-count",
|
|
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=["word_count"],
|
|
rust_extensions=[RustExtension("word_count.word_count", "Cargo.toml", debug=True)],
|
|
install_requires=install_requires,
|
|
tests_require=tests_require,
|
|
setup_requires=setup_requires,
|
|
include_package_data=True,
|
|
zip_safe=False,
|
|
cmdclass={
|
|
'test': PyTest,
|
|
'sdist': CargoModifiedSdist,
|
|
},
|
|
)
|