一步步來用C語言來寫python擴充套件

pythontab發表於2013-08-30
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

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


相關文章