Cython如何检测C头文件存在并实现条件编译兼容旧库？

我的库新版本新增了旧版本没有的C函数及对应的mylib.h头文件，头文件内容如下：

#define MYLIB_VERSION "dev"

const char *mylib_version(void) { return MYLIB_VERSION; }

在新版Cython代码中可以这样调用该函数：

cdef extern from "mylib.h":
    const char *mylib_version()

但如果系统安装的是旧版本库，上述代码编译会失败。我尝试用Cython条件编译编写降级逻辑，但代码报错：

cdef extern from *:
    #if __has_include("mylib.h")
        #include "mylib.h"
        #define MYLIB_H_EXISTS 1
    #else
        #define MYLIB_H_EXISTS 0
    #endif

    IF MYLIB_H_EXISTS:
        cdef extern from "mylib.h":
            char *mylib_version()
    ELSE:
        cdef inline char *mylib_version():
            return "0.0.0"

报错信息：

Compile-time name MYLIB_H_EXISTS not defined

请问如何实现Cython条件编译以兼容旧版本库？

报错的核心问题是：Cython的IF指令属于自身编译时逻辑，无法直接读取C预处理器（#define）定义的宏。你在cdef extern from *里通过C预处理器定义的MYLIB_H_EXISTS，Cython编译阶段无法识别，因此提示名称未定义。

方法一：用C预处理器实现兼容逻辑 #

直接借助C预处理器的条件判断能力，将兼容逻辑嵌入C代码块，让C编译器处理分支选择：

cdef extern from *:
    """
    #if __has_include("mylib.h")
        #include "mylib.h"
    #else
        static const char *mylib_version(void) {
            return "0.0.0";
        }
    #endif
    """
    const char *mylib_version()

逻辑说明：

• 用__has_include检测mylib.h是否存在
• 若存在则引入头文件，使用原版函数；若不存在则定义静态替代函数返回默认版本号
• Cython只需统一声明一次函数，无论哪种情况都能匹配到对应实现

方法二：构建阶段传递编译宏给Cython #

如果希望保留Cython层面的条件逻辑，可以在构建脚本中检测头文件存在性，再传递宏给Cython：

1. 修改Cython代码 #

使用Cython的IFDEF指令判断编译宏：

IFDEF HAVE_MYLIB_H:
    cdef extern from "mylib.h":
        const char *mylib_version()
ELSE:
    cdef inline const char *mylib_version():
        return "0.0.0"

2. 修改setup.py #

添加头文件检测逻辑，传递编译宏：

from setuptools import setup, Extension
from Cython.Build import cythonize
import os

def check_mylib_header_exists():
    # 编译测试程序判断头文件是否存在
    try:
        with open("test_mylib.c", "w") as f:
            f.write("#include <mylib.h>\nint main() { return 0; }")
        # 静默编译测试文件
        os.system("cc test_mylib.c -o test_mylib > /dev/null 2>&1")
        exists = os.path.exists("./test_mylib")
        # 清理临时文件
        if exists:
            os.remove("./test_mylib")
        os.remove("test_mylib.c")
        return exists
    except:
        os.remove("test_mylib.c")
        return False

have_mylib_h = check_mylib_header_exists()

extensions = [
    Extension(
        "your_module_name",
        ["your_module.pyx"],
        define_macros=[("HAVE_MYLIB_H", 1)] if have_mylib_h else []
    )
]

setup(
    name="your_package",
    ext_modules=cythonize(extensions)
)

逻辑说明：

• 构建阶段先检测mylib.h是否存在
• 若存在则传递HAVE_MYLIB_H宏给Cython
• Cython编译时通过IFDEF选择对应代码分支

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