使用C++擴展Python的功能詳解

//-------------------------------------------------------------------------
// 函數    : initPyExt
// 功能    : 初始化函數
// 返回值   :PyMODINIT_FUNC
// 附注    : 注意，這個函數的名字不能改動。必須是init+模塊名字，
// 我們的模塊名字是PyExt，所以函數名是initPyExt。Python在導入
// 我們的PyExt模塊時，會找到這個函數，并調用。這個函數實現的
// 功能很簡單，通過調用Py_InitModule將模塊名字和映射表結合起
// 來，它的意思是說PyExt這個模塊使用PyExtMethods這個映射表。
//-------------------------------------------------------------------------
PyMODINIT_FUNCinitPyExt()
{
    Py_InitModule("PyExt",PyExtMethods);
}

/*
   方法列表，這個是一個C結構數組。把需要擴展的函數都映射到這個表里。
   那么Python就知道你的這個擴展模塊支持一些什么方法了。表的第一個字
   段是方法名字，也是通過Python來調用時的名字。第二個字段是導出函數，
   是真正調用的函數，也是C\C++實現的函數。第三個參數是指明Python向
   C\C++函數傳遞參數的形式?？蛇x的兩種方式是METH_VARARGS和
   METH_KEYWORDS，其中METH_VARARGS是參數傳遞的標準形式，它通
   過Python的元組在Python解釋器和C函數之間傳遞參數，若采用
   METH_KEYWORD方式，則Python解釋器和C函數之間將通過Python的字典
   類型在兩者之間進行參數傳遞。第四個字段是這個函數的說明。如果你在
   python里來help這個函數，將顯示這個說明。相當于在python里的函數的文檔說明。
*/
staticPyMethodDefPyExtMethods[]=
{
    {"Add", Add,METH_VARARGS,"Addtwo number - edit by magictong."},
    {"ExecSystem",ExecSystem,METH_VARARGS,"Execute a shell command - edit bymagictong." },
    {NULL,NULL, 0,NULL}
};

//-------------------------------------------------------------------------
// 函數    : Add
// 功能    : 這是一個加法函數
// 返回值   :PyObject*
// 參數    : PyObject*self 這個參數我們暫時不用理會
// 參數    : PyObject*args 是一個參數列表，我們需要從它解析出參數
// 附注    :
// 所有的導出函數都具有相同的原型：
// PyObject*method(PyObject* self, PyObject* args);
//PyArg_ParseTuple來完成解析參數任務。它的第一個參數是args，
// 就是我們要轉換的參數。第二個是格式符號。"s"代表是個string。
// 從args里提取一個參數就寫"s"，兩個的話就寫"s|s"，如果是一個
// string，一個int，就寫"s|i"，有點和printf類似哦。第三個參數就是
// 提取出來的參數放置的真正位置。必須傳遞這個參數的地址。
//-------------------------------------------------------------------------
staticPyObject*Add(PyObject*self,PyObject*args)
{
    intx = 0 ;
    inty = 0;
    intz = 0;
    if(!PyArg_ParseTuple(args,"i|i", &x, &y))
       returnNULL;
    z=x +y;
    returnPy_BuildValue("i",z);
    /*
      調用完之后我們需要返回結果。這個結果是c的type或者是我們自己定義的類型。
      必須把他轉換成PyObject，讓python認識。這個用Py_BuildValue來完成。他
      是PyArg_ParseTuple的逆過程。他的第一個參數和PyArg_ParseTuple的第二個
      參數一樣，是個格式化符號。第三個參數是我們需要轉換的參數。Py_BuildValue
      會把所有的返回只組裝成一個tutple給python。
 
      如果對應的C函數沒有返回值（即返回值類型為void），則應返回一個全局的None
      對象(Py_None)，并將其引用計數增，如下所示：
      Py_INCREF(Py_None);
      returnPy_None;
   */
}

intcmd(constchar* arg)
{
    returnsystem(arg);
}
 
staticPyObject*ExecSystem(PyObject*self,PyObject*args)
{
    constchar*command;
    if(!PyArg_ParseTuple(args,"s", &command))
       returnNULL;
    intn =cmd(command);
    returnPy_BuildValue("i",n);
}