• 9.3 解除一个装饰器
    • 问题
    • 解决方案
    • 讨论

    9.3 解除一个装饰器

    问题

    一个装饰器已经作用在一个函数上,你想撤销它,直接访问原始的未包装的那个函数。

    解决方案

    假设装饰器是通过 @wraps (参考9.2小节)来实现的,那么你可以通过访问 wrapped 属性来访问原始函数:

    1. >>> @somedecorator
    2. >>> def add(x, y):
    3. ... return x + y
    4. ...
    5. >>> orig_add = add.__wrapped__
    6. >>> orig_add(3, 4)
    7. 7
    8. >>>

    讨论

    直接访问未包装的原始函数在调试、内省和其他函数操作时是很有用的。但是我们这里的方案仅仅适用于在包装器中正确使用了 @wraps 或者直接设置了 wrapped 属性的情况。

    如果有多个包装器,那么访问 wrapped 属性的行为是不可预知的,应该避免这样做。在Python3.3中,它会略过所有的包装层,比如,假如你有如下的代码:

    1. from functools import wraps
    2.  
    3. def decorator1(func):
    4. @wraps(func)
    5. def wrapper(*args, **kwargs):
    6. print('Decorator 1')
    7. return func(*args, **kwargs)
    8. return wrapper
    9.  
    10. def decorator2(func):
    11. @wraps(func)
    12. def wrapper(*args, **kwargs):
    13. print('Decorator 2')
    14. return func(*args, **kwargs)
    15. return wrapper
    16.  
    17. @decorator1
    18. @decorator2
    19. def add(x, y):
    20. return x + y

    下面我们在Python3.3下测试:

    1. >>> add(2, 3)
    2. Decorator 1
    3. Decorator 2
    4. 5
    5. >>> add.__wrapped__(2, 3)
    6. 5
    7. >>>

    下面我们在Python3.4下测试:

    1. >>> add(2, 3)
    2. Decorator 1
    3. Decorator 2
    4. 5
    5. >>> add.__wrapped__(2, 3)
    6. Decorator 2
    7. 5
    8. >>>

    最后要说的是,并不是所有的装饰器都使用了 @wraps ,因此这里的方案并不全部适用。特别的,内置的装饰器 @staticmethod@classmethod 就没有遵循这个约定(它们把原始函数存储在属性 func 中)。

    原文:

    http://python3-cookbook.readthedocs.io/zh_CN/latest/c09/p03_unwrapping_decorator.html