扩展C ++函数的奇怪行为

时间:2015-09-12 19:57:31

标签: python c++ extending

我正在为C ++函数repeating_count::count(string str)编写Python包装器。此函数的算法并不重要,但始终返回int。 我的包装器获取list[str]作为输入并返回list[int]。我的包装器的代码如下:

PyObject* count_rep_list(PyObject *mod, PyObject *args){
    PyObject* inputList = PyList_GetItem(args, 0);
    PyObject* outputList = PyList_New(0);
    cout << PyList_Size(inputList);
    char* str;
    for(size_t i = 0; i < PyList_Size(inputList); ++i){
        PyObject* list_item = PyList_GetItem(inputList, i);
        if(!PyArg_Parse(list_item, "s", &str)){
            return NULL;
        }
        PyList_Append(outputList, PyLong_FromSize_t(repeating_count::count(string(str))));
    }
    return outputList;
}

然后我用这个函数创建了一个python模块:

PyMODINIT_FUNC PyInit_repeating_count() {
    static PyMethodDef ModuleMethods[] = {
            {"count_rep_list", count_rep_list, METH_VARARGS, "counting repeatings for list of strings"},
            { NULL, NULL, 0, NULL}
    };
    static PyModuleDef ModuleDef = {
            PyModuleDef_HEAD_INIT,
            "repeating_count",
            "counting repeatings",
            -1, ModuleMethods,
            NULL, NULL, NULL, NULL
    };
    PyObject * module = PyModule_Create(&ModuleDef);
    return module;
}

我在.so文件中成功编译并链接了此模块。但是当我想从Python 3调用该函数时,我会抓住

  

分段错误(核心转储)

那么,我做错了什么?

用于调用count()的Python代码片段:

from repeating_count import *
print(count_rep_list(["sdfsf", "sf", "sdfvgsdgsd"]))

1 个答案:

答案 0 :(得分:0)

我解决了我的问题。代码中只有一个错误。

我必须使用PyTuble_GetItem(args, 0)代替PyList_GetItem(args, 0) beacuse args是一个元组。