python-3.x - 使用 python 装饰器调用递归函数的次数

Question

我有以下关于 的代码块功能装饰我想在哪里初始化变量 wrapper.calls从我第二次调用 fib() 开始为 0知道在第一次运行时它给出了正确的答案，之后，它的行为就像对以前的输出求和一样。我的问题是如何通过仅更 retrofit 饰器分析器来解决此问题，好吗？

这是我的代码:

from functools import wraps
import time


def profiler(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        wrapper.calls += 1
        return func(*args, **kwargs)
    wrapper.calls = 0
    return wrapper

@profiler
def fib(n):
    if n == 0:
        return 0
    if n == 1:
        return 1
    else:
        return fib(n-1) + fib(n-2)


fib(1)
print(fib.calls)
fib(2)
print(fib.calls)

该程序执行后的输出:

1
4

预期产量

1 
3

Answer 1

def profiler(func):

    calls = 0

    @wraps(func)
    def wrapper(*args, **kwargs): 
        nonlocal calls
        if not calls:
            wrapper.calls = 0

        calls += 1
        result = func(*args, **kwargs)
        calls -= 1
        wrapper.calls += 1

        return result
    return wrapper