Python无参装饰器

Python无参装饰器

需求:想要在test_func函数前后执行一些代码

  1. 第一步(定义函数,将调用原函数,使用新函数替换)
def test_func():
    return 'test_func'

def test_wrapper_func():
    print 'before'
    rt = test_func()
    print 'after'
    return rt
    
print test_wrapper_func()

2.第二 步(原函数可以定义参数)

def test_func():
    return 'test_func'

def test_wrapper_func(*args, **kwargs):
    print 'before'
    rt = test_func(*args, **kwargs)
    print 'after'
    return rt
    
  
print test_wrapper_func()

3.第三步(定义工厂函数,返回wapper函数)

def test_func():
    return 'test_func'


def test_wrapper_func(*args, **kwargs):
    print 'before'
    rt = test_func(*args, **kwargs)
    print 'after'
    return rt

def test_wrapper():
    return test_wrapper_func


test_wrapper_func_temp = test_wrapper()

print test_wrapper_func_temp == test_wrapper_func
print test_wrapper_func_temp()
print test_wrapper_func()

4.第四步(去掉共有的wapper函数,放在工厂函数内部)

def test_func():
    return 'test_func'


def test_wrapper():
    def wrapper(*args, **kwargs):
        print 'before'
        rt = test_func(*args, **kwargs)
        print 'after'
        return rt    
    return wrapper


test_wrapper_func = test_wrapper()
print test_wrapper_func()

5.第五步(可以为所有函数添加同一功能前后执行代码)

def test_func()
    return 'test_func'

def test_wrapper(func):
    def wrapper(*args, **kwargs):
        print 'before'
        rt = func(*args, **kwargs)
        print 'after'
        return rt
    return wrapper


test_wrapper_func = test_wrapper(test_func)
print test_wrapper_func()

6.第六步(使用装饰器)

def test_wrapper(func):
    def wrapper(*args, **kwargs):
        print 'before'
        rt = func(*args, **kwargs)
        print 'after'
        return rt    
    return wrapper
    

@test_wrapper
def test_func():
    return 'test_func'


print test_func()

7.第七步(修正多个装饰器时,使用函数fun名称问题)

from functools import wraps

def test_wrapper(func):    
    
    @wraps(func)
    def wrapper(*args, **kwargs):
        print 'before'
        rt = func(*args, **kwargs)
        print 'after'
        return rt    
    return wrapper
    
    
@test_wrapper
def test_func():
    return 'test_func'


print test_func()
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容