python函数修饰符@的使用方法解析

时间:2021-05-22

python函数修饰符@的作用是为现有函数增加额外的功能,常用于插入日志、性能测试、事务处理等等。

创建函数修饰符的规则:

(1)修饰符是一个函数

(2)修饰符取被修饰函数为参数

(3)修饰符返回一个新函数

(4)修饰符维护被维护函数的签名

例子1:被修饰函数不带参数

def log(func): def wrapper(): print('log开始 ...') func() print('log结束 ...') return wrapper@logdef test(): print('test ..')test()

运行结果:

log开始 ...test ..log结束 ...

例子2:使用functools模块提供的修改函数属性的方法wraps

def log(func): def wrapper(): print('log开始 ...') func() print('log结束 ...') return wrapper@logdef test1(): print('test1 ..')def test2(): print('test2 ..')print(test1.__name__)print(test2.__name__)

运行结果:

wrappertest2

可见test1的函数名称变了,如果某些代码用到就会出问题,可以使用functools模块提供的修改函数属性的方法wraps

from functools import wrapsdef log(func): @wraps(func) def wrapper(): print('log开始 ...') func() print('log结束 ...') return wrapper@logdef test1(): print('test1 ..')def test2(): print('test2 ..')print(test1.__name__)print(test2.__name__)

运行结果:

test1test2

例子3:被修饰函数带参数

from functools import wrapsdef log(func): @wraps(func) def wrapper(*args,**kwargs): print('log开始 ...',func.__name__) ret = func(*args,**kwargs) print('log结束 ...') return ret return wrapper@logdef test1(s): print('test1 ..', s) return s@logdef test2(s1, s2): print('test2 ..', s1, s2) return s1 + s2test1('a')test2('a','bc')

运行结果:

log开始 ... test1test1 .. alog结束 ...log开始 ... test2test2 .. a bclog结束 ...

例子4:修饰符带参数,需要比上面例子多一层包装

from functools import wrapsdef log(arg): def _log(func): @wraps(func) def wrapper(*args,**kwargs): print('log开始 ...',func.__name__, arg) ret = func(*args,**kwargs) print('log结束 ...') return ret return wrapper return _log @log('module1')def test1(s): print('test1 ..', s) return s@log('module1')def test2(s1, s2): print('test2 ..', s1, s2) return s1 + s2test1('a')test2('a','bc')

运行结果:

log开始 ... test1 module1test1 .. alog结束 ...log开始 ... test2 module1test2 .. a bclog结束 ...

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章