使用Python的装饰器实现功能增强

原创 朱雀 2025-02-10 17:33 60阅读 0赞

在Python中,装饰器是一种特殊类型的函数,它可以在不修改原函数代码的情况下,给原函数增加新的功能。

以下是一个简单的装饰器例子,该装饰器用于计算函数执行的时间:

  1. import time
  2. def timer_decorator(func):
  3. def wrapper(*args, **kwargs):
  4. start_time = time.time()
  5. result = func(*args, **kwargs))
  6. end_time = time.time()
  7. print(f"Function {func.__name__}} took {end_time - start_time} seconds to execute.")
  8. return result
  9. return wrapper
  10. @timer_decorator
  11. def slow_function(n):
  12. time.sleep(n)
  13. return n
  14. print(slow_function(2)) # 输出函数执行时间

在这个例子中,timer_decorator是一个装饰器,它接受一个函数作为参数,并返回一个新的函数。新函数在执行原功能的同时,还会记录并打印出执行时间。

文章版权声明:注明蒲公英云原创文章,转载或复制请以超链接形式并注明出处。

发表评论

表情:
评论列表 (有 0 条评论,60人围观)

还没有评论,来说两句吧...

相关阅读