创建包装函数
在这一步中，你将学习如何使用包装函数来跟踪函数调用。包装函数是一种包裹另一个函数的函数，它允许你在原函数执行前后添加额外的功能。这是一种强大的技术，可用于监控函数调用，而无需修改原函数的代码。
让我们继续使用上一步中的 greet 函数。你将创建一个包装函数来跟踪 greet 函数的调用次数。
修改 ~/project 目录下的 my_function.py 文件，使其包含以下代码：
## ~/project/my_function.py
invocation_count = 0
def greet(name):
"""This function greets the person passed in as a parameter."""
print(f"Hello, {name}!")
def greet_wrapper(func):
"""This is a wrapper function that tracks the number of times the wrapped function is called."""
def wrapper(*args, **kwargs):
global invocation_count
invocation_count += 1
result = func(*args, **kwargs)
return result
return wrapper
## Apply the wrapper to the greet function
greet = greet_wrapper(greet)
## Example usage
greet("Alice")
greet("Bob")
print(f"The greet function was called {invocation_count} times.")
在这段代码中：
- 定义了一个
greet_wrapper 函数，它接受一个函数 (func) 作为输入。
- 在
greet_wrapper 内部，定义了另一个名为 wrapper 的函数。这个 wrapper 函数是实际的包装器，当调用被包装的函数时，它将被执行。
wrapper 函数会增加 invocation_count，使用传递给它的任何参数调用原函数 (func)，并返回结果。- 然后，通过
greet = greet_wrapper(greet) 将包装器应用到 greet 函数上。这会用包装后的版本替换原 greet 函数。
使用相同的命令再次运行脚本：
python ~/project/my_function.py
你应该会看到以下输出：
Hello, Alice!
Hello, Bob!
The greet function was called 2 times.
这个输出与上一步相同，但现在你使用的是包装函数来跟踪调用。这种方法更加灵活，因为你可以轻松地将同一个包装器应用到多个函数上。
现在，让我们添加另一个函数，并将相同的包装器应用到它上面。在 my_function.py 文件中添加以下函数：
## ~/project/my_function.py
invocation_count = 0
def greet(name):
"""This function greets the person passed in as a parameter."""
print(f"Hello, {name}!")
def add(x, y):
"""This function adds two numbers and returns the result."""
print(f"Adding {x} and {y}")
return x + y
def greet_wrapper(func):
"""This is a wrapper function that tracks the number of times the wrapped function is called."""
def wrapper(*args, **kwargs):
global invocation_count
invocation_count += 1
result = func(*args, **kwargs)
return result
return wrapper
## Apply the wrapper to the greet function
greet = greet_wrapper(greet)
add = greet_wrapper(add)
## Example usage
greet("Alice")
greet("Bob")
add(5, 3)
print(f"The greet function was called {invocation_count} times.")
你添加了一个 add 函数，并将 greet_wrapper 也应用到了它上面。现在，invocation_count 将跟踪 greet 和 add 两个函数的总调用次数。
再次运行脚本：
python ~/project/my_function.py
你应该会看到类似以下的输出：
Hello, Alice!
Hello, Bob!
Adding 5 and 3
The greet function was called 3 times.
如你所见，invocation_count 现在反映了 greet 和 add 两个函数的总调用次数。包装函数提供了一种简洁且可复用的方式来监控函数调用。
