谈到python装饰器,一般都先提到函数式编程,但是其实运用python装饰器,和知不知道他是函数式编程没啥关系。
例子
def decorator(func): def wrapper(*args, **kwargs): print('我是装饰器呦') return func(*args, **kwargs) return wrapper @decorator def test(): print('我是test') test()
输出如下
我是装饰器呦 我是test
其实就这么简单,完全可以把代码里的@理解为我们聊天中用的@,在调用某一个函数的时候@你想调用的装饰器函数,然后就可以在函数自身代码不变的情况下、增添一些具有普适性的功能。
应用
那么装饰器能做什么呢?最简单的一个应用,计算函数运行时间啊。
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()
输出如下
(1000001, 0.08555388450622559)
当然,装饰器的应用不止如此,它还可以传递参数,著名的用法比如web框架flask的路由配置,有兴趣可以阅读一下它的源码,很漂亮。
所需基础
为了彻底理解装饰器,有两个基础知识要巩固一下。一个是闭包函数,一个是可变参数。
补充
我们给上面的装饰器加一个参数:
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( )
Copyright © 2019- ovod.cn 版权所有 湘ICP备2023023988号-4
违法及侵权请联系:TEL:199 1889 7713 E-MAIL:2724546146@qq.com
本站由北京市万商天勤律师事务所王兴未律师提供法律服务