需求:想要在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.第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.第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()