setup.py

import contextlib
from contextlib import ExitStack
import os
from pathlib import Path
import shutil
import subprocess
import sys

try:
    from setuptools import setup
except ImportError:
    sys.exit("Please install setuptools by following the instructions at\n"
             "    https://pypi.python.org/pypi/setuptools")

from setuptools import Extension
from setuptools.command.build_ext import build_ext


if sys.platform == "win32":
    PACKAGE_DATA = ["dds-32.dll", "dds-64.dll"]
else:
    # On a POSIX system, libdds.so will be moved to its correct location by
    # make_build.
    PACKAGE_DATA = []


@contextlib.contextmanager
def patched_path(path, old, new):
    contents = path.read_text()
    if old not in contents:
        raise Exception(f"Invalid patch: {old}")
    try:
        path.write_text(contents.replace(old, new))
        yield
    finally:
        path.write_text(contents)


patches = [
    ("Makefiles/Makefile_linux_shared",
     "$(COMPILE_FLAGS)", "$(COMPILE_FLAGS) $(CFLAGS)"),
    ("System.cpp",  # redeal issue 19.
     "free -k | tail -n+3 | head -n1 | awk '{print $NF}'",
     r"grep -Po 'MemAvailable:\\s*\\K[0-9]*' /proc/meminfo || "
     r"grep -Po 'MemFree:\\s*\\K[0-9]*' /proc/meminfo"),
    ("dds.cpp",  # dds issue #91.
     "FreeMemory();", ""),
    ("Makefiles/Makefile_Mac_clang_shared",
     "$(COMPILE_FLAGS)", "$(COMPILE_FLAGS) $(CFLAGS)"),
    ("Makefiles/Makefile_Mac_clang_shared",
     "$(LINK_FLAGS)", "$(LINK_FLAGS) -lc++"),
]


class build_ext(build_ext):
    def finalize_options(self):
        super().finalize_options()
        # Needs to be computed here because setuptools patches out inplace.
        self.__dest_dir = Path(self.get_ext_fullpath("redeal._")).parent

    def build_extensions(self):
        self.distribution.ext_modules[:] = []
        super().build_extensions()
        if os.name == "posix":
            dds_src = Path(__file__).resolve().parent / "dds/src"
            if not dds_src.exists():
                sys.exit("""\
DDS sources are missing.

If you are using a git checkout, run
    git submodule init && git submodule update

On a Unix system, do not use the zip archives from github.""")
            with ExitStack() as stack:
                for name, old, new in patches:
                    stack.enter_context(patched_path(dds_src / name, old, new))
                if sys.platform.startswith("linux"):
                    subprocess.check_call(
                        ["make", "-f", "Makefiles/Makefile_linux_shared",
                         "THREADING=", "THREAD_LINK="], cwd=dds_src)
                elif sys.platform == "darwin":
                    subprocess.check_call(
                        ["make", "-f", "Makefiles/Makefile_Mac_clang_shared",
                         "CC=gcc", "THREADING=", "THREAD_LINK="], cwd=dds_src)
            shutil.copy2(dds_src / "libdds.so", self.__dest_dir)


setup(
    cmdclass={"build_ext": build_ext},
    name="redeal",
    version="0.2.0",
    author="Antony Lee",
    author_email="anntzer.lee@gmail.com",
    packages=["redeal"],
    package_data={"redeal": PACKAGE_DATA},
    entry_points={
        "console_scripts": ["redeal = redeal.__main__:console_entry"],
        "gui_scripts": ["redeal-gui = redeal.__main__:gui_entry"],
    },
    url="http://github.com/anntzer/redeal",
    license="LICENSE.txt",
    description="A reimplementation of Thomas Andrews' Deal in Python.",
    long_description=Path("README.rst").read_text(encoding="utf-8"),
    python_requires=">=3.6",
    install_requires=["colorama>=0.2.4"],
    ext_modules=[Extension("", [])]
)