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

短命女 2023-07-07 11:14 7阅读 0赞

这篇文章主要介绍了python函数修饰符@的使用方法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下  
python函数修饰符@的作用是为现有函数增加额外的功能，常用于插入日志、性能测试、事务处理等等。

创建函数修饰符的规则：  
（1）修饰符是一个函数  
（2）修饰符取被修饰函数为参数  
（3）修饰符返回一个新函数  
（4）修饰符维护被维护函数的签名

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

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

运行结果：

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

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

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

运行结果：

wrapper
    test2

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

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

运行结果：

test1
    test2

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

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

运行结果：

log开始 ... test1
    test1 .. a
    log结束 ...
    log开始 ... test2
    test2 .. a bc
    log结束 ...

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

from functools import wraps
     
    def 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 + s2
    test1('a')
    test2('a','bc')

运行结果：

log开始 ... test1 module1
    test1 .. a
    log结束 ...
    log开始 ... test2 module1
    test2 .. a bc
    log结束 ...

推荐我们的python学习基地，看前辈们是如何学习的！从基础的python脚本、爬虫、django、数据挖掘等编程技术，还有整理零基础到项目实战的资料，送给每一位爱学习python的小伙伴！每天都有老前辈定时讲解Python技术，分享一些学习的方法和需要留意的小细节，点击加入我们的 [python学习者聚集地][python]

[python]: https://jq.qq.com/?_wv=1027&k=5ipV5It