python 装饰器用法

pythont天坑

装饰器

• 定义装饰器

def demo(func):
    print("1111111")

    def wrapper(*args, **kwargs):
        print("2222222")
        return func()
    print("3333333")
    return wrapper

• 使用装饰器

@demo
def test_wrapper():
    """测试装饰器"""
    print("4444444")
    pass

• 装饰器执行流程

1111111
3333333
2222222
4444444

• 使用装饰器注意点
  • 未使用装饰器前，函数对象名字及文档打印结果如下

    print(test_wrapper.__name__)
    print(test_wrapper.__doc__)
    
    test_wrapper
    测试装饰器
  

  • 使用装饰器会改变函数对象的一些特性，名字以及文档

    @demo
    def test_wrapper():
        """测试装饰器"""
        pass
    
    
    # test_wrapper()
    print(test_wrapper.__name__)
    print(test_wrapper.__doc__)
    
    wrapper
    None
  
    
  

  • 引入functools中的wraps解决

    from functools import wraps
    def demo(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            return func()
        return wrapper
  
  
    @demo
    def test_wrapper():
        """测试装饰器"""
        pass
    
    
    # test_wrapper()
    print(test_wrapper.__name__)
    print(test_wrapper.__doc__)
    
    
    运行结果
    test_wrapper
    测试装饰器