装饰器本质是一个python函数,它可以让其他函数在不做任何代码改动的前提下增加额外功能,装饰器返回值也是一个函数对象,它经常用于有切面要求的场景:插入日志,性能测试,事务处理,权限校验等。有了装饰器我们可以抽离出大量与函数功能本身无关的雷同代码继续重用。
def foo():
print('aaa')
现在有一个新的需求,希望记录下函数执行的日志,于是在代码中添加日志代码:
def foo():
print('aaa')
logging.info("foo is running")
假如 foo1()、foo2()也有相同的需求,每个函数都需要加相同的代码,为了减少重写的代码,第一种方法:重新定义一个函数,专门处理日志,之后再执行真正的业务代码
def use_logging(func):
logging.info("%s is running"% func.__name__)
func()
def foo():
print('aaa')
use_logging(bar)
缺点:每次都要将一个函数作为参数传递给use_logging函数。而且这种方式破坏了原来的逻辑性,原本执行业务逻辑:执行foo(),但现在要改为执行use_logging(bar)
方法二:装饰器
def use_logging(func):
def wrapper(*args,**kw):
logging.info("%s is running"%func.__name__)
return func(*args)
return wrapper
@use_logging
def foo():
print('aaa')
use_logging就是装饰器,它把真正执行业务的代码的包裹在函数里,看起来foo被use_logging装饰了,在这个例子中函数进入和退出时,被称为一个横切面,这种编程方式被称为面向切面的编程(Aspect-Orineted Programmign)。