在python中访问装饰器属性?

时间:2018-06-17 10:12:49

标签: python decorator python-decorators

是否可以在Python 3中访问装饰器属性?

例如:在调用装饰的斐波那契方法后是否可以访问self.misses

class Cache:
    def __init__(self, func):
        self.func = func
        self.cache = {}
        self.misses = 0    
    def __call__(self, *args):
        if not (args in self.cache):
            self.misses += 1
            self.cache[args] = self.func(*args)
        return self.cache[args]

@Cache    
def fibonacci(n):
    return n if n in (0, 1) else fibonacci(n - 1) + fibonacci(n - 2)

fibonacci(20)
### now we want to print the number of cache misses ###

1 个答案:

答案 0 :(得分:5)

当你装饰一个函数(或类或其他任何东西)时,你实际上用装饰器的返回值替换装饰对象。这意味着:

def fibonacci(n):
    return n if n in (0, 1) else fibonacci(n - 1) + fibonacci(n - 2)

fibonacci = Cache(fibonacci)

相当于:

fibonacci

因此,Cache现在是>>> fibonacci <__main__.Cache object at 0x7fd4d8b63e80> 实例,而不是函数:

fibonacci

因此,为了获得缓存未命中数,您只需访问misses的{​​{1}}属性:

>>> fibonacci.misses
21