C语言编写Python扩展时如何检测AST对象的PyType？

API 探路者

2026-10-3

问题原因

PyAST_Check 不属于CPython对外公开的C API，仅为解释器内部使用的接口，不会被导出到libpython的符号表中，因此外部C扩展在链接时无法找到该符号，会触发你遇到的导入错误。

可行解决方法

最稳定且跨版本兼容的方案是通过公共C API调用Python层面的ast.AST基类做实例检测，具体实现如下：

实现步骤

在模块初始化阶段预先获取ast模块的AST基类对象，缓存为全局静态变量，避免每次调用重复导入产生性能损耗
在检测函数中使用PyObject_IsInstance公共API判断传入对象是否为AST的子类实例

完整代码示例

#include <Python.h>

// 缓存ast.AST类型对象的全局静态变量
static PyObject *Ast_Type = NULL;

static PyObject *python_is_ast(PyObject *Py_UNUSED(self), PyObject *obj) {
    // 用公共API做类型检测
    int ret = PyObject_IsInstance(obj, Ast_Type);
    if (ret < 0) {
        // 检测过程抛出异常的话返回NULL
        return NULL;
    }
    return PyBool_FromLong(ret);
}

// 方法表
static PyMethodDef module_methods[] = {
    {"is_ast", python_is_ast, METH_O, "Check if an object is a Python AST node"},
    {NULL, NULL, 0, NULL} // 哨兵
};

// 模块定义
static struct PyModuleDef module_def = {
    PyModuleDef_HEAD_INIT,
    "_module",
    NULL,
    -1,
    module_methods
};

// 模块初始化函数
PyMODINIT_FUNC PyInit__module(void) {
    PyObject *module = PyModule_Create(&module_def);
    if (!module) return NULL;

    // 导入ast模块并获取AST基类
    PyObject *ast_module = PyImport_ImportModule("ast");
    if (!ast_module) {
        Py_DECREF(module);
        return NULL;
    }
    Ast_Type = PyObject_GetAttrString(ast_module, "AST");
    Py_DECREF(ast_module);
    if (!Ast_Type) {
        Py_DECREF(module);
        return NULL;
    }

    return module;
}