谈到python装饰器，一般都先提到函数式编程，但是其实运用python装饰器，和知不知道他是函数式编程没啥关系。
例子
1 2 3 4 5 6 7 8 9 10 11 | def decorator(func):
def wrapper(*args, **kwargs):
print('我是装饰器呦')
return func(*args, **kwargs)
return wrapper
@decorator
def test():
print('我是test')
test()
|
输出如下
其实就这么简单，完全可以把代码里的@理解为我们聊天中用的@，在调用某一个函数的时候@你想调用的装饰器函数，然后就可以在函数自身代码不变的情况下、增添一些具有普适性的功能。
应用
那么装饰器能做什么呢？最简单的一个应用，计算函数运行时间啊。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import time
def run_time(func):
def wrapper(*args, **kw):
t = time.time()
ans = func(*args, **kw)
t = time.time() - t
return ans, t
return wrapper
@run_time
def test():
x=1
for i in range(10 ** 6):
x += 1
return x
test()
|
输出如下
1 | (1000001, 0.08555388450622559)
|
当然，装饰器的应用不止如此，它还可以传递参数，著名的用法比如web框架flask的路由配置，有兴趣可以阅读一下它的源码，很漂亮。
所需基础
为了彻底理解装饰器，有两个基础知识要巩固一下。一个是闭包函数，一个是可变参数。
补充
我们给上面的装饰器加一个参数:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | import time
def run_time(text):
def decorator(func):
def wrapper(*args, **kwargs):
t = time.time()
ans = func(*args, **kwargs)
t = time.time() - t
if t>text:
return ans, '太慢'
else:
return ans, '正常'
return wrapper
return decorator
@run_time(1)
def test():
x=1
for i in range(10 ** 6):
x += 1
return x
test( )
|
