在扩展Python的C库中调用json.loads返回Python对象的方法

没问题！我刚好做过类似的Python C扩展开发，把JSON解析逻辑移到C层直接返回字典确实能提升性能，还能简化Python端的代码。下面是具体的实现思路和伪代码，完全基于Python官方的C扩展规范来写：

• 第一步：在C扩展中获取Python标准库的json模块，以及它的loads函数
• 第二步：在你的C函数中生成JSON字符串（和原来get_json_string的逻辑一致）
• 第三步：将C字符串转换成Python的str对象
• 第四步：调用json.loads解析这个字符串，得到Python字典对象
• 第五步：返回这个字典对象，并处理可能的异常（比如JSON格式错误）

#include <Python.h>

// 全局变量存储json模块和loads函数，避免每次调用都重新获取
static PyObject *json_module = NULL;
static PyObject *json_loads = NULL;

// 模块初始化时获取json模块和loads函数
static int my_module_init(PyObject *module) {
    // 导入json模块
    json_module = PyImport_ImportModule("json");
    if (json_module == NULL) {
        return -1; // 自动抛出ImportError异常
    }

    // 获取json.loads函数
    json_loads = PyObject_GetAttrString(json_module, "loads");
    if (json_loads == NULL || !PyCallable_Check(json_loads)) {
        Py_DECREF(json_module);
        json_module = NULL;
        return -1; // 自动抛出AttributeError异常
    }

    return 0;
}

// 你的核心函数：直接返回Python字典
static PyObject *my_module_get_result(PyObject *self, PyObject *args) {
    // 1. 生成JSON字符串（替换成你实际的JSON生成逻辑）
    const char *json_str = generate_your_json_content(); 
    if (json_str == NULL) {
        PyErr_SetString(PyExc_RuntimeError, "Failed to generate JSON content");
        return NULL;
    }

    // 2. 将C字符串转为Python str对象
    PyObject *py_json_str = PyUnicode_FromString(json_str);
    if (py_json_str == NULL) {
        // 若json_str是动态分配的，记得释放内存
        free((void*)json_str);
        return NULL;
    }

    // 3. 调用json.loads解析字符串
    PyObject *result_dict = PyObject_CallFunctionObjArgs(json_loads, py_json_str, NULL);

    // 清理临时对象和资源
    Py_DECREF(py_json_str);
    free((void*)json_str); // 动态分配的C字符串必须释放

    // 4. 检查解析是否成功
    if (result_dict == NULL) {
        // json.loads抛出的异常会自动保留，直接返回NULL即可
        return NULL;
    }

    // 5. 返回解析后的字典对象
    return result_dict;
}

// 模块方法列表
static PyMethodDef MyModuleMethods[] = {
    {"get_result", my_module_get_result, METH_NOARGS, "Return parsed JSON as a Python dictionary"},
    // 可以添加其他方法
    {NULL, NULL, 0, NULL}
};

// 模块定义
static struct PyModuleDef my_module = {
    PyModuleDef_HEAD_INIT,
    "my_module",
    NULL,
    -1,
    MyModuleMethods,
    NULL,
    NULL,
    NULL,
    NULL
};

// 模块初始化入口
PyMODINIT_FUNC PyInit_my_module(void) {
    PyObject *module = PyModule_Create(&my_module);
    if (module == NULL) {
        return NULL;
    }

    // 初始化json模块依赖
    if (my_module_init(module) == -1) {
        Py_DECREF(module);
        return NULL;
    }

    return module;
}

• 资源管理：一定要用Py_DECREF释放不再使用的Python对象，避免内存泄漏；如果你的JSON字符串是动态分配的（比如用malloc），也要记得在合适的时机释放。
• 异常处理：所有Python API调用失败后都会返回NULL并设置异常，你只需要传递这个异常即可，不需要额外处理（除非有特殊业务逻辑）。
• 性能优化：把json_module和json_loads设为全局变量，在模块初始化时只获取一次，避免每次调用get_result都重新导入模块，提升执行效率。

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