Python装饰器是一种在不修改原始函数代码的情况下,为函数增加新功能的方法。它们通过在原始函数前添加一个包装函数来实现这一目的。装饰器可以简化代码,提高代码的可读性和可维护性。
以下是一个简单的装饰器示例:
def my_decorator(func): def wrapper(): print("Something is happening before the function is called.") func() print("Something is happening after the function is called.") return wrapper @my_decorator def say_hello(): print("Hello!") say_hello()
在这个例子中,my_decorator
是一个装饰器函数,它接受一个函数 func
作为参数。wrapper
函数是实际的包装函数,它在调用原始函数之前和之后执行一些额外的操作。通过使用 @my_decorator
语法,我们将 say_hello
函数与 my_decorator
装饰器关联起来。
当调用 say_hello()
时,实际上是在调用 wrapper
函数,因此会看到以下输出:
Something is happening before the function is called. Hello! Something is happening after the function is called.
装饰器可以简化代码,例如:
def log_time(func):
import time
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs)
end_time = time.time()
print(f"{func.__name__} took {end_time - start_time:.2f} seconds to execute.")
return result
return wrapper
@log_time
def my_function():
time.sleep(2)
print("Function executed.")
my_function()
在这个例子中,log_time
装饰器用于记录函数的执行时间。通过使用装饰器,我们可以轻松地为其他函数添加相同的日志功能,而无需修改它们的代码。