Python のクラスベースデコレータ

デコレータは関数だけでなく、クラスとしても実装できます。状態を保持したい場合や、複雑なロジックを整理したい場合に有効です。

基本構造

クラスベースのデコレータは __init__ で関数を受け取り、__call__ でラッパー処理を行います。

from functools import wraps

class CountCalls:
    def __init__(self, func):
        wraps(func)(self)
        self.func = func
        self.count = 0
    
    def __call__(self, *args, **kwargs):
        self.count += 1
        print(f"{self.func.__name__} は {self.count} 回呼ばれました")
        return self.func(*args, **kwargs)

使い方は関数デコレータと同じです。

@CountCalls
def say_hello():
    print("Hello!")

say_hello()  # say_hello は 1 回呼ばれました / Hello!
say_hello()  # say_hello は 2 回呼ばれました / Hello!

クラスなので、self.count のように状態を自然に保持できます。

引数を取るクラスデコレータ

引数を取る場合は、__init__ で引数を受け取り、__call__ で関数を受け取ります。

class Repeat:
    def __init__(self, n):
        self.n = n
    
    def __call__(self, func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            for _ in range(self.n):
                result = func(*args, **kwargs)
            return result
        return wrapper

@Repeat(3)
def greet():
    print("Hi!")

@Repeat(3) でインスタンスが生成され、その __call__ にデコレート対象の関数が渡されます。関数ベースの 3 層ネストよりも構造が明確になることがあります。