装饰器本身的形式是处理其他的可调用对象的可调用的对象(函数或者类)。装饰器通过自动把函数和类名重绑到其他的可调用对象来的实现效果。装饰器提供了一种方法,在函数或者类定义语句的入口或者退点插入自动运行代码(比如参数验证,计时,追踪,管理对类属性的访问等操作),在需要的时候拦截随后的函数调用或者实例创建并处理它们。
函数装饰器设计用来只增强一个特定函数或方法调用,而不是一个完整的对象接口。类装饰器更好地充当后一种角色——因为它们可以拦截实例创建调用,它们可以用来实现任意的对象接口扩展或管理任务。
- 函数装饰器在函数定义的时候进行名称重绑定,提供一层逻辑层来管理灌输和方法或随后对它们的调用
- 类装饰器在类定义的时候进行名称重绑定,提供一层逻辑层,或随后调用它们所创建的实例。
1、函数装饰器:
import time
from functools import wraps
def collectDuration(func):
@wraps(func)
def wrapper(*args,**kargs):
start = time.time()
result = func(*args,**kargs)
end = time.time()
print(func.__name__,"duration:%f"%(end-start))
return result
return wrapper
@collectDuration
def counter(n):
sum =0
while n >0:
sum +=n
n -=1
return sum
counter(1000000)
>>counter duration:0.256015
语法 @标识了装饰器
当执行:
counter(1000000)
等同于执行了如下的代码,首先将counter(1000000)对象作为参数传递给装饰函数collectDuration(),然后将该函数重新赋值为counter
counter = collectDuration(counter(1000000))
2、类装饰器:
类装饰器是管理类的一种,或者管理或者扩展类所创建方式的实例。
class CollectDuration:
def __init__(self,func):
print("inside __init__")
self.func = func
def __call__(self,*args,**kargs):
print("inside __call__")
start = time.time()
number = self.func(*args,**kargs)
end = time.time()
print(self.func.__name__,"duration:%f"%(end-start))
return number
@CollectDuration
def adder(n):
number = 0
while n > 0:
number +=n
n -= 1
return number
number = adder(10000)
>>inside __init__
>>inside __call__
>>adder duration:0.003000
执行adder(5)方法,等价执行如下的代码:
adder=CollectDuration(adder(5))
调用adder 的时候,它确实会调用装饰器所创建的实例的c a l l运算符重载方法;然后,c a l l方法可能运行最初的adder ,因为它在一个实例属性中仍然可用。当按照这种方式编写代码的时候,每个装饰的函数都会产生一个新的实例来保持状态。