Pythonのデコレーターで関数を修飾してみよう

解説

>>> def logger(func):

デコレーターとしてlogger関数を定義しています。引数の「func」は、デコレート対象の関数が入ります。

...        print("--- call {0} ---".format(func.__name__))
...        func(*args)
...        print("--- call {0} ---".format(func.__name__))

関数内の「func(*args)」は、ネスト（入れ子に）した関数を定義しています。

「__name__」属性からは、関数名を取得できます。 wrapper関数は、可変長引数（引数をいくつでも受け取れる）になっており、それをそのままデコレート対象の関数の引数として渡しています。（func.__name__）

そして最後に、「return wrapper」でwrapper関数を返しています。

解説

>>> @logger
... def add(x, y):
...     print(x+y)
... 
>>> add(2, 3)

①関数をデコレートするには、関数定義の前に”@(デコレーター名)”を指定します。後は通常通り関数を呼び出すだけです。

②すると、先程のデコレーターが呼び出され、関数呼び出しの前後に処理が追加されます。

出力結果：

--- call add ---
5
--- call add ---

解説

>>> decorate = logger(add) 

logger関数にデコレートする関数を渡すと、それをラップしたwrapper関数が返ってきます。

>>> decorate(2, 3)

後はそれに通常通りの引数を渡します。するとwrapper関数の内部で、追加の処理とデコレートする関数（add）が呼び出されます。

解説

...     def __init__(self, func):
...         self.func = func

クラスでデコレーターを定義する場合には、「__init__」でデコレートする関数を受け取って自身の変数に保存しておきます。

...     def __call__(self, *args):
...         start = time()
...         self.func(*args)
...         difftime = time() - start
...         print("execute time: {0}".format(difftime))

デコレーターが呼び出されると「__call__」が実行されます。

先ほどのwrapper関数と同じように追加の処理とデコレートする関数（　self.func(*args)　）を呼び出します。

>>> @timer
... def slooop():
（省略）

デコレートする関数は、同じように”@(デコレーター名)”を指定するだけです。