标签:
实现装饰器的最主要的原因是python中一切皆为对象,我们会把方法看做一个对象包装起来
首相,我们先来看一个简单的例子,有一个add方法:
def add(x, y): print(‘result is {}‘.format(x + y))
现在有一个新的需求,我们想在这个方法执行完之后print一句话证明他执行成功了,同样我们又不想修改这个add方法,那我们可以新建一个方法:
def printres(func): func() print(‘Done‘) printres(add)
那这样是ok的。但是这样的话,我们每次都要将一个函数作为参数传递给printres函数。而且这种方式已经破坏了原有的代码逻辑结构,之前执行业务逻辑时,执行运行add(),但是现在不得不改成printres(add)。那么有没有更好的方式的呢?当然有,答案就是装饰器。我就直接贴代码:
# coding=utf-8 """ 学习装饰器 2016-08-20 by oldman """ from functools import wraps """ 先来看一个简单的装饰器 """ def decor1(func): @wraps(func) def wrapper(*args, **kwargs): func(*args, **kwargs) print(‘i am a decorate‘) return wrapper @decor1 def add(x, y): print(‘result is {}‘.format(x + y)) # if __name__ =="__main__": # add(2,3) """ 再来看一个带参数的装饰器 """ def decor2(parm): """ parm为装饰器用到的参数 """ def _decor2(func): # func为被装饰的函数 @wraps(func) def wrapper(*args, **kwargs): if parm: func(*args) print(‘I am true‘) else: print(‘sorry, false‘) return wrapper return _decor2 @decor2(True) # 这里装饰器的参数为True def add1(x, y): print(‘result is {}‘.format(x + y)) if __name__ == ‘__main__‘: add1(2, 3)
注意到functools.wraps,wraps本身也是一个装饰器,它能把原函数的元信息拷贝到装饰器函数中,这使得装饰器函数也有和原函数一样的元信息了,如果不在装饰器里使用@wrap的话,被装饰的函数的元信息会丢失。
这是一个很好的解决方案,不得不感叹python的灵活,人生苦短,我用python。
参考文章:http://www.zhihu.com/question/26930016
学习交流群 :226704167
python cookbook 学习系列(一) python中的装饰器
标签:
原文地址:http://www.cnblogs.com/lip0121/p/5792176.html