【Python】python扩展

当python的基本功能无法满足要求。或者是为了保密源码（.py）、遇到性能瓶颈时，我们经常要扩展python，扩展语言能够是C/C++、Java、C#等。

为python创建扩展须要三个基本的步骤：创建应用程序代码；利用样板来包装代码；编译与測试。

1、 创建应用程序代码

我们创建一个C代码PythonEx.c，实现两个函数fac()和reverse()，分别用来求阶乘和逆转字符串，test()函数是用来測试fac()和reverse()基本功能的。以防问题带入python。

// PythonEx.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h> 

int fac(int n)
{
    if (n < 2) {
        return (1);
    }
    else {
        return (n) * fac(n - 1);
    }
} 

char* reverse(char *s)
{
    register char t;
    register char *p = s;
    register char *q = (s + strlen(s) -1); 

    while (p < q) {
        t = *p;
        *p++ = *q;
        *q-- = t;
    } 

    return s;
} 

int test()
{
    char s[BUFSIZ]; 

    printf("3! = %d\n", fac(3));
    printf("6! = %d\n", fac(6));
    printf("9! = %d\n", fac(9)); 

    strcpy(s, "abcdefg");
    printf("'abcdefg' after reversing is '%s'\n", reverse(s));
    strcpy(s, "python");
    printf("'python' after reversing is '%s'\n", reverse(s)); 

    return 0;
}

2、 用样板包装代码

样板是扩展代码与python解释器之间进行交互的桥梁，主要分为以下4步。

a.包括python的头文件。

找到python头文件位置并确保编译器的訪问权限，然后在代码中inlcude这个头文件，例如以下：

#include “Python.h”

b.为每个模块的每个函数添加一个型如PyObject* Module_func()的包装函数。

这部分须要为全部想被python环境訪问的函数都添加一个静态的函数。函数的返回值类型为PyObject*，函数名前面要加上模块名和一个下划线。

包装函数的用处就是先把python的值传递给C，然后调用我们想要调用的相关函数。当这个函数完毕要返回python的时候，把函数的计算结果转换成python的对象，然后返回给python。那么。在从python到C的转换就用PyArg_Parse*系列函数，在从C转到python的时候，就用Py_BuildValue()函数。

以下我们包装fac()和reverse()函数，如果其模块名为CustomPy，例如以下：

static PyObject* CustomPy_fac(PyObject *self, PyObject *args)
{
    int num;
    if (!PyArg_ParseTuple(args, "i", &num)) { // i i.e. int->int
        return NULL;
    }
    return (PyObject*)Py_BuildValue("i", fac(num)); // i i.e. int->int
} 

static PyObject* CustomPy_reverse(PyObject *self, PyObject *args)
{
    char *orig_str; // original
    char *dup_str; // reversed
    PyObject *retval;
    if (!PyArg_ParseTuple(args, "s", &orig_str)) { // s i.e. str->char*
        return NULL;
    }
    retval = (PyObject*)Py_BuildValue(
            "ss", // s i.e. char*->str
            orig_str,
            dup_str = reverse(strdup(orig_str))); // strdup
    free(dup_str); // free after strdup
    return retval; // return tuple(orig_str, dup_str)
}

static PyObject* CustomPy_test(PyObject *self, PyObject *args)
{
    test();
    return (PyObject*)Py_BuildValue("");
}

以下是Python和C/C++之间的数据格式——

FormatCode PythonType C/C++Type

s str char*

z str/None char*/NULL

i int int

l long long

c str char

d float double

D complex Py_Complex*

o (any) PyObject*

S str PyStringObject

c.为每个模块添加一个型如PyMethodDef ModuleMethods[]的数组。

static PyMethodDef CustomPyMethods[] = {
    {"fac", CustomPy_fac, METH_VARARGS},
    {"reverse", CustomPy_reverse, METH_VARARGS},
    {"test", CustomPy_test, METH_VARARGS},
    {NULL, NULL},
};

完毕包装函数后。把他们添加到一个数组中，以便于python解释器能够导入并调用它们。

每个数组都包括了函数在python中的名字、对应的包装函数的名字以及一个METH_VARARGS常量，这个常量表示參数以tuple形式传入。最后是一个NULL数组表示列表结束。

d.添加模块初始化函数void initModule()

void initCustomPy()
{
    Py_InitModule("CustomPy", CustomPyMethods);
}

这部分代码在模块被导入的时候被解释器调用。

另外。创建扩展还能够先写包装代码，使用桩函数、測试函数或哑函数，在开发过程中慢慢地把这些函数用有实际功能的函数替换。

3、编译与測试

为了让新python扩展能被创建。须要把它们与python库放在一起编译，曾经可能要用到Makefile，如今使用distutils模块就能够了，能够方便地编译、安装和分发这些模块、扩展和包。仅仅需创建一个setup.py脚本即可。

#!/usr/bin/env python 

from distutils.core import setup, Extension 

MOD = 'CustomPy'
setup(name = MOD, ext_modules = [Extension(MOD, sources = [PythonEx.c])])

以下直接执行setup.py脚本来创建自己定义模块。或者是install到python环境里，终于会生成一个so文件，接着import这个模块就能够使用了。

还有两点须要注意的是，python的垃圾自己主动回收策略是引用计数。以及线程安全操作，这些都能够添加到自己定义的模块中。

上面的样例中用C扩展了Python，我们还能够用Java扩展Jython，使用C#或者VB .NET扩展IronPython。如果在win32下，python还能够使用其COM（市场名字为ActiveX）操作Microsoft Office。