pythondecoratorwrapper_python的装饰器(Decorator)

先看一个例子

#这个是一个闭包&＃xff0c;闭包前面已经说了

deflog(func):def wrapper(*args, **kw):print(&＃39;call %s():&＃39; % func.__name__)return func(*args, **kw)returnwrapper

&＃64;logdefnow():print(&＃39;2015-3-25&＃39;)#调用函数

now()#输出

call now():2015-3-25

在我们调用这个函数的时候&＃xff0c;第一件事并不是执行这个函数&＃xff0c;而是将这个函数做为参数传入它头顶上这顶帽子&＃xff0c;这顶帽子我们称之为装饰函数 或 装饰器&＃xff0c;如上面的log()函数

装饰器的使用方法&＃xff1a;

1.先定义一个装饰器(帽子)&＃xff0c;可以用类或者函数实现

2.再定义你的业务函数或者类

3.最后把帽子戴在这个函数头上

上面的例子log()函数是一个装饰器&＃xff0c;返回的是一个函数&＃xff0c;如果装饰器本身需要传入参数&＃xff0c;则需要定义一个三层嵌套的函数

如&＃xff1a;

deflog(text):defdecorator(func):def wrapper(*args, **kw):print(&＃39;%s %s():&＃39; % (text, func.__name__)) #函数真正执行的地方

return func(*args, **kw)returnwrapperreturndecorator

&＃64;log(&＃39;execute&＃39;) #log函数传入参数&＃39;execute&＃39;

defnow():print(&＃39;2015-3-25&＃39;)

now()#输出

execute now():2015-3-25

我们来剖析上面的语句&＃xff0c;首先执行log(&＃39;execute&＃39;)&＃xff0c;返回的是decorator函数&＃xff0c;再调用返回的函数&＃xff0c;参数是now函数&＃xff0c;返回值最终是wrapper函数

使用funltools标准库中的wraps装饰器

#使用函数的属性__name__

now.__name__

#输出

&＃39;wrapper&＃39;

函数对象有一个__name__属性&＃xff0c;可以拿到函数的名字&＃xff0c;为啥返回值是wrapper&＃xff0c;这是因为不管怎么嵌套&＃xff0c;最终还是回到最里面那层&＃xff0c;为了避免理解错误&＃xff0c;就使用wraps装饰器

functools .wraps 装饰器&＃xff0c;它的作用就是将 被修饰的函数(now) 的一些属性值赋值给 修饰器函数(log) &＃xff0c;最终让属性的显示更符合我们的直觉

importfunctoolsdeflog(func):

&＃64;functools.wraps(func)def wrapper(*args, **kw):print(&＃39;call %s():&＃39; % func.__name__)return func(*args, **kw)returnwrapper#带参数的装饰器

importfunctoolsdeflog(text):defdecorator(func):

&＃64;functools.wraps(func)def wrapper(*args, **kw):print(&＃39;%s %s():&＃39; % (text, func.__name__))return func(*args, **kw)returnwrapperreturn decorator

wraps 其实是一个偏函数对象(partial)&＃xff0c;源码如下

defwraps(wrapped,

assigned&＃61;WRAPPER_ASSIGNMENTS,

updated&＃61;WRAPPER_UPDATES):return partial(update_wrapper, wrapped&＃61;wrapped,

assigned&＃61;assigned, updated&＃61;updated)

练习&＃xff1a;请设计一个decorator&＃xff0c;它可作用于任何函数上&＃xff0c;并打印该函数的执行时间&＃xff1a;

importtime, functoolsdefmetric(fn):

&＃64;functools.wraps(fn)def wrapper(*args, **kw):

start_time&＃61;time.time()

e&＃61; fn(*args, **kw)

end_time&＃61;time.time()print(&＃39;%s executed in %s ms&＃39; % (fn.__name__, end_time-start_time))returnereturnwrapper

&＃64;metricdeffast(x, y):

time.sleep(0.0012)return x &＃43;y;

&＃64;metricdefslow(x, y, z):

time.sleep(0.1234)return x * y *z;

f&＃61; fast(11, 22)

s&＃61; slow(11, 22, 33)if f !&＃61; 33:print(&＃39;测试失败!&＃39;)elif s !&＃61; 7986:print(&＃39;测试失败!&＃39;)

类的装饰器后面在聊