快捷導(dǎo)航

python函數(shù)裝飾器構(gòu)造和參數(shù)傳遞

更新時(shí)間：2022年03月08日 10:40:57 作者：python老鳥(niǎo)

這篇文章主要介紹了python函數(shù)裝飾器構(gòu)造和參數(shù)傳遞，下面通過(guò)一個(gè)小案例來(lái)簡(jiǎn)單的理解什么是裝飾器，需要的小伙伴可以參考一下

這時(shí)候有一個(gè)需求，必須是貓和狗的主人呼喊它們才會(huì)做出以上動(dòng)作，就需要對(duì)指令發(fā)出者進(jìn)行身份驗(yàn)證。如果直接在判斷上采用身份驗(yàn)證，這樣代碼重用度會(huì)很低，如果在上面兩個(gè)函數(shù)中寫(xiě)，如果驗(yàn)證代碼過(guò)多，可能需要寫(xiě)好幾遍。這時(shí)候我們可以再創(chuàng)建一個(gè)函數(shù)，在調(diào)用dog和cat函數(shù)的時(shí)候先調(diào)用身份驗(yàn)證函數(shù)，但是這樣，我們的dog函數(shù)用在其他地方時(shí)如果不需要驗(yàn)證就會(huì)有冗余代碼。上面幾種方案都有自己的缺點(diǎn)，我們可以試試前面學(xué)習(xí)的閉包函數(shù)來(lái)實(shí)現(xiàn)這個(gè)功能。

一.閉包函數(shù)

def func(f):
? ? def test():
? ? ? ? print('主人身份驗(yàn)證')
? ? ? ? f()
? ? return test
? ??
def dog():
? ? print('搖尾巴')
dog = func(dog) # 這里的dog其實(shí)是test函數(shù)
?
def cat():
? ? print('喵喵喵')
cat = func(cat)
call = '狗'
if call == '狗':
? ? dog() # ★★★這里的dog函數(shù)其實(shí)是test函數(shù)，所以先執(zhí)行身份驗(yàn)證，然后又調(diào)用f()函數(shù)，也就是原來(lái)的dog()函數(shù),也可以給這行的dog函數(shù)換個(gè)名字，好理解★★★
else:
? ? cat()

二.python裝飾器構(gòu)造

python提供一種簡(jiǎn)單的裝飾器寫(xiě)法，叫做語(yǔ)法糖，

如下：

def func(f):
? ? def test():
? ? ? ? print('主人身份驗(yàn)證')
? ? ? ? f()
? ? return test
? ??
@func
def dog():
? ? print('搖尾巴')
# dog = func(dog)
?
@func
def cat():
? ? print('喵喵喵')# cat = func(cat)
call = '狗'
if call == '狗':
? ? dog()
else:
? ? cat()

函數(shù)體不發(fā)生改變，增加了額外的功能，重用性高。裝飾器內(nèi)部必須使用閉包函數(shù)，否則當(dāng)使用@時(shí)，裝飾器就會(huì)被直接執(zhí)行，注意執(zhí)行順序。

三. python裝飾器疊加

# 裝飾器可以被疊加使用
def func(f):
? ? def test():
? ? ? ? print('主人身份驗(yàn)證')
? ? ? ? f()
? ? return test
? ??
def func2(f):
? ? def test2():
? ? ? ? print('=======')
? ? ? ? f()
?return test2
?
@func2
@func ?# 可以疊加使用裝飾器，先執(zhí)行上面的裝飾器
def dog():
? ? print('搖尾巴')
dog() # 這里的dog函數(shù)其實(shí)是test和test2兩個(gè)函數(shù)，而test和test2又返回來(lái)調(diào)用上面的dog()原始函數(shù)

四.python裝飾器傳參

1.裝飾器單個(gè)參數(shù)傳遞

def test(f):
? ? def test1(x):
? ? ? ? print('==========')
? ? ? ? f(x)
? ? return test1
? ??
@test
def func1(m):
? ? print(m)
? ??
func1(10)

2.裝飾器多個(gè)參數(shù)傳遞

def test(f):
? ? def test1(x, y):
? ? ? ? print('==========')
? ? ? ? f(x, y)
? ? return test1
? ??
@test
def func2(m, n):
? ? print(m, n)
? ??
func2(10, 5)

3.裝飾器的不定長(zhǎng)參數(shù)

def test(f):
? ? def test1(*args, **kwargs):
? ? ? ? print('==========')
? ? ? ? f(*args, **kwargs)
? ? return test1
?
@test
def func2(a, b, c):
? ? # print(args, kwargs)
? ? print('*********')
func2(10, 5, c=6) # 這里的c和上面func2的第三個(gè)形參名要一致

五、帶返回值的裝飾器

def test(f):
? ? def test1(*args, **kwargs): # 這里的test1函數(shù)要和被裝飾函數(shù)func2的結(jié)構(gòu)保持一致
? ? ? ? print('==========')
? ? ? ? res = f(*args, **kwargs) # 這里相當(dāng)于把被裝飾函數(shù)的結(jié)果拿過(guò)來(lái)賦值,f(*args, **kwargs)的執(zhí)行結(jié)果就是func2的返回值
? ? ? ? return res ?# 沒(méi)有返回值也可以這樣寫(xiě)，返回結(jié)果就是None
? ? return test1
? ??
@test
def func2(a, b, c):
? ? # print(args, kwargs)
? ? print('*********')
? ? return a + b + c
print(func2(10, 5, c=88))

到此這篇關(guān)于python函數(shù)裝飾器構(gòu)造和參數(shù)傳遞的文章就介紹到這了,更多相關(guān)python函數(shù)裝飾器內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: