[python]装饰器及其使用

心已赠人 2022-03-15 10:08 263阅读 0赞

装饰器

  • 最简装饰器

    1. def deco(func):
    2.     def wrap(*args, **kwargs):
    3.         return func(*args, **kwargs)
    4.     return wrap
    6. @deco
    7. def foo(a, b):
    8.     return a ** b

  • 原理

    • 对比被装饰前后的 foo.__name__ 和 foo.__doc__

      1. from functools import wraps
      3. def deco(func):
      4.     '''i am deco'''
      5.     @wraps(func)  # 还原被装饰器修改的原函数属性
      6.     def wrap(*args, **kwargs):
      7.         '''i am wrap'''
      8.         return func(*args, **kwargs)
      9.     return wrap

    • 简单过程

      1. fn = deco(func)
      2. foo = fn
      3. foo(*args, **kwargs)

    • 多个装饰器叠加调用的过程

      1. @deco1
      2. @deco2
      3. @deco3
      4. def foo(x, y):
      5.     return x ** y
      7. # 过程拆解 1
      8. fn3 = deco3(foo)
      9. fn2 = deco2(fn3)
      10. fn1 = deco1(fn2)
      11. foo = fn1
      12. foo(3, 4)
      14. # 过程拆解 2
      15. # 单行: deco1( deco2( deco3(foo) ) )(3, 2)
      16. deco1(
      17.     deco2(
      18.         deco3(foo)
      19.     ) 
      20. )(3, 4)

  • 带参数的装饰器

    1. def deco(n):
    2.     def wrap1(func):
    3.         def wrap2(*args, **kwargs):
    4.             return func(*args, **kwargs)
    5.         return wrap2
    6.     return wrap1
    8. # 调用过程
    9. wrap1 = deco(n)
    10. wrap2 = wrap1(foo)
    11. foo = wrap2
    12. foo()
    14. # 单行形式
    15. check_result(30)(foo)(4, 8)

  • 装饰器类和 __call__

    1. class Deco:
    2.     def __init__(self, func):
    3.         self.func = func
    5.     def __call__(self, *args, **kwargs):
    6.         return self.func(*args, **kwargs)
    8. @Deco
    9. def foo(x, y):
    10.     return x ** y
    12. # 过程拆解
    13. fn = Deco(foo)
    14. foo = fn
    15. foo(12, 34)

  • 使用场景

    • 参数、结果检查
    • 缓存、计数
    • 日志、统计
    • 权限管理
    • 重试
    • 其他

  • 例1: 写一个 timer 装饰器, 计算出被装饰函数调用一次花多长时间, 并把时间打印出来

    1. import time
    2. from functools import wraps
    4. def timer(func):
    5.     @wraps(func)  # 修正 docstring
    6.     def wrap(*args, **kwargs):
    7.         time0 = time.time()
    8.         result = func(*args, **kwargs)
    9.         time1 = time.time()
    10.         print(time1 - time0)
    11.         return result
    12.     return wrap

  • 例2: 写一个 Retry 装饰器

    1. import time
    3. class retry(object):
    4.     def __init__(self, max_retries=3, wait=0, exceptions=(Exception,)):
    5.         self.max_retries = max_retries
    6.         self.exceptions = exceptions
    7.         self.wait = wait
    9.     def __call__(self, func):
    10.         def wrapper(*args, **kwargs):
    11.             for i in range(self.max_retries + 1):
    12.                 try:
    13.                     result = func(*args, **kwargs)
    14.                 except self.exceptions:
    15.                     time.sleep(self.wait)
    16.                     continue
    17.                 else:
    18.                     return result
    19.         return wrapper

发表评论

表情：
评论列表 （有 0 条评论，263人围观）

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

相关阅读

    相关 Python 装饰器如何使用

    装饰器（Decorator） 是 Python 里的一种特殊工具，它为我们提供了一种在函数外部修改函数的灵活能力。它有点像一顶画着独一无二 @ 符号的神奇帽子，只要将它戴在函数

    墨蓝墨蓝/ 0 赞/ 136 阅读