我如何通过额外的参数给一个Python装饰？我如何通过额外的参数给一个Python装饰？(How d

我有一个像下方的装饰。

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'

但是这个代码给出了错误，

类型错误：myDecorator（）恰恰2参数（1给出）

为什么没有功能自动传递？ 如何明确地传递功能的装饰功能？

Answer 1:

既然你在呼唤像功能的装饰，它需要返回另一个函数，它是实际的装饰：

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调用拷贝之类的名称和文档字符串包装的功能，使其更类似于原始功能。

实例：

>>> @log_decorator(True)
... def f(x):
...     return x+1
...
>>> f(4)
Calling Function: f
5

Answer 2:

只需提供一个不同的观点：语法

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

相当于

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

特别是， expr可以是任何你喜欢的，只要它的计算结果为一个可调用。 尤其特别， expr可以是装饰厂：你给它的一些参数，它给你一个装饰。 因此，也许有更好的方式来了解你的情况是因为

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

然后可以缩短到

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

当然，因为它看起来像decorator_factory是一个装饰，人们往往将它命名，以反映。 当你尝试按照间接的水平，这可能会造成混淆。

Answer 3:

只想补充一些有用的技巧，让做装饰参数可选。 它还将alows重用装饰和减少嵌套

import functools

def myDecorator(test_func=None,logIt=None):
    if not test_func:
        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)

Answer 4:

做装饰的另一种方式。 我觉得这种方式最容易绕到我的头。

class NiceDecorator:
    def __init__(self, param_foo='a', param_bar='b'):
        self.param_foo = param_foo
        self.param_bar = param_bar

    def __call__(self, func):
        def my_logic(*args, **kwargs):
            # whatever logic your decorator is supposed to implement goes in here
            print('pre action baz')
            print(self.param_bar)
            # including the call to the decorated function (if you want to do that)
            result = func(*args, **kwargs)
            print('post action beep')
            return result

        return my_logic

# usage example from here on
@NiceDecorator(param_bar='baaar')
def example():
    print('example yay')


example()

文章来源: How do I pass extra arguments to a Python decorator?