Python扩展|极客教程

Python扩展

Python是一种简单而强大的编程语言，但有时候我们需要扩展Python的功能以满足特定需求。Python提供了多种扩展的方式，如使用C语言编写Python扩展模块、调用C/C++代码、使用外部库等。本文将从多个角度详细介绍如何扩展Python的功能。

使用C语言编写Python扩展模块

准备工作

在使用C语言编写Python扩展之前，我们需要准备好C编译器和Python的开发包。对于Windows系统，可以使用Visual Studio或MinGW作为C编译器；对于Linux系统，可以使用GCC。此外，还需要安装Python的开发包，通常可以通过系统的包管理器安装。

编写扩展模块

首先创建一个新的C源文件，可以命名为mymodule.c，编写如下代码：

#include <Python.h>

static PyObject* mymodule_hello(PyObject* self, PyObject* args)
{
    const char* name;

    if (!PyArg_ParseTuple(args, "s", &name))
        return NULL;

    printf("Hello, %s!\n", name);

    Py_RETURN_NONE;
}

static PyMethodDef mymodule_methods[] = {
    {"hello", mymodule_hello, METH_VARARGS, "Print a greeting message."},
    {NULL, NULL, 0, NULL}
};

static struct PyModuleDef mymodule = {
    PyModuleDef_HEAD_INIT,
    "mymodule",
    "A simple Python C extension module",
    -1,
    mymodule_methods
};

PyMODINIT_FUNC PyInit_mymodule(void)
{
    return PyModule_Create(&mymodule);
}

这段代码定义了一个简单的Python扩展模块，包含一个名为hello的函数，用于打印问候信息。接下来编写一个setup.py文件用于构建和安装这个扩展模块：

from distutils.core import setup, Extension

module = Extension("mymodule", sources=["mymodule.c"])

setup(name="mymodule",
      version="1.0",
      description="A simple Python C extension module",
      ext_modules=[module])

构建和安装扩展模块

在终端中执行以下命令来构建和安装扩展模块：

$ python setup.py build
$ python setup.py install

使用扩展模块

现在可以在Python中使用这个扩展模块了：

import mymodule

mymodule.hello("Alice")

运行以上代码将输出Hello, Alice!，这说明我们成功使用了C语言编写的Python扩展模块。

调用C/C++代码

除了编写Python扩展模块外，还可以直接调用C/C++代码来扩展Python的功能。这种方式通常用于优化性能或利用现有的C/C++库。

使用ctypes调用动态链接库

在Python中，可以使用ctypes模块调用动态链接库中的C函数。下面是一个简单的示例，调用动态链接库libm中的cos函数：

import ctypes

libc = ctypes.CDLL("libm.so.6")  # 加载libm.so.6库

x = 1.0
result = libc.cos(x)

print(result)

运行以上代码将输出0.5403023058681398，这是cos(1.0)的值，说明我们成功调用了动态链接库中的C函数。

使用Cython调用C/C++代码

Cython是一个Python库，可以在Python代码中直接嵌入C/C++代码。下面是一个使用Cython调用C函数的示例：

# example.pyx
cdef extern from "math.h":
    double cos(double x)

def my_cos(x):
    return cos(x)

要编译Cython代码，可以使用cython命令：

$ cython example.pyx

然后编译生成的C代码：

$ gcc -shared -pthread -fPIC -fwrapv -O2 -Wall -fno-strict-aliasing -I/usr/include/python3.6m -o example.so example.c -lm

最后在Python中调用这个Cython扩展模块：

import example

result = example.my_cos(1.0)
print(result)

使用外部库

Python还提供了多种方法来调用外部库，如ctypes、Cython、CFFI等。这些库可以帮助我们方便地调用C/C++库或系统API。下面以ctypes为例，演示如何使用外部库扩展Python的功能。

调用系统API

可以使用ctypes调用系统API来扩展Python的功能。下面是一个简单的示例，调用libc中的printf函数：

import ctypes

libc = ctypes.CDLL("libc.so.6")  # 加载libc.so.6库

format = b"Hello, %s!\n"
name = b"Bob"

libc.printf(format, name)

运行以上代码将输出Hello, Bob!，这是通过调用系统API中的printf函数实现的。

调用第三方库

除了调用系统API外，还可以使用ctypes调用第三方库。下面是一个示例，调用libcrypto库中的SHA256函数计算哈希值：

import ctypes

libcrypto = ctypes.CDLL("libcrypto.so.1.1")  # 加载libcrypto.so.1.1库

data = b"hello"
digest = (ctypes.c_ubyte * 32)()

libcrypto.SHA256(data, len(data), digest)

for byte in digest:
    print(hex(byte), end="")

运行以上代码将输出b'2c0749d89304c8496a61f3a727a3680ec3f183823735d674940b4e44'，这是SHA256("hello")的哈希值。