[Python] 改寫 Structure 的 __repr__ 函式,顯示 struct 的實際內容
在用 python 呼叫 C/C++ 的 DLL 時,常常會需要傳遞一個 struct 給 C/C++ 的函式,
struct 在 python 裡可以用 ctypes.Structure 來定義,
例如下面定義了一個 struct,對應到一個裡面有 char*, int, 和 bool 的 C struct:
import ctypes class TestStruct(ctypes.Structure): _fields_ = [("pattern", ctypes.c_char_p), ("build", ctypes.c_int), ("load", ctypes.c_bool)]
賦值給這個 struct 很簡單,但是在 python 端要查看 struct 裡有什麼東西的話,
一般的 print 只會印出這個 struct 物件的位址,沒啥路用:
>>> s = TestStruct("/tmp/pattern", 1000, True) >>> print s <__main__.TestStruct object at 0x10e6fd440>
上網查了一下,看到網友提供了很好的解決方法,
那就是覆寫 __repr__() 函式,在裡面把 struct 裡所有的資料 __fields__ 都顯示出來,
舉例來說,我們可以把 __repr__() 寫成像下面這樣:
class TestStruct(ctypes.Structure): _fields_ = [("pattern", ctypes.c_char_p), ("build", ctypes.c_int), ("load", ctypes.c_bool)]
def __repr__(self): return "%s(%s)" % (self.__class__.__name__, dict((name, getattr(self, name)) for name, _ in self._fields_))
這樣的話,print 這個 struct 時,就能將各欄位的值都印出來了,對於除錯很有幫助喔:
>>> s = TestStruct("/tmp/pattern", 1000, True) >>> print s TestStruct({'load': True, 'pattern': '/tmp/pattern', 'build': 1000})
參考資料:ctypes tricks
(本頁面已被瀏覽過 383 次)