120

以下のようなデコレータがあります。

def myDecorator(test_func):
    return callSomeWrapper(test_func)
def callSomeWrapper(test_func):
    return test_func
@myDecorator
def someFunc():
    print 'hello'

このデコレーターを拡張して、以下のような別の引数を受け入れたい

def myDecorator(test_func,logIt):
    if logIt:
        print "Calling Function: " + test_func.__name__
    return callSomeWrapper(test_func)
@myDecorator(False)
def someFunc():
    print 'Hello'

しかし、このコードはエラーを出し、

TypeError: myDecorator() は正確に 2 つの引数を取ります (1 つ指定)

関数が自動的に渡されないのはなぜですか? 関数をデコレータ関数に明示的に渡すにはどうすればよいですか?

4

5 に答える 5

203

関数のようにデコレータを呼び出しているため、実際のデコレータである別の関数を返す必要があります。

def my_decorator(param):
    def actual_decorator(func):
        print("Decorating function {}, with parameter {}".format(func.__name__, param))
        return function_wrapper(func)  # assume we defined a wrapper somewhere
    return actual_decorator

外側の関数には、明示的に渡す引数が与えられ、内側の関数を返す必要があります。内部関数は装飾する関数を渡され、変更された関数を返します。

通常、ラッパー関数でラップすることにより、デコレータで関数の動作を変更する必要があります。関数が呼び出されたときにオプションでロギングを追加する例を次に示します。

def log_decorator(log_enabled):
    def actual_decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kwargs):
            if log_enabled:
                print("Calling Function: " + func.__name__)
            return func(*args, **kwargs)
        return wrapper
    return actual_decorator

このfunctools.wraps呼び出しは、名前や docstring などをラッパー関数にコピーして、元の関数により似たものにします。

使用例:

>>> @log_decorator(True)
... def f(x):
...     return x+1
...
>>> f(4)
Calling Function: f
5
于 2012-04-16T14:41:28.717 に答える
54

別の視点を提供するためだけに: 構文

@expr
def func(...): #stuff

と同等です

def func(...): #stuff
func = expr(func)

特に、exprcallable に評価される限り、好きなものを指定できます。特に、exprデコレータ ファクトリにすることができます。いくつかのパラメータを指定すると、デコレータが提供されます。したがって、状況を理解するためのより良い方法は次のとおりです。

dec = decorator_factory(*args)
@dec
def func(...):

これは次のように短縮できます

@decorator_factory(*args)
def func(...):

もちろん、デコレーターのように見えるdecorator_factoryので、それを反映して名前を付ける傾向があります。インダイレクションのレベルに従おうとすると、混乱する可能性があります。

于 2012-04-16T15:09:02.050 に答える
34

デコレータ引数をオプションにすることができる便利なトリックを追加したいだけです。また、デコレータを再利用してネストを減らすこともできます

import functools

def myDecorator(test_func=None,logIt=None):
    if test_func is None:
        return functools.partial(myDecorator, logIt=logIt)
    @functools.wraps(test_func)
    def f(*args, **kwargs):
        if logIt==1:
            print 'Logging level 1 for {}'.format(test_func.__name__)
        if logIt==2:
            print 'Logging level 2 for {}'.format(test_func.__name__)
        return test_func(*args, **kwargs)
    return f

#new decorator 
myDecorator_2 = myDecorator(logIt=2)

@myDecorator(logIt=2)
def pow2(i):
    return i**2

@myDecorator
def pow3(i):
    return i**3

@myDecorator_2
def pow4(i):
    return i**4

print pow2(2)
print pow3(2)
print pow4(2)
于 2016-04-20T09:44:01.997 に答える