第二次调用时无法从python模块获取函数

时间:2013-04-02 07:25:07

标签: python python-c-api

下面是c ++代码,它将调用我的python脚本' rule.py'在同一目录中。

#include <Python.h>
#include <iostream>
using namespace std;

int dou(int a)
{
    PyObject *pModule, *pDict, *pFunc, *pArgs, *pRetVal;
    if( !Py_IsInitialized() ){
        cout<<"Can't initialize"<<endl;
        return -1;
    }
    PyRun_SimpleString("import sys");
    PyRun_SimpleString("sys.path.append('.')");
    pModule = PyImport_ImportModule("rule");
    if( !pModule ){
        cout<<"can' import the module "<<endl;
        return -1;
    }
    pDict = PyModule_GetDict(pModule);
    if( !pDict ){
        cout<<"can't get the dict"<<endl;
        return -1;
    }
    pFunc = PyDict_GetItemString(pDict, "fun");
    if( !pFunc || !PyCallable_Check(pFunc) ){
        cout<<"can't get the function"<<endl;
        return -1;
    }

    pArgs = PyTuple_New(1);
    PyTuple_SetItem(pArgs,0,Py_BuildValue("i",a));
    pRetVal = PyObject_CallObject(pFunc,pArgs);

    int result = PyInt_AsLong(pRetVal);

    Py_DECREF(pModule);
    Py_DECREF(pDict);
    Py_DECREF(pFunc);
    Py_DECREF(pArgs);
    Py_DECREF(pRetVal);

    return result;
}
int main(void)
{
    Py_Initialize();
    cout<<dou(2)<<endl;
    cout<<dou(3)<<endl;
    Py_Finalize();
    return 0;
}

这是简单的python代码。

def fun(a):
    return 2*a
if __name__ == "__main__":
    print fun(2)

输出结果为:

4
can't get the function.
-1

它在第一时间工作,但在第二次调用它时无法获得该功能。也许我错过了什么?

1 个答案:

答案 0 :(得分:2)

PyModule_GetDict()返回借用的引用。这意味着您不得减少其引用计数,否则模块dict将被破坏。

您的代码片段中存在许多错误。我想你可以试试boost :: python。并仔细阅读Python手册,特别是在引用计数管理中。