Python C 语言扩展

测试积点老人

1. 创建应用程序代码，用样板包装代码

1). 包含 Python 头文件。

2) 为每个模版的每一个函数增加 PyObject * Module_func() 的包装函数。

3) 为每个模块的每一个函数增加 PyMethodDef * ModuleMethods[] 的数组。

4) 增加模块初始化函数 void initModule()。

模块名为 Extest，包含 Add 和 Sub 方法。

1. #include <Python.h>
   
2. 
   
3. /* C 函数 */
   
4. double Add(double a, double b)
   
5. {
   
6.         return a + b;
   
7. }
   
8. 
   
9. double Sub(double a, double b)
   
10. {
    
11.         return a - b;
    
12. }
    
13. 
    
14. /* 为每个模块的每一个函数增加包装函数  */
    
15. static PyObject * Extest_Add(PyObject *self, PyObject *args)
    
16. {
    
17.         double a;
    
18.         double b;
    
19. 
    
20.         /* PyArg_ParseTuple 把 Python传过来的参数转为 C */
    
21.         if (!PyArg_ParseTuple(args, "dd", &a, &b))
    
22.                 return NULL;
    
23. 
    
24.         /* Py_BuildValue 把 C 的数据转为 Python 的一个或一组对象，然后返回 */
    
25.         return (PyObject *)Py_BuildValue("d", Add(a, b));
    
26. }
    
27. 
    
28. static PyObject * Extest_Sub(PyObject *self, PyObject *args)
    
29. {
    
30.         double a;
    
31.         double b;
    
32. 
    
33.         if (!PyArg_ParseTuple(args, "dd", &a, &b))
    
34.                 return NULL;
    
35. 
    
36.         return (PyObject *)Py_BuildValue("d", Sub(a, b));
    
37. }
    
38. 
    
39. /* 为每个模块增加一个PyMethodDef数组 */
    
40. static PyMethodDef ExtestMethods[] = {
    
41.         { "Add", Extest_Add, METH_VARARGS },
    
42.         { "Sub", Extest_Sub, METH_VARARGS },
    
43.         {NULL, NULL}
    
44. };
    
45. 
    
46. /* 增加模块初始化函数 void initModule() */
    
47. void initExtest()
    
48. {
    
49.         Py_InitModule("Extest", ExtestMethods);
    
50. }

复制代码

2. 编译
2.1 创建setup.py

1. #!/usr/bin/env python
   
2. 
   
3. from distutils.core import setup, Extension
   
4. 
   
5. MOD = 'Extest'
   
6. setup(name=MOD, ext_modules=[Extension(MOD, sources=['Extest.c'])])

复制代码

2.2 通过运行 setup.py 来编译和链接代码

编译

1. $ python setup.py build

复制代码

安装模块

1. $ python setup.py install

复制代码

2.3 导入并执行

1. >>> import Extest
   
2. >>> Extest.Add(4,5)
   
3. 9.0
   
4. >>> Extest.Sub(5,4)
   
5. 1.0
   
6. >>>

复制代码

PS：Windows 编译出错

如果出现

1. F:\01_Code\01_Python\Python_C\Python_C>python setup.py build
   
2. running build
   
3. running build_ext
   
4. building 'Extest' extension
   
5. error: Unable to find vcvarsall.bat

复制代码

如果为 Python2.7，并且安装了VS2010，则可以通过在环境变量中添加：

变量名：VS90COMNTOOLS

变量值：%VS100COMNTOOLS%

PPS：Linux 缺少头文件

如果缺少头文件，需要安装 python-dev

1. sudo apt-get install python-dev

复制代码

Miss_love

支持分享