Python無參裝飾器
需求：想要在test_func函數(shù)前后執(zhí)行一些代碼
- 第一步(定義函數(shù)，將調(diào)用原函數(shù)，使用新函數(shù)替換)
def test_func():
return 'test_func'
def test_wrapper_func():
print 'before'
rt = test_func()
print 'after'
return rt
print test_wrapper_func()
2.第二 步(原函數(shù)可以定義參數(shù))
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.第三步(定義工廠函數(shù)，返回wapper函數(shù))
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函數(shù)，放在工廠函數(shù)內(nèi)部)
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.第五步(可以為所有函數(shù)添加同一功能前后執(zhí)行代碼)
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.第七步（修正多個裝飾器時，使用函數(shù)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()
