装饰器
- 定义:就是给已有函数增加额外功能的函数,它本质上就是一个闭包函数。
def 装饰器函数(原函数fn):
def inner():
'''执行函数之前'''
fn() // 执行被装饰的目标函数
'''执行函数之后'''
return inner
@装饰器函数
def f1():
pass
f1 = 装饰器函数(f1) 等价于 函数上写@装饰器函数
- 作用:实现代码的重用
- 特点:
- 不修改已有函数的源代码;
- 不修改已有函数的调用方式;
- 给已有函数增加额外的功能。
语法糖
- 如果又多个函数都需要添加某种功能,每次都需要
func = out_func(func)
这样的代码对已有函数进行装饰很麻烦。 Python
提供了一个装饰函数更加简单的写法,那就是语法糖,语法糖的书写格式是:@装饰器名字
,使用“@
”语法糖精简装饰器的代码的同时也可以完成对已有函数的装饰。- 说明:
@check
等价于comment = check(comment)
- 装饰器的执行时间是加载模块时立即执行。
通用装饰器
- 需求:上面的普通装饰器,原函数既及不带参数也不带返回值,但是怎么设计可以不管函数有没有参数、返回值都可以兼容?
def 装饰器函数(原函数引用fn):
def inner(*args, **kwargs):
// 执行前扩展功能
result = fn(*args, **kwargs)
// 执行后扩展功能
return result
return inner
@装饰器函数
def fn(arg1, arg2,...):
pass
return xxxx