本文介绍如何用 C 语言来扩展 python。所举的例子是,为 python 添加一个设置字符串到 windows 的剪切板(Clipboard)的功能。我在写以下代码的时候用到的环境是:windows xp,gcc.exe 4.7.2,Python 3.2.3。
第一步 撰写C语言的DLL
// 设置 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,NULL}
};
// 定义 python 的 model
static struct PyModuleDef clipmodule = {
PyModuleDef_HEAD_INIT,"clip",-1,ClipMethods
};
// 初始化 python model
PyMODINIT_FUNC PyInit_clip(void)
{
return PyModule_Create(&clipmodule);
}
# End www.jb51.cc
第二步 写 python 的 setup.py
# @param 一步步来用C语言来写python扩展
# @author 编程之家 jb51.cc|www.www.jb51.cc
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])
# End www.jb51.cc
第三步 用 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,内容如下:
# @param 一步步来用C语言来写python扩展
# @author 编程之家 jb51.cc|www.www.jb51.cc
# -*- encoding: gbk -*-
import clip
clip.setclip("Hello python")
# End www.jb51.cc
运行
python test.py
再到任何一个地方粘贴,即可验证是否正确。