Python修饰器是一种非常强大的工具,它允许你在不修改原有函数定义的情况下,给函数添加额外的功能。修饰器本质上是一个函数,它接收一个函数作为参数,并返回一个新的函数。这使得你可以在原有函数前后附加代码,实现诸如日志记录、性能测试、事务处理、缓存等功能。接下来,我将通过一些例程逐步讲解修饰器的使用方法和原理。
1. 简单修饰器
首先,我们创建一个修饰器,让它在函数调用前后打印一些信息:
# 定义一个修饰器
def simple_decorator(func):
# 定义一个包装函数,它将被用来替换(或装饰)原始函数
def wrapper():
print("Before calling the function.")
a = func() # 调用原始函数
print("After calling the function get:{}".format(a))
return wrapper # 返回包装函数
# 使用刚才定义的修饰器来装饰这个简单的函数
@simple_decorator
def say_hello():
print("Hello!")
return 1
# 调用被装饰的函数
say_hello()
输出:
Before calling the function.
Hello!
After calling the function get:1
在这个例子中,simple_decorator是一个修饰器,它接收一个函数func作为参数。wrapper()是一个内嵌函数,它在func被调用前后添加了一些额外的操作。通过在say_hello()函数定义前使用@simple_decorator语法,我们实际上是将say_hello()函数作为参数传递给了simple_decorator,并将返回的新函数wrapper()赋给了say_hello()。因此,当我们调用say_hello()时,实际上是在调用wrapper()。
2. 带参数的修饰器
修饰器也可以接收参数。这需要在外面再包一层函数,用来接收这些参数:
# 定义一个接收参数的修饰器
def repeat(num_times):
def decorator_repeat(func):
def wrapper(*args, **kwargs