setup.py

#!/usr/bin/env python3
from setuptools import setup, Extension
from setuptools.command.build_clib import build_clib
from setuptools.command.build_py import build_py
import os
import sys
import platform
import shutil
import os.path
import extract_icsneo40_defines
import generate_icsneo40_structs
from ics_utility import get_pkg_version, create_version_py, create_ics_init
import pathlib
from typing import List

# I had this in build_py but there are certain times its not called, for now lets force
# it to run every single time we call setup. Its quick enough where it shouldn't hurt.
create_ics_init()
create_version_py()
generate_icsneo40_structs.generate_all_files()

# Check for clang stuff here, read the docs doesn't have this so use what is in the repo
if not os.getenv("READTHEDOCS"):
    if not shutil.which("clang"):
        raise RuntimeError("clang is required for building python_ics.")
    if not shutil.which("clang-format"):
        raise RuntimeError("clang-format is required for building python_ics.")

def get_version():
    major = int(get_pkg_version().split(".")[0])
    minor = int(get_pkg_version().split(".")[1])
    return major, minor

class _build_libicsneo(build_clib):
    def run(self):
        # checkout and build libicsneo
        if platform.system().upper() in ("DARWIN", "LINUX"):
            import build_libicsneo
            
            print("Checking out libicsneo...")
            build_libicsneo.checkout()
            print("Building libicsneo...")
            build_libicsneo.build()
            print("Copying libicsneo...")
            build_libicsneo.copy()
        super().run()


class _build_ics_py(build_py):
    def run(self):
        print("Generating python defines from icsnVC40.h...")
        extract_icsneo40_defines.extract()

        print("Copy python source files over to gen...")
        src_path = pathlib.Path("./src/")
        gen_path = pathlib.Path("./gen/")
        for dirpath, dirnames, filenames in os.walk(src_path):
            dirpath = pathlib.Path(dirpath)
            for name in filenames:
                if pathlib.Path(name).suffix == ".py" and 'icsdebug' not in name:
                    src = (dirpath / name)
                    dest = (gen_path / dirpath.relative_to(src_path) / name)
                    print(f"Copying {src} to {dest}")
                    shutil.copy(src, dest)
        self.run_command("build_libicsneo")
        super().run()


def get_ics_extension_compiler_arguments() -> List[str]:
    """Return a list of compiler arguments to append for compiling the ics extension module"""
    # gcc and clang arguments
    GCC_COMPILE_ARGS = [
        "-fpermissive",
        "-Wno-unused-variable",
        "-Wno-unused-function",
        "-Wno-write-strings",
    ]
    # Set compiler flags here
    if "WINDOWS" in platform.system().upper():
        compile_args = []
        # mingw and clang python builds won't have MSC in the version string
        if "MSC" not in sys.version:
            compile_args = GCC_COMPILE_ARGS
    elif "LINUX" in platform.system().upper():
        compile_args = GCC_COMPILE_ARGS
    elif "DARWIN" in platform.system().upper():
        # Mac doesn't respect the compiler args, but will append with ARCHFLAGS environment variable
        os.environ["ARCHFLAGS"] += " -std=c++17"
        compile_args = []
    else:
        compile_args = []
    return compile_args

ics_extension = Extension(
    "ics.ics",
    define_macros=[("MAJOR_VERSION", get_version()[0]), ("MINOR_VERSION", get_version()[1]), ("EXTERNAL_PROJECT", 1)],
    include_dirs=["include", "include/ics", "src/ice/include", "src/ice/include/ice"],
    libraries=[],
    library_dirs=["/usr/local/lib"],
    sources=[
        "src/object_spy_message.cpp",
        "src/defines.cpp",
        "src/exceptions.cpp",
        "src/dll.cpp",
        "src/setup_module_auto_defines.cpp",
        "src/main.cpp",
        "src/methods.cpp",
        "src/ice/src/ice_library_manager.cpp",
        "src/ice/src/ice_library_name.cpp",
        "src/ice/src/ice_library.cpp",
    ],
    extra_compile_args=get_ics_extension_compiler_arguments(),
)

package_data = {}
if "DARWIN" in platform.system().upper():
    package_data["ics"] = ["*.dylib"]
elif "LINUX" in platform.system().upper():
    package_data[""] = ["*.so"]

setup(
    name="python_ics",
    version=get_pkg_version(),
    cmdclass={
        "build_libicsneo": _build_libicsneo,
        "build_py": _build_ics_py,
    },
    download_url="https://github.com/intrepidcs/python_ics/releases",
    packages=["ics", "ics.structures"],
    package_dir={"ics": "gen/ics", "ics.structures": "gen/ics/structures"},
    package_data=package_data,
    include_package_data=True,
    ext_modules=[ics_extension],
)