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

Python  /  管理员 发布于 7年前   138

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结束 ...

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