setup.py

import os
import typing

import setuptools
from pipenv.project import Project
from pipenv.utils import convert_deps_to_pip
from setuptools.command.sdist import sdist as SdistCommand
from setuptools.command.test import test as TestCommand
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):
        """Stage 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):
    """Handle running tests during setup."""

    # user_options = []

    def run(self):
        """Run tests."""
        self.run_command("test_rust")
        import subprocess

        subprocess.check_call(["pytest", "tests"])


setup_requires = ["setuptools-rust>=0.10.1", "wheel"]
install_requires: typing.List[str] = []
# tests_require = install_requires + ["pytest", "pytest-benchmark"]

pfile = Project(chdir=False).parsed_pipfile
# install_requires = convert_deps_to_pip(pfile["packages"], r=False)
tests_require = convert_deps_to_pip(pfile["dev-packages"], r=False)


setuptools.setup(
    name="pip-crev",
    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=["pip_crev", "crev"],
    rust_extensions=[
        RustExtension("crev.py_crev_lib", "py-crev-lib/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},
    entry_points={
        "console_scripts": ["pip-crev=pip_crev.pip_crev:parse_command_line_arguments"]
    },
)