Python で引数を取るデコレータの作り方
デコレータ自身がパラメータを受け取りたい場合、関数をもう一段ネストさせます。「デコレータを返す関数」を作るイメージです。
基本構造
引数を取るデコレータは 3 層構造になります。
def repeat(n):
def decorator(func):
def wrapper(*args, **kwargs):
for _ in range(n):
result = func(*args, **kwargs)
return result
return wrapper
return decorator使い方は以下のとおりです。
@repeat(3)
def say_hello():
print("Hello!")
say_hello()実行すると「Hello!」が 3 回出力されます。@repeat(3) は repeat(3) を呼び出してデコレータを取得し、それを関数に適用しています。つまり say_hello = repeat(3)(say_hello) と同じ意味です。
実用例：リトライ処理
失敗時に再試行するデコレータを作ってみましょう。
import time
def retry(max_attempts, delay=1):
def decorator(func):
def wrapper(*args, **kwargs):
for attempt in range(max_attempts):
try:
return func(*args, **kwargs)
except Exception as e:
if attempt < max_attempts - 1:
print(f"リトライ {attempt + 1}/{max_attempts}: {e}")
time.sleep(delay)
else:
raise
return wrapper
return decorator
@retry(max_attempts=3, delay=2)
def fetch_data():
# ネットワーク処理など
passこのように引数付きデコレータを使うと、デコレータの挙動を柔軟にカスタマイズできます。
21views
