如何将额外的参数传递给Python装饰器?


93

我有一个如下的装饰器。

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个)

为什么不能自动传递函数?如何将函数显式传递给装饰器函数?


3
巴尔基:请避免使用布尔值作为参数,这不是gd方法,并且会降低代码的可读性
Kit Ho

8
@KitHo-这是一个布尔标志,因此使用布尔值是正确的方法。
AKX 2012年

2
@KitHo-什么是“ gd”?好吗”?
Rob Bednark

Answers:


164

由于您像装饰器一样调用装饰器,因此它需要返回另一个函数,即实际的装饰器:

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

11
并且functools.wraps建议使用-保留包装函数的原始名称,文档字符串等。
AKX 2012年

@AKX:谢谢,我将此添加到第二个示例。
interjay 2012年

1
因此,基本上装饰器总是只接受一个参数,即函数。但是装饰器可以是可能带有参数的函数的返回值。这样对吗?
balki 2012年

2
@balki:是的,这是正确的。令人困惑的是,许多人还将外部函数(myDecorator这里)称为装饰器。这对于装饰器的用户来说很方便,但是在您尝试编写装饰器时可能会造成混淆。
interjay 2012年

1
小细节让我感到困惑:如果您log_decorator使用默认参数,则不能使用@log_decorator,它必须是@log_decorator()
Stardidi

44

只是提供不同的观点:语法

@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一个装饰器,因此人们倾向于将其命名以反映这一点。当您尝试遵循间接级别时,这可能会造成混淆。


谢谢,这确实帮助我了解了发生了什么的原因。
Aditya Sriram

24

只是想添加一些有用的技巧,使装饰器参数成为可选。它还将允许重用装饰器并减少嵌套

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)

16

只是做装饰的另一种方法。我发现这种方式最容易缠住我的头。

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()

谢谢!一直在寻找一些令人费解的“解决方案”约30分钟,这是第一个真正有意义的解决方案。
canhazbits

0

现在,如果您要function1使用装饰器调用函数,decorator_with_arg并且在这种情况下,该函数和装饰器都接受参数,

def function1(a, b):
    print (a, b)

decorator_with_arg(10)(function1)(1, 2)
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.