ft_utils/setup.py at main · facebookincubator/ft_utils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
# Copyright (c) Meta Platforms, Inc. and affiliates.

import os
import sysconfig

from setuptools import Extension, setup
from setuptools.command.build_ext import build_ext


class BuildExtInplace(build_ext):
    """Custom build_ext to ensure extensions are placed in the correct package directory."""

    def get_ext_fullpath(self, ext_name: str) -> str:
        """Override to ensure extensions go to ft_utils/ directory."""
        fullpath = super().get_ext_fullpath(ext_name)
        # If running with --inplace, adjust the path to put files in ft_utils/ directory
        if self.inplace:
            # Change from ./_concurrency.so to ./ft_utils/_concurrency.so
            filename = os.path.basename(fullpath)
            # Get the package path from the extension name
            parts = ext_name.split(".")
            if parts[0] == "ft_utils":
                # Place in the ft_utils package directory
                fullpath = os.path.join("ft_utils", filename)
        return fullpath


def build_extensions() -> list[Extension]:
    """Build list of Extension objects for C extensions."""
    python_include = sysconfig.get_paths()["include"]
    include_dirs = [
        python_include,
        os.path.join(python_include, "internal"),
        "native/include",
        "native/src",
    ]

    extensions = [
        Extension(
            "ft_utils._concurrency",
            sources=[
                "native/src/_concurrency.c",
                "native/src/ft_core.c",
                "native/src/concurrent_data_structures/concurrent_dict.c",
                "native/src/concurrent_data_structures/atomic_int64.c",
                "native/src/concurrent_data_structures/atomic_reference.c",
                "native/src/concurrent_data_structures/concurrent_deque.c",
            ],
            include_dirs=include_dirs,
        ),
        Extension(
            "ft_utils._weave",
            sources=["native/src/_weave.c"],
            include_dirs=include_dirs,
        ),
        Extension(
            "ft_utils.local",
            sources=["native/src/local.c"],
            include_dirs=include_dirs,
        ),
        Extension(
            "ft_utils.synchronization",
            sources=["native/src/synchronization.c"],
            include_dirs=include_dirs,
        ),
        Extension(
            "ft_utils._test_compat",
            sources=["ft_utils/tests/_test_compat.c"],
            include_dirs=include_dirs,
        ),
        Extension(
            "ft_utils._test_weave",
            sources=["ft_utils/tests/_test_weave.c"],
            include_dirs=include_dirs,
        ),
    ]

    return extensions


def invoke_main() -> None:
    with open("README.md") as readme_file:
        long_descr = readme_file.read()

    setup(
        name="ft_utils",
        version="0.1.0",
        description="A utility library for Free Threaded Python programming",
        long_description=long_descr,
        long_description_content_type="text/markdown",
        author="Meta Platforms, Inc.",
        author_email="open-source@fb.com",
        url="https://github.com/facebookincubator/ft_utils",
        license="MIT",
        packages=["ft_utils", "ft_utils.tests"],
        package_dir={"ft_utils": "ft_utils"},
        ext_modules=build_extensions(),
        cmdclass={"build_ext": BuildExtInplace},
        classifiers=[
            "Development Status :: 4 - Beta",
            "Intended Audience :: Developers",
            "License :: OSI Approved :: MIT License",
            "Programming Language :: Python :: 3.12",
            "Programming Language :: Python :: 3.13",
            "Programming Language :: Python :: 3.14",
        ],
    )


if __name__ == "__main__":
    invoke_main()