如何在setup.py中根据操作系统条件切换setuptools编译器？

核心思路 #

通过检测当前操作系统，在Windows环境下动态修改setuptools的编译器配置，同时配合CFFI的构建逻辑完成编译，无需改动原有C代码。

具体实现步骤 #

1. 导入依赖模块
   导入os、sys、setuptools的构建类以及CFFI的FFI类，为后续逻辑做准备。

2. 自定义构建类替换编译器
   继承setuptools的build_ext类，重写build_extensions方法，在Windows系统下强制指定编译器为mingw32，还可按需添加编译参数。

3. 集成CFFI绑定逻辑
   正常编写CFFI的函数声明与绑定代码，将构建逻辑与自定义编译类关联。

完整示例代码 #

import sys
from setuptools import setup
from setuptools.command.build_ext import build_ext
from cffi import FFI

# 自定义编译类，实现Windows下的编译器切换
class MinGWBuildExt(build_ext):
    def build_extensions(self):
        if sys.platform.startswith('win32'):
            # 强制设置Windows使用mingw32编译器
            self.compiler = 'mingw32'
            # 添加C99支持参数（VLA是C99特性，MinGW需显式开启）
            for ext in self.extensions:
                ext.extra_compile_args.append('-std=c99')
        super().build_extensions()

# CFFI绑定逻辑
ffibuilder = FFI()
# 替换为你的C函数声明
ffibuilder.cdef("""
    void vla_based_function(int array_size);
""")

# 绑定C代码或外部源文件
ffibuilder.set_source("_your_cffi_module",
    """
    #include "your_c_code.h"
    """,
    sources=['your_c_code.c']
)

setup(
    name="your-package-name",
    version="0.1.0",
    ext_modules=[ffibuilder.distutils_extension()],
    cmdclass={'build_ext': MinGWBuildExt},
    install_requires=['cffi>=1.0.0'],
)

关键说明 #

• 编译器切换逻辑：仅在Windows系统触发mingw32编译器设置，Linux下仍保留默认GCC编译流程，不影响跨平台兼容性。
• C99支持：MinGW默认可能未开启C99标准，添加-std=c99参数可确保VLA相关代码正常编译。
• 环境要求：Windows环境需提前安装MinGW工具链，并将其路径添加到系统PATH变量中，否则setuptools无法找到编译器。

额外注意事项 #

• 上传PyPI时，建议使用build工具配合wheel构建跨平台包，Windows下需确保MinGW环境配置正确，生成兼容的wheel文件。

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