Python装饰器计数函数调用

Question

我刷新了我尚未得到的一些python功能的记忆，我正在从此python教程中学习，并且有一个我不完全了解的示例。这是关于装饰器计算对函数的调用的代码，这是代码：

def call_counter(func):
    def helper(x):
        helper.calls += 1
        return func(x)
    helper.calls = 0
    return helper

@call_counter
def succ(x):
    return x + 1

if __name__ == '__main__':
    print(succ.calls)
    for i in range(10):
        print(succ(i))
    print(succ.calls)

我在这里没有得到的是为什么我们增加函数包装器的调用（helper.calls + = 1）而不是函数调用本身，为什么它真正起作用？

Answer 1

要记住的装饰最重要的是，装饰是一个函数接受一个函数作为参数，并返回另一个功能。返回的值（另一个函数）是调用原始函数的名称时所调用的值。

这个模型可以很简单：

def my_decorator(fn):
    print("Decorator was called")
    return fn

在这种情况下，返回的函数与传入的函数相同。但这通常不是您要做的。通常，您返回一个完全不同的函数，或者返回以某种方式链接或包装原始函数的函数。

在您的示例（这是一个非常常见的模型）中，您将返回一个内部函数：

def helper(x):
    helper.calls += 1
    return func(x)

此内部函数调用原始函数（return func(x)），但也会增加调用计数器。

对于正在装饰的任何功能，此内部功能都将作为“替换”插入。因此，当您foo.succ()查找模块函数时，结果是对装饰器返回的内部帮助器函数的引用。该函数将递增调用计数器，然后调用最初定义的succ函数。

Answer 2

当你装饰一个函数时，你用包装器“替换”你的函数。

在这个例子中，在装饰之后，当你调用时，succ你实际上是在调用helper. 因此，如果您正在计算呼叫次数，则必须增加helper呼叫次数。

您可以通过检查装饰函数的属性 _ name _ 来检查装饰函数后名称是否已绑定到包装器：

def call_counter(func):
    def helper(*args, **kwargs):
        helper.calls += 1
        print(helper.calls)
        return func(*args, **kwargs)
    helper.calls = 0
    return helper

@call_counter
def succ(x):
    return x + 1

succ(0)
>>> 1
succ(1)
>>> 2
print(succ.__name__)
>>> 'helper'
print(succ.calls)
>>> 2