測開之函數進階· 第7篇《裝飾器裝飾類,通用裝飾器,有啥區別呢?》
堅持原創輸出,點擊藍字關注我吧
作者:清菡
博客:oschina、雲+社區、知乎等各大平台都有。
由於微信公眾號推送改為了信息流的形式,防止走丟,請給加個星標 ⭐,你就可以第一時間接收到本公眾號的推送!
目錄
- 一、什麼是裝飾器
- 1.開放封閉原則(面向對象原則的核心)
- 2.裝飾器的作用
- 二、實現一個裝飾器
- 1.不帶參數的裝飾器
- 2.裝飾器的原理
- 3.組裝方便,拆卸也方便
- 4.帶參數的裝飾器
- 三、通用裝飾器
- 四、裝飾器裝飾類
- 1.不帶參數的
- 2.帶參數的
- 五、裝飾器的應用場景
- 六、補充
- 1.*號的作用
- 2.裝飾器裝飾類和裝飾函數的不同點
一、什麼是裝飾器
1.開放封閉原則(面向對象原則的核心)
對已經實現的功能(項目已經上線了),在這個基礎上增加新功能,也可以在它的基礎上進行拓展,這個就是開放。如果你要去再修改它內部的代碼,這個時候是不允許的,對內部的修改是封閉的。
就是你實現的功能可以拓展,但是你不要去修改它內部的代碼。
比如index()
是接口,返回是「這個是網站的首頁」,只要調用這個接口就會返回一個「這個是網站的首頁。」
突然有個需求,在進入網站之前需要先登錄校驗一下。這個時候需要拓展,如何拓展?就需要用裝飾器了。
def index():
print("這個是網站的首頁")
這個已經實現的接口,不能去修改的。
2.裝飾器的作用
裝飾器可在不更改這個函數裏面任何代碼的基礎上,給它添加新的功能。
二、實現一個裝飾器
1.不帶參數的裝飾器
裝飾器其實就是一種閉包的應用。要使用裝飾器,可以先定義個閉包函數。把登錄校驗的功能寫在了閉包函數的內部。
把閉包函數當成裝飾器來用的話,外面接收的參數需要傳一個函數,你要裝飾哪個函數,你就傳哪個函數。
# 開放封閉原則
def login(func):
def fun():
# 簡單的校驗
username = 'python01'
password = 'qinghan'
user = input("請輸入賬號:")
pw = input("請輸入密碼:")
# 判斷下賬號密碼對不對
if username == user and pw == password:
func() # 登錄得賬號密碼都正確的情況下,調用這個函數
else:
print("賬號或密碼錯誤")
return fun
@login # 艾特一下這個裝飾器
def index():
print("這個是網站的首頁")
index()
2.裝飾器的原理
將被裝飾的函數當作一個參數傳到裝飾器中,並且讓被裝飾的函數名指向裝飾器內部的函數,在裝飾器的內部函數中用接收到的參數再調用被裝飾的函數。
@login
是 Python 中的一個語法糖。它的作用是:index=login(index)
。傳入index
,然後被index
接收。
如何做到通過func()調用原函數?
@login
等於index=login(index)
。
自動將index
當作參數傳入login
這個函數裏面,去執行login(func)
這個函數,檢測到這個fun()
函數,將這段代碼:
def fun():
# 簡單的校驗
username = 'python01'
password = 'qinghan'
user = input("請輸入賬號:")
pw = input("請輸入密碼:")
# 判斷下賬號密碼對不對
if username == user and pw == password:
func() # 登錄得賬號密碼都正確的情況下,調用這個函數
else:
print("賬號或密碼錯誤")
直接跳過。
return fun
直接將結果返回出來。結果返回出來又給index()
接收,調用index()
的時候實際上是調用fun()
函數。
執行fun()
函數裏面的代碼。通過func()
調用原函數,怎麼做到的?
fun()
函數是放在index.__closure__
這個屬性裏面。
然後在下面調用func()
的時候,就是去index.__closure__
這個屬性裏面找到對應存儲的那塊代碼。
存儲的代碼就是這個:
def index():
print("這個是網站的首頁")
以上,就是裝飾器裝飾的流程。
3.組裝方便,拆卸也方便
我想改成不用登錄也可以訪問,直接去掉@login
這個裝飾器就可以了。
# 開放封閉原則
def login(func):
def fun():
# 簡單的校驗
username = 'python01'
password = 'qinghan'
user = input("請輸入賬號:")
pw = input("請輸入密碼:")
# 判斷下賬號密碼對不對
if username == user and pw == password:
func() # 登錄得賬號密碼都正確的情況下,調用這個函數
else:
print("賬號或密碼錯誤")
return fun
# @login # 艾特一下這個裝飾器
def index():
print("這個是網站的首頁")
# index.__closure__
index()
這樣操作,不會對原來有什麼影響。
4.帶參數的裝飾器
實現兩個數相加後,又有新的需求,需要可以相乘、相除。
def add(func):
def fun(a, b):
print("相乘", a * b)
print("相除", a / b)
func(a, b)
return fun
@add
def add_num(a, b):
# 打印兩個數相加
print("相加:", a + b)
add_num(11, 22)
三、通用裝飾器
如果同一個裝飾器既要裝飾有參數的函數,又要裝飾無參數的函數。
那麼我們在傳參的時候就設置成不定長參數,這樣不管被裝飾的函數有沒有參數都能用。
# 通用裝飾器
def add(func):
def fun(*args, **kwargs):
print("裝飾器的功能代碼:登錄")
func(*args,**kwargs)
return fun
@add
def index():
print("這個是網站的首頁")
@add
def good_list(num):
print("這個是商品列表第{}頁".format(num))
index()
print("------------")
good_list(9)
四、裝飾器裝飾類
1.不帶參數的
#裝飾器裝飾類
def add(func):
def fun(*args, **kwargs):
print("裝飾器的功能代碼:登錄")
return func(*args,**kwargs)
return fun
@add # MyClass=add(MyClass)
class MyClass:
def __init__(self):
pass
m = MyClass()
print("m的值:",m)
把類當作一個參數傳到裝飾器裏面。return fun
返回的是fun
,MyClass
接收到的是fun
。
MyClass()
調用的是fun
。
執行代碼:
def fun(*args, **kwargs):
print("裝飾器的功能代碼:登錄")
return func(*args,**kwargs)
這裏面的功能。
先執行裝飾器的功能,return func(*args,**kwargs)
,func()
來自def add(func)
。
調用MyClass
這個類,return func(*args,**kwargs)
創建了個對象,MyClass()
調用完了接收,m 就能接收這個對象了。
這個就是裝飾器裝飾類的一個原理。
2.帶參數的
#裝飾器裝飾類
def add(func):
def fun(*args, **kwargs):
print("裝飾器的功能代碼:登錄")
return func(*args,**kwargs)
return fun
@add # MyClass=add(MyClass)
class MyClass:
def __init__(self,name,age):
self.name=name
self.age=age
m = MyClass("qinghan","18")
print("m的值:",m)
這裡用的是不定行參數,所以不管你裝飾的類是有參數的還是沒參數的,都可以。
五、裝飾器的應用場景
1.登錄校驗。(在裝飾器裏面判斷下你有沒有登錄)
2.函數運行時間統計。
3.執行函數之前做準備工作。
4.執行函數後清理功能。
六、補充
1.*號的作用
*
是進行拆包作用的。把每個元素拿出來,當作參數進行傳遞。
一個*
是對元組形式的位置參數進行拆包,兩個**
對關鍵字參數進行拆包。
2.裝飾器裝飾類和裝飾函數的不同點
類需要把對象返回出來。
公眾號清菡軟件測試首發,更多原創文章:清菡軟件測試 116+原創文章,歡迎關注、交流,禁止第三方擅自轉載。