使用Python的装饰器实现功能增强
在Python中,装饰器是一种特殊类型的函数,它可以在不修改原函数代码的情况下,给原函数增加新的功能。
以下是一个简单的装饰器例子,该装饰器用于计算函数执行的时间:
import time
def timer_decorator(func):
def wrapper(*args, **kwargs):
start_time = time.time()
result = func(*args, **kwargs))
end_time = time.time()
print(f"Function {func.__name__}} took {end_time - start_time} seconds to execute.")
return result
return wrapper
@timer_decorator
def slow_function(n):
time.sleep(n)
return n
print(slow_function(2)) # 输出函数执行时间
在这个例子中,timer_decorator
是一个装饰器,它接受一个函数作为参数,并返回一个新的函数。新函数在执行原功能的同时,还会记录并打印出执行时间。
还没有评论,来说两句吧...