Python如何將裝飾器定義為類

2022-10-04 13:48:28 字數 2522 閱讀 9670

問題

你想使用乙個裝飾器去包裝函式,但是希望返回乙個可呼叫的例項。 你需要讓你的裝飾器可以同時工作在類定義的內部和外部。

解決方案

為了將裝飾器定義成乙個例項,你需要確保它實現了 __call__() 和 __get__() 方法。 例如,下面的**定義了乙個類,它在其他函式上放置乙個簡單的記錄層:

import www.cppcns.comtypes

from functools import wraps

class profiled:

def __init__(self, func):

wraps(func)(self)

self.ncalls = 0

defsylimthk __call__(self, *args, **kwargs):

self.ncalls += 1

return self.__wrapped__(*args, **kwargs)

def __get__(self, instance, cls):

if instance is none:

return self

else:

return types.methodtype(self, instance)

你可以將它當做乙個普通的裝飾器來使用,在類裡面或外面都可以:

@profiled

def add(x, y):

return x + y

class spam:

@profiled

def bar(self, x):

print(self, x)

在互動環境中的使用示例:

>>> add(2, 3)

5>>> add(4, 5)

9>>> add.ncalls

2>>> s = spam()

>>> s.bar(1)

<__main__.spam object at> 1

>>> s.bar(2)

<__main__.spam object at> 2

>>> s.bar(3)

<__main__.spam object at> 3

>>> spam.bar.ncalls3討論

將裝飾器定義成類通常是很簡單的。但是這裡還是有一些細節需要解釋下,特別是當你想將它作用在例項方法上的時候。

首先,使用 functools.wraps() 函式的作用跟之前還是一樣,將被包裝函式的元資訊複製到可呼叫例項中去。

其次,通常很容易會忽視上面的 __get__() 方法。如果你忽略它,保持其他**不變再次執行, 你會發現當你去呼叫被裝飾例項方法時出現很奇怪的問題。例如:

>>> s = spam()

>>> s.bar(3)

traceback (most recent call last):

...typeerror: bar() missing 1 required positional argument: 'x'

出錯原因是當方法函式在乙個類中被查詢時,它們的 __get__() 方法依據描述器協議被呼叫, 在8.9小節已經講述過描述器協議了。在這裡,__get__() 的目的是建立乙個繫結方法物件 (最終會給這個方法傳遞self引數)。下面是乙個例子來演示底層原理:

>>> s = spam()

>>> def grok(self, x):

...www.cppcns.com pass

...>>> grok.__get__(s, spam)

>

>>>

__get__() 方法是為了確保繫結方法物件能被正確的建立。 type.methodtype() 手動建立乙個繫結方法來使用。只有當例項被使用的時候繫結方法才會被建立。 如果這個方法是在類上面來訪問, 那麼 __get__() 中的instance引數會被設定成none並直接返回 profiled 例項本身。 這樣的話我們就可以提取它的 ncalls 屬性了。

如果你想避免一些混亂,也可以考慮另外乙個使用閉包和 nonlocal 變數實現的裝飾器,這個在9.5小節有講到。例如:

import types

from functools import wraps

def profiled(func):

ncalls = 0

@wraps(func)

def wrapper(*args, **kwargs):

nonlocal ncalls

ncalls += 1

return func(*args, **kwargs)

wrapper.ncalls = lambda: ncalls

return wrapper

# example

@profiled

def add(x, y):

return x + y

這個方式跟之前的效果幾乎一樣,除了對於 ncalls 的訪問現在是通過乙個被繫結為屬性的函式來實現,例如:

>>> add(2, 3)

5>>> add(4, 5)

9>>> add.ncalls()

2>>>

如何將Py檔案打包為exe

1 安裝pip 2 安裝distribute,pip install distribute 3 按照pyinstaller,pip install pyinstaller 將所要打包的程式 即.py檔案和引用的資源檔案 放在乙個資料夾裡,用cmd進入這個目錄,輸入 pyinstaller f w p...

Python 如何定義帶引數的裝飾器?

案例 實現乙個裝飾器,用它來檢查被裝飾函式的引數型別。需求 裝飾器可以通過函式,指明函式引數型別,進行函式呼叫的時候,傳入引數,檢測到不匹配時,丟擲異常 如何解決這個問題?先要獲取函式的簽名,並且獲得裝飾器中引數,然後把函式簽名和裝飾器中引數對應繫結 把呼叫函式時候傳入的引數和函式簽名進行繫結 把實...

python如何定義帶引數的裝飾器

案例 實現乙個裝飾器,用它來檢查被裝飾函式的引數型別。需求 裝飾器可以通過程式設計客棧函式,指明函式引數型別,進行函式呼叫的時候,傳入引數,檢測到不匹配時,丟擲異常 如何解決這個問題?先要獲取函式的簽名,並且獲得裝飾器中引數,然後把函式簽名和裝飾器中引數對應繫結 把呼叫函式時候傳入的引數和函式簽名進...