『Python CoolBook』C扩展库_其一_用法讲解

时间:2021-06-01 20:20:26

不依靠其他工具,直接使用Python的扩展API来编写一些简单的C扩展模块。

本篇参考PythonCookbook第15节和Python核心编程完成,值得注意的是,Python2.X和Python3.X在扩展库写法上略有不同,我们研究的是3.X写法。

一、源文件

Extest2.c

C函数本体

c文件头必须包含"Python.h"头,以调用接口函数

这里面写了两个c函数,模块名称定为Extest

#include "Python.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h> int fac(int n)
{
if (n<2) return(1);
return (n)*fac(n-1);
} char *reverse(char *s)
{
register char t,
*p = s,
*q = (s + (strlen(s)-1));
while (p < q)
{
t = *p;
*p++ = *q;
*q-- = t;
}
return s;
}

Python API封装

对这两个函数采取C对Python API封装,

  • 封装函数为静态函数,输入返回均为PyObject*,且输入须有一个self用于处理类对自身的引用,函数名须为模块名_c函数名格式(模块名不是强制的)
    • static PyObject*
      模块名称_C函数名称(PyObject *self, PyObject *args)
  • Python对象->C对象,用于C函数输入,使用PyArg_ParseTuple为元组输入解析类型,另有PyArg_ParseTupleAndKeywords为字典类型解析类型
    • PyArg_ParseTuple(args, "i", &num)  //PyObject参数,类型指定,C参数存放地址1,[C参数存放地址2……]
  • C对象->Python对象,用于C函数输出,使用(PyObject*)Py_BuildValue进行类型转换
    • (PyObject*)Py_BuildValue("ss", orig_str, \
      dupe_str = reverse(strdup(orig_str))); //类型指定,C参数1,[C参数2……]
  • return Python对象,由于是个元组,所以上一步多少个C输出都没有关系
static PyObject*
Extest_fac(PyObject *self, PyObject *args)
{
int num;
if (!PyArg_ParseTuple(args, "i", &num))
return NULL;
return (PyObject*)Py_BuildValue("i", fac(num));
} static PyObject*
Extest_doppel(PyObject *self, PyObject *args)
{
char *orig_str;
char *dupe_str;
PyObject* retval;
if (!PyArg_ParseTuple(args, "s", &orig_str)) return NULL;
retval = (PyObject*)Py_BuildValue("ss", orig_str, \
dupe_str = reverse(strdup(orig_str)));
free(dupe_str);
return retval;
}

首先,在扩展模块中,你写的函数都是像下面这样的一个普通原型:

static PyObject *py_func(PyObject *self, PyObject *args) {
...
}

PyObject 是一个能表示任何Python对象的C数据类型。 在一个高级层面,一个扩展函数就是一个接受一个Python对象 (在 PyObject *args中)元组并返回一个新Python对象的C函数。 函数的 self 参数对于简单的扩展函数没有被使用到, 不过如果你想定义新的类或者是C中的对象类型的话就能派上用场了。比如如果扩展函数是一个类的一个方法, 那么 self 就能引用那个实例了。

PyArg_ParseTuple() 函数被用来将Python中的值转换成C中对应表示。 它接受一个指定输入格式的格式化字符串作为输入,比如“i”代表整数,“d”代表双精度浮点数, 同样还有存放转换后结果的C变量的地址。 如果输入的值不匹配这个格式化字符串,就会抛出一个异常并返回一个NULL值。 通过检查并返回NULL,一个合适的异常会在调用代码中被抛出。

Py_BuildValue() 函数被用来根据C数据类型创建Python对象。 它同样接受一个格式化字符串来指定期望类型。 在扩展函数中,它被用来返回结果给Python。 Py_BuildValue() 的一个特性是它能构建更加复杂的对象类型,比如元组和字典。 在 py_divide() 代码中,一个例子演示了怎样返回一个元组。不过,下面还有一些实例:

return Py_BuildValue("i", 34);      // Return an integer
return Py_BuildValue("d", 3.4); // Return a double
return Py_BuildValue("s", "Hello"); // Null-terminated UTF-8 string
return Py_BuildValue("(ii)", 3, 4); // Tuple (3, 4)

库信息记录

库信息以及初始化信息

/* 记录函数信息,{函数在Python中名称,函数对应封装,参数格式(此处表示参数以元组格式传入)} */
static PyMethodDef
ExtestMethods[] =
{
{"fac", Extest_fac, METH_VARARGS},
{"doppel", Extest_doppel, METH_VARARGS},
{NULL, NULL},
}; /* Module structure */
static struct PyModuleDef Extestmodule = {
PyModuleDef_HEAD_INIT, "Extest", /* 库名称 */
"A sample module", /* Doc string (may be NULL) */
-1, /* Size of per-interpreter state or -1 */
ExtestMethods /* 函数信息 */
}; /* Module initialization function */
PyMODINIT_FUNC
PyInit_Extest(void) { /* PyInit_库名称 */
return PyModule_Create(&Extestmodule); /* 参数为Module structure名词 */
}

在扩展模块底部,你会发现一个函数表,比如本节中的 ExtestMethods 表。 这个表可以列出C函数、Python中使用的名字、文档字符串。 所有模块都需要指定这个表,因为它在模块初始化时要被使用到。

最后的函数 PyInit_Extest() 是模块初始化函数,但该模块第一次被导入时执行。 这个函数的主要工作是在解释器中注册模块对象。

二、编译文件setup.py

from distutils.core import setup, Extension

MOD = "Extest"
setup(name=MOD, # 一个名字参数表示要编译哪个东西
    ext_modules=[ # 一个list对象列出编译对象
      Extension(MOD, sources=['Extest2.c']) # 完整扩展名(可能含有.),源文件
   ])

三、测试调用

调用:

python setup.py build
python setup.py install

测试:

『Python CoolBook』C扩展库_其一_用法讲解