如何在pyproject.toml中为Python C扩展配置动态参数？

因为setuptools的pyproject.toml动态元数据确实不支持ext-modules字段，要实现你需要的主机适配和环境变量驱动的配置，推荐以下两种可行方案：

方案1：保留精简版setup.py处理扩展配置 #

不需要完全抛弃setup.py，pyproject.toml指定setuptools为构建后端时，依然会读取setup.py中的ext_modules配置。你可以把版本、作者等静态配置放在pyproject.toml里，仅在setup.py中处理C扩展的动态生成：

步骤： #

1. 在pyproject.toml中保留基础配置，指定构建后端：

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "your-package"
version = "0.1.0"
# 其他静态元数据...

2. 创建/保留setup.py，在其中动态生成扩展配置：

from setuptools import setup, Extension
import os
import sysconfig

def get_extension():
    # 适配构建主机的include-dirs和library-dirs
    include_dirs = [sysconfig.get_path("include")]
    # 按需添加平台专属路径，比如Linux/Windows/macOS的差异
    if sysconfig.get_platform().startswith("linux"):
        include_dirs.append("/usr/local/include")
    
    library_dirs = [sysconfig.get_path("lib")]
    if sysconfig.get_platform().startswith("linux"):
        library_dirs.append("/usr/local/lib")
    
    # 根据DEBUG环境变量调整libraries和编译参数
    libraries = []
    extra_compile_args = []
    if os.environ.get("DEBUG") in ("1", "true", "yes"):
        libraries.append("m")  # 示例：添加数学库
        extra_compile_args.extend(["-g", "-O0", "-Wall"])
    else:
        extra_compile_args.extend(["-O2"])
    
    return Extension(
        "your_package.extension_module",
        sources=["src/extension.c"],
        include_dirs=include_dirs,
        library_dirs=library_dirs,
        libraries=libraries,
        extra_compile_args=extra_compile_args
    )

setup(
    ext_modules=[get_extension()]
)

这种方式兼容性最好，不需要额外依赖，完全利用setuptools的现有机制。

方案2：使用setuptools构建钩子（Build Hooks） #

如果你想尽可能减少setup.py的代码，可以使用setuptools的build.py钩子脚本，在构建阶段动态生成扩展配置：

步骤： #

1. 在pyproject.toml中添加钩子配置：

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
# 静态元数据...

[tool.setuptools.build]
scripts = ["build.py"]

2. 创建build.py脚本，实现build_extensions钩子：

from setuptools import Extension
import os
import sysconfig

def build_extensions(ext):
    # 替换原扩展列表为动态生成的配置
    include_dirs = [sysconfig.get_path("include")]
    if sysconfig.get_platform().startswith("linux"):
        include_dirs.append("/usr/local/include")
    
    library_dirs = [sysconfig.get_path("lib")]
    if sysconfig.get_platform().startswith("linux"):
        library_dirs.append("/usr/local/lib")
    
    libraries = []
    extra_compile_args = []
    if os.environ.get("DEBUG") in ("1", "true", "yes"):
        libraries.append("m")
        extra_compile_args.extend(["-g", "-O0", "-Wall"])
    else:
        extra_compile_args.extend(["-O2"])
    
    new_ext = [Extension(
        "your_package.extension_module",
        sources=["src/extension.c"],
        include_dirs=include_dirs,
        library_dirs=library_dirs,
        libraries=libraries,
        extra_compile_args=extra_compile_args
    )]
    ext.extensions = new_ext

这种方式更符合现代pyproject.toml的配置风格，把动态逻辑分离到专门的构建脚本中。

注意事项 #

• 要在不同平台（Linux、Windows、macOS）上测试主机适配逻辑，注意sysconfig.get_platform()返回字符串的差异。
• 可以扩展DEBUG环境变量的判断逻辑，支持多种输入格式（比如DEBUG=true或DEBUG=1）。
• 复杂平台检测可以结合platform模块辅助实现。

内容的提问来源于stack exchange，提问作者user3758232