Python C扩展编译时如何设置自定义预处理器FLAG

你设置os.environ['FLAG']不生效是因为C编译器不会默认读取系统环境变量作为预处理宏，必须通过编译器参数显式传递宏定义，有两种实现方式：

• 方式1：直接在extra_compile_args中添加编译器参数
  给extra_compile_args列表加-DFLAG参数即可，修改后的Extension定义如下：

  cmodule = Extension('mytest',
                      sources=['src/code.c'],
                      include_dirs=['src'],
                      extra_compile_args=['-fstrict-aliasing', '-O3', '-std=c99', '-Wno-unknown-pragmas', '-D_GNU_SOURCE', '-fPIC', '-DFLAG'],
                      extra_link_args=[],
                      )
  

  注意你原代码中extra_compile_args行末尾缺少逗号，会触发语法错误，上述代码已补全。
• 方式2：使用Extension自带的define_macros参数（更规范，跨平台兼容性更好）
  define_macros是Extension专门用来定义预处理宏的参数，接收元组组成的列表，每个元组第一个元素是宏名，第二个元素是宏的值（如果只需要定义宏不需要赋值，填None即可），示例如下：

  cmodule = Extension('mytest',
                      sources=['src/code.c'],
                      include_dirs=['src'],
                      extra_compile_args=['-fstrict-aliasing', '-O3', '-std=c99', '-Wno-unknown-pragmas', '-D_GNU_SOURCE', '-fPIC'],
                      define_macros=[('FLAG', None)],
                      extra_link_args=[],
                      )
  

如果你需要动态控制宏是否开启，比如根据环境变量判断，可以在setup.py中先读取环境变量再决定是否添加对应配置：

import os
from setuptools import Extension

macros = []
if os.getenv('FLAG') == '1':
    macros.append(('FLAG', None))

cmodule = Extension('mytest',
                    sources=['src/code.c'],
                    include_dirs=['src'],
                    extra_compile_args=['-fstrict-aliasing', '-O3', '-std=c99', '-Wno-unknown-pragmas', '-D_GNU_SOURCE', '-fPIC'],
                    define_macros=macros,
                    extra_link_args=[],
                    )

这样你执行编译前先执行export FLAG=1（Linux/macOS）或set FLAG=1（Windows cmd），再运行python setup.py build_ext就会自动开启FLAG宏定义。

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