多应用+插件架构,代码干净,二开方便,首家独创一键云编译技术,文档视频完善,免费商用码云13.8K 广告
[TOC] # 简介 假设我们要增强now()函数的功能,比如,在函数调用前后自动打印日志,但又不希望修改now()函数的定义,这种在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)。 本质上,decorator就是一个返回函数的高阶函数。所以,我们要定义一个能打印日志的decorator,可以定义如下: ~~~ def log(func): def wrapper(*args, **kw): print('call %s():' % func.__name__) return func(*args, **kw) return wrapper ~~~ 观察上面的log,因为它是一个decorator,所以接受一个函数作为参数,并返回一个函数。我们要借助Python的@语法,把decorator置于函数的定义处: ~~~ @log def now(): print('2015-3-25') ~~~ 调用now()函数,不仅会运行now()函数本身,还会在运行now()函数前打印一行日志: ~~~ >>> now() call now(): 2015-3-25 ~~~ 把`@log`放到now()函数的定义处,相当于执行了语句: ~~~ now = log(now) ~~~ # 多个装饰器 外函数由下面先执行,内函数由上面先执行 ~~~ def add_qx(func): print('---开启进行装饰权限1的功能---') def call_func(*args, **kwargs): print('---这是权限验证1---') return func(*args, **kwargs) return call_func def add_xx(func): print('---开启进行装饰xx的功能---') def call_func(*args, **kwargs): print('---这是xx的功能---') return func(*args, **kwargs) return call_func @add_qx @add_xx def test1(): print('test1---') test1() ~~~ 输出 ~~~ ---开启进行装饰xx的功能--- ---开启进行装饰权限1的功能--- ---这是权限验证1--- ---这是xx的功能--- test1--- ~~~ # 类装饰 ~~~ class Test(object): def __init__(self, func): self.func = func def __call__(self): print('这里是装饰器添加的功能') return self.func() @Test def get_str(): return 'hello' print(get_str()) ~~~ 输出 ~~~ 这里是装饰器添加的功能 hello ~~~ # 装饰器传参数 ~~~ def log(text): def decorator(func): def wrapper(*args, **kwargs): print('%s %s():' % (text, func.__name__)) return func(*args, **kwargs) return wrapper return decorator @log('execute') def now(): print('2015-3-25') ~~~ 执行结果如下: ~~~ >>> now() execute now(): 2015-3-25 ~~~ 和两层嵌套的decorator相比,3层嵌套的效果是这样的: ~~~ >>> now = log('execute')(now) ~~~