python函数修饰符@的使用方法解析
python函数修饰符@的作用是为现有函数增加额外的功能，常用于插入日志、性能测试、事务处理等等。
创建函数修饰符的规则：
（1）修饰符是一个函数
（2）修饰符取被修饰函数为参数
（3）修饰符返回一个新函数
（4）修饰符维护被维护函数的签名
例子1：被修饰函数不带参数
deflog(func):
defwrapper():
print('log开始...')
func()
print('log结束...')
returnwrapper
@log
deftest():
print('test..')
test()
运行结果：
log开始... test.. log结束...
例子2：使用functools模块提供的修改函数属性的方法wraps
deflog(func):
defwrapper():
print('log开始...')
func()
print('log结束...')
returnwrapper
@log
deftest1():
print('test1..')
deftest2():
print('test2..')
print(test1.__name__)
print(test2.__name__)
运行结果：
wrapper test2
可见test1的函数名称变了，如果某些代码用到就会出问题，可以使用functools模块提供的修改函数属性的方法wraps
fromfunctoolsimportwraps
deflog(func):
@wraps(func)
defwrapper():
print('log开始...')
func()
print('log结束...')
returnwrapper
@log
deftest1():
print('test1..')
deftest2():
print('test2..')
print(test1.__name__)
print(test2.__name__)
运行结果：
test1 test2
例子3：被修饰函数带参数
fromfunctoolsimportwraps
deflog(func):
@wraps(func)
defwrapper(*args,**kwargs):
print('log开始...',func.__name__)
ret=func(*args,**kwargs)
print('log结束...')
returnret
returnwrapper
@log
deftest1(s):
print('test1..',s)
returns
@log
deftest2(s1,s2):
print('test2..',s1,s2)
returns1+s2
test1('a')
test2('a','bc')
运行结果：
log开始...test1 test1..a log结束... log开始...test2 test2..abc log结束...
例子4：修饰符带参数，需要比上面例子多一层包装
fromfunctoolsimportwraps
deflog(arg):
def_log(func):
@wraps(func)
defwrapper(*args,**kwargs):
print('log开始...',func.__name__,arg)
ret=func(*args,**kwargs)
print('log结束...')
returnret
returnwrapper
return_log
@log('module1')
deftest1(s):
print('test1..',s)
returns
@log('module1')
deftest2(s1,s2):
print('test2..',s1,s2)
returns1+s2
test1('a')
test2('a','bc')
运行结果：
log开始...test1module1 test1..a log结束... log开始...test2module1 test2..abc log结束...
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。