1. 程式人生 > >Python cffi 初探

Python cffi 初探

新建立一個 ext_build.py 用於生成擴充套件

# ext_build.py
import numpy as np
import cffi

ffi = cffi.FFI()

ffi.cdef("""int add(int a, int b);""")
ffi.cdef("""int sub(int a, int b);""")

ffi.set_source('_ext', """
    int add(int a, int b)
    {
        return a + b;
    }
    int sub(int a, int b)
    {
        return a - b;
    }
""")

if __name__ == '__main__':
    ffi.compile(verbose=True)

ffi.cdef 表示宣告一個函式,ffi.set_source 第一個引數為擴充套件模組的名字,第二個引數為函式的定義。
然後執行 python ext_build.py

在 Windows 系統下大概會有如下顯示

generating .\_ext.c
(already up-to-date)
the current directory is '**省略**'
running build_ext
building '_ext' extension
C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\BIN\x86_amd64\cl.exe /c /nologo /Ox /W3 /GL /DNDEBUG /MD -IC:\ProgramData\Anaconda3\include -IC:\ProgramData\Anaconda3\include "-IC:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\INCLUDE" "-IC:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\ATLMFC\INCLUDE" "-IC:\Program Files (x86)\Windows Kits\10\include\10.0.10240.0\ucrt" "-IC:\Program Files (x86)\Windows Kits\NETFXSDK\4.6.1\include\um" "-IC:\Program Files (x86)\Windows Kits\8.1\include\\shared" "-IC:\Program Files (x86)\Windows Kits\8.1\include\\um" "-IC:\Program Files (x86)\Windows Kits\8.1\include\\winrt" "-ID:\Microsoft Visual Studio\VC98\atl\include" "-ID:\Microsoft Visual Studio\VC98\mfc\include" "-ID:\Microsoft Visual Studio\VC98\include" /Tc_ext.c /Fo.\Release\_ext.obj
_ext.c
C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\BIN\x86_amd64\link.exe /nologo /INCREMENTAL:NO /LTCG /DLL /MANIFEST:EMBED,ID=2 /MANIFESTUAC:NO /LIBPATH:C:\ProgramData\Anaconda3\libs /LIBPATH:C:\ProgramData\Anaconda3\PCbuild\amd64 "/LIBPATH:C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\LIB\amd64" "/LIBPATH:C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\ATLMFC\LIB\amd64" "/LIBPATH:C:\Program Files (x86)\Windows Kits\10\lib\10.0.10240.0\ucrt\x64" "/LIBPATH:C:\Program Files (x86)\Windows Kits\NETFXSDK\4.6.1\lib\um\x64" "/LIBPATH:C:\Program Files (x86)\Windows Kits\8.1\lib\winv6.3\um\x64" "/LIBPATH:D:\Microsoft Visual Studio\VC98\mfc\lib" "/LIBPATH:D:\Microsoft Visual Studio\VC98\lib" /EXPORT:PyInit__ext .\Release\_ext.obj /OUT:.\_ext.cp36-win_amd64.pyd /IMPLIB:.\Release\_ext.cp36-win_amd64.lib
_ext.obj : warning LNK4197: 多次指定匯出“PyInit__ext”;使用第一個規範
  正在建立庫 .\Release\_ext.cp36-win_amd64.lib 和物件 .\Release\_ext.cp36-win_amd64.exp
正在生成程式碼
已完成程式碼的生成

然後目錄下生成了 _ext.cp36-win_amd64.pyd 這一個包。

新建一個 demo 測試一下

# demo.py
from _ext import ffi, lib

print(lib.add(2, 3))
print(lib.sub(2, 3))

輸出

5
-1

通過。