一个例子解释python装饰器

一个例子解释python装饰器

谈到python装饰器,一般都先提到函数式编程,但是其实运用python装饰器,和知不知道他是函数式编程没啥关系。

例子

defdecorator(func):
defwrapper(*args,**kwargs):
print('我是装饰器呦')
returnfunc(*args,**kwargs)
returnwrapper

@decorator
deftest():
print('我是test')

test()

输出如下

我是装饰器呦
我是test

其实就这么简单,完全可以把代码里的@理解为我们聊天中用的@,在调用某一个函数的时候@你想调用的装饰器函数,然后就可以在函数自身代码不变的情况下、增添一些具有普适性的功能。

应用

那么装饰器能做什么呢?最简单的一个应用,计算函数运行时间啊。

importtime
defrun_time(func):
defwrapper(*args,**kw):
t=time.time()
ans=func(*args,**kw)
t=time.time()-t
returnans,t
returnwrapper

@run_time
deftest():
x=1
foriinrange(10**6):
x+=1
returnx

test()

输出如下

(1000001,0.08555388450622559)

当然,装饰器的应用不止如此,它还可以传递参数,著名的用法比如web框架flask的路由配置,有兴趣可以阅读一下它的源码,很漂亮。

所需基础

为了彻底理解装饰器,有两个基础知识要巩固一下。一个是闭包函数,一个是可变参数。

补充

importtime
defrun_time(text):
defdecorator(func):
defwrapper(*args,**kwargs):
t=time.time()
ans=func(*args,**kwargs)
t=time.time()-t
ift>text:
returnans,'太慢'
else:
returnans,'正常'
returnwrapper
returndecorator

@run_time(1)
deftest():
x=1
foriinrange(10**6):
x+=1
returnx

test()