python - 在两个 python 装饰器之间传递变量

Question

有没有办法在应用于同一函数的两个Python装饰器之间传递变量？目标是让其中一个装饰器知道另一个装饰器也被应用。我需要下面示例中的 decobar_present() 之类的东西:

def decobar(f):
    def wrap():
        return f() + "bar"
    return wrap

def decofu(f):
    def wrap():
        print decobar_present() # Tells me whether decobar was also applied
        return f() + "fu"
    return wrap

@decofu
@decobar
def important_task():
    return "abc"

更一般地说，我希望能够根据是否也应用了 decobar 来修改 decofu 的行为。

Answer 1

要在两个 python 装饰器之间传递变量，您可以使用装饰函数的关键字参数字典。只是不要忘记在从第二个装饰器中调用函数之前从那里弹出添加的参数。

    def decorator1(func):
        def wrap(*args, **kwargs):
            kwargs['cat_says'] = 'meow'
            return func(*args, **kwargs)
        return wrap

    def decorator2(func):
        def wrap(*args, **kwargs):
            print(kwargs.pop('cat_says'))
            return func(*args, **kwargs)
        return wrap


    class C:
        @decorator1
        @decorator2
        def spam(self, a, b, c, d=0):
            print("Hello, cat! What's your favourite number?")
            return a + b + c + d

    x=C()
    print(x.spam(1, 2, 3, d=7))