一步步來用C語言來寫python擴展

本文介紹如何用 C 語言來擴展 python。所舉的例子是，為 python 添加一個設置字符串到 windows 的剪切板(Clipboard)的功能。我在寫以下代碼的時候用到的環境是：windows xp, gcc.exe 4.7.2, Python 3.2.3。

第一步 撰寫C語言的DLL

創建一個 clip.c 文件，內容如下：

// 設置 UNICODE 庫，這樣的話才可以正確複製寬字符集

#define UNICODE

#include <windows.h>

#include <python.h>

// 設置文本到剪切板(Clipboard)

static PyObject *setclip(PyObject *self, PyObject *args)

{

LPTSTR lptstrCopy;

HGLOBAL hglbCopy;

Py_UNICODE *content;

int len = 0;

// 將 python 的 UNICODE 字符串及長度傳入

if (!PyArg_ParseTuple(args, "u#", &content, &len))

return NULL;

Py_INCREF(Py_None);

if (!OpenClipboard(NULL))

return Py_None;

EmptyClipboard();

hglbCopy = GlobalAlloc(GMEM_MOVEABLE, (len+1) * sizeof(Py_UNICODE));

if (hglbCopy == NULL) {

CloseClipboard();

return Py_None;

}

lptstrCopy = GlobalLock(hglbCopy);

memcpy(lptstrCopy, content, len * sizeof(Py_UNICODE));

lptstrCopy[len] = (Py_UNICODE) 0;

GlobalUnlock(hglbCopy);

SetClipboardData(CF_UNICODETEXT, hglbCopy);

CloseClipboard();

return Py_None;

}

// 定義導出給 python 的方法

static PyMethodDef ClipMethods[] = {

{"setclip", setclip, METH_VARARGS,

"Set string to clip."},

{NULL, NULL, 0, NULL}

};

// 定義 python 的 model

static struct PyModuleDef clipmodule = {

PyModuleDef_HEAD_INIT,

"clip",

NULL,

-1,

ClipMethods

};

// 初始化 python model

PyMODINIT_FUNC PyInit_clip(void)

{

return PyModule_Create(&clipmodule);

}

第二步 寫 python 的 setup.py

創建一個 setup.py 文件，內容如下：

from distutils.core import setup, Extension

module1 = Extension('clip',

sources = ['clip.c'])

setup (name = 'clip',

version = '1.0',

description = 'This is a clip package',

ext_modules = [module1])

第三步 用 python 編譯

運行以下命令：

python setup.py build --compiler=mingw32 install

在我的環境中會提示以下錯誤：

gcc: error: unrecognized command line option '-mno-cygwin'

error: command 'gcc' failed with exit status 1

打開 %PYTHON安裝目錄%/Lib/distutils/cygwinccompiler.py 文件，將裡面的 -mno-cygwin 刪除掉，然後再運行即可。

正常運行後，會生成一個 clip.pyd 文件，並將該文件複製到 %PYTHON安裝目錄%/Lib/site-packages 目錄中

第四步 測試該擴展

寫一個 test.py, 內容如下：

# -*- encoding: gbk -*-

import clip

clip.setclip("Hello python")

運行

python test.py

再到任何一個地方粘貼，即可驗證是否正確。

更多技巧請《轉發 + 關注》哦！