裝飾器 (1)什麼是裝飾器: 器指的是工具,可以定義成函數 裝飾指的是為其他事務添加額外的東西來點綴 上面兩者合到一起: 裝飾器指的是定義一個函數,該函數用來為其他函數添加額外的功能 函數裝飾器分為: 無參裝飾器和有參裝飾兩種,二者的實現原理一樣,都是’函數嵌套+閉包+函數對象’的組合使用的產物。 ...
裝飾器
(1)什麼是裝飾器:
-
器指的是工具,可以定義成函數
-
裝飾指的是為其他事務添加額外的東西來點綴
上面兩者合到一起:
- 裝飾器指的是定義一個函數,該函數用來為其他函數添加額外的功能
函數裝飾器分為:
- 無參裝飾器和有參裝飾兩種,二者的實現原理一樣,都是’函數嵌套+閉包+函數對象’的組合使用的產物。
(2)為何要用裝飾器
開放封閉原則
開放:指的是對擴展功能是開放的
封閉:指的是對修改源代碼是封閉的
裝飾器就是在不修改被裝飾器對象的源代碼以及調用方式的前提下,為被裝飾對象添加新功能
(3)裝飾器實現思路
無參裝飾器
- 方案一:失敗
沒有修改被裝飾對象的調用方式,但是改變了源代碼
def index(x, y):
start = time.time()
time.sleep(2)
print(' index %s %s ' % (x, y))
end = time.time()
print(end - start)
index(1, 2)
# index()
- 方案二
# 問題:沒有修改被裝修飾對象的源代碼,也滅有修改調用方式,但是代碼冗餘
def index(x, y):
time.sleep(2)
print(' index %s %s ' % (x, y))
start = time.time()
index(11, 22)
end = time.time()
start = time.time()
index(11, 22)
end = time.time()
- 方案三
問題:解決了代碼榮譽,但是函數的調用方式改變了
def index(x, y):
time.sleep(2)
print(' index %s %s ' % (x, y))
def wrapper():
start = time.time()
index(11, 22)
end = time.time()
wrapper()
wrapper()
wrapper()
- 方案三的優化一:
在方案三基礎上優化代碼:將Index寫活了(參數寫活了)
def index(x, y,z):
time.sleep(2)
print(' index %s %s %s ' % (x, y,z))
def wrapper(*args,**kwargs):
start=time.time()
index(*args,**kwargs)
stop=time.time()
print(stop-start)
wrapper(11,22,44)
wrapper(111,y=111,z=4545)
-
方案三的優化二:
-
在優化一的基礎上把被裝飾對象寫活,原來只能裝飾Index
def index(x, y, z):
time.sleep(2)
print(' index %s %s %s ' % (x, y, z))
def home(name):
time.time()
print('welcome %s to home page' % name)
def outer(func): # func=index的記憶體地址
# func = index
def wrapper(*args, **kwargs):
start = time.time()
func(*args, **kwargs) # index的記憶體地址()
stop = time.time()
print(stop - start)
return wrapper
home=outer(home)
home('zhao')
# f = outer(index) # f=outer(index的記憶體地址)
# f(x=1, y=2, z=3)
index = outer(index) # 偷梁換柱.>>此時的index指向的是wrapper的記憶體地址
print(index)
index(x=1, y=2, z=3)
-
方案三的優化三
將wrapper做的跟被裝飾器一摸一樣,以假亂真
def index(x, y, z):
time.sleep(2)
print(' index %s %s %s ' % (x, y, z))
def home(name):
time.time()
print('welcome %s to home page' % name)
return 1234
def outer(func):
def wrapper(*args, **kwargs):
start = time.time()
res = func(*args, **kwargs)
stop = time.time()
print(stop - start)
return res
return wrapper
home = outer(home)
res = home('zhao')
print('返回值:>>>', res)
(4) 語法糖:
在被裝飾對象正上方的單獨一行寫 @裝飾器名字
# 裝飾器
def timmer(func):
def wrapper(*args, **kwargs):
start = time.time()
res = func(*args, **kwargs)
stop = time.time()
print(stop - start)
return res
return wrapper
@timmer # index=timmer(index)
def index(x, y, z):
time.sleep(2)
print(' index %s %s %s ' % (x, y, z))
@timmer # home = timmer(home)
def home(name):
time.sleep(2)
print('welcome %s to home page' % name)
return 1234
index(x=1,y=2,z=3)
home('zhao')
(5)偷梁換柱
即將原函數名指向的記憶體地址偷梁換柱,所以應該將wrapper做的跟原函數一樣才行
手動的將原函數的屬性值賦值給wrapper,需要一個一個的去加,太麻煩
def outter(func):
def wrapper(*args, **kwargs):
#手動的將原函數的屬性值賦值給wrapper
# 函數名wrapper.__name__ =原函數.__name__
# 函數名wrapper.__doc__ = 原函數.__doc__
wrapper.__name__ = func.__name__
wrapper.__doc__ = func.__doc__
res = func(*args, **kwargs)
return res
return wrapper
@outter # index=outter(index)
def index(x, y):
"""
:param x:
:param y:
:return:
"""
print('index:', x, y)
index(1, 2)
print(index.__name__)
print(index.__doc__) # help(index)
自動的將原函數的所有屬性值賦值給wrapper
from functools import wraps
def outter(func):
@wraps(func)
def wrapper(*args, **kwargs):
#手動的將原函數的屬性值賦值給wrapper
# 函數名wrapper.__name__ =原函數.__name__
# 函數名wrapper.__doc__ = 原函數.__doc__
# wrapper.__name__ = func.__name__
# wrapper.__doc__ = func.__doc__
res = func(*args, **kwargs)
return res
return wrapper
@outter # index=outter(index)
def index(x, y):
"""
:param x:
:param y:
:return:
"""
print('index:', x, y)
index(1, 2)
print(index.__name__)
print(index.__doc__) # help(index)
- 無參裝飾器模板
def outter(func):
def wrapper(*args,**kwargs):
res=func(*args,**kwargs)
return res
return wrapper()
- 統計時間的裝飾器
def timmer(func):
def wrapper(*args,**kwargs):
start=time.time()
res=func(*args,**kwargs)
end=time.time()
print(end-start)
return res
return wrapper()
- 認證功能
def auth(func):
def wrapper(*args, **kwargs):
name = input('your name :').strip()
passwd = input('your password:').strip()
if name == 'zhao' and passwd == '132':
res = func(*args, **kwargs)
return res
else:
print("your name or your password is error")
return wrapper
@auth
def index():
print('from index')
index()
- 有參裝飾器模板
def 有參裝飾器(x,y,z)
def outter(func):
def wrapper(*args,**kwargs):
res=func(*args,**kwargs)
return res
return wrapper()
@有參裝飾器(1,y=2,z=3)
def 被裝飾對象():
pass
- 認證功能改進
def auth(db_type):
def deco(func):
def wrapper(*args, **kwargs):
username = input('your name:').strip()
password = input('your paddword').strip()
if db_type == 'file':
print('基於文件驗證')
if username == 'zhao' and password == '133':
print('login successful')
res = func(*args, **kwargs)
return res
else:
print('username or password is error')
elif db_type == 'mysql':
print("基於資料庫")
elif db_type == 'ldap':
print("基於ldap")
else:
print('不支持該db_type')
return wrapper
return deco
@auth(db_type='file')#@deco #index=dexo(index)
def index(x, y):
print('index:>>%s %s' % (x, y))
@auth(db_type='mysql')
def home(name):
print('home :>>%s' % name)
@auth(db_type='ldap')
def transfer():
print("transfer:>>>%s" % transfer)
index(1, 2)
home('zhao')
transfer()
- 疊加多個裝飾器分析
def deco1(func1): # func1=wrapper2的記憶體地址
def wrapper1(*args, **kwargs):
print('deco1.wrapper1')
res1 = func1(*args, **kwargs)
return res1
return wrapper1
def deco2(func2): # func2=wrapper3的記憶體地址
def wrapper2(*args, **kwargs):
print('deco1.wrapper2')
res2 = func2(*args, **kwargs)
return res2
return wrapper2
def deco3(x):
def outter(func3): # func3=被裝飾對象index函數的記憶體地址
def wrapper3(*args, **kwargs):
print('deco3.outter.wrapper3')
res3 = func3(*args, **kwargs)
return res3
return wrapper3
return outter
# 載入順序:自下而上
@deco1 # index=deco1(wrapper2的記憶體地址) ===》index=wrapper1的記憶體地址
@deco2 # index=deco2(wrapper3的記憶體地址)===》index=wrapper2的記憶體地址
@deco3(11) # @outer===>@index=outer(index)===>index=wrapper3的記憶體地址
def index(x, y):
print('from index %s %s' % (x, y))
print(index)
# 執行順序:自上而下即 wrapper1>wrapper2>wrapper3
#
index(1, 2) # wrapper1(1,2)
本文來自博客園,作者:Expiredsaury,轉載請註明原文鏈接:https://www.cnblogs.com/saury/p/16739191.html