python3 裝飾器,python3裝飾
如果想在程式中加入惡意代碼,裝飾器是不錯的選擇!
- 一來裝飾器函數一般在主程式外定義,不易被發現;
- 二來裝飾器函數在主程式運行之前執行,可以在主程式運行之前,獲得系統分配的資源.
裝飾器的文法規則很有趣,可以在不改變原有函數代碼的前提下進行擴充,實現程式的友好迭代升級其實裝飾器的功能不只是更改函數,如果你願意,裝飾器還能"智能"的選擇函數在網站的後台,有這樣一種需求:需要把前端頁面的請求分類處理:把形似(1345.html, 4567.html)等以數字命名的網頁邏輯交由一個函數處理;把形似(index.html, center.html)等以字母命名的網頁交由另一個函數處理;直白的講,就是對不同頁面的請求進行分類處理.對資料進行分類匹配,Regex是首選的方式!我們先定義兩個負責處理不同邏輯的函數,並附帶"Regex"匹配規則的裝飾器
# 匹配"字元名稱網頁"的函數@route(r"[a-z]+\.html")def char(file_name, url=None): print("收到了'名稱為字元'的網頁") print("接收的檔案名稱為",file_name) print("匹配的正則url運算式為:",url) print("~"*20)#匹配"數字名稱網頁"的函數@route(r"[\d]+\.html")def num(file_name, url=None): print("收到了'名稱為數字'的網頁") print("接收的檔案名稱為",file_name) print("匹配的正則url運算式為:",url) print("~"*20)
然後定義裝飾器函數
# route用於匹配函數式def route(url): # set_func用於接收函數的引用 def set_func(func): # 將Regex與函數綁定 route_dic[url] = func print("--為route_dic--添加了一個索引值對") # call_func用於建立新的函數,並將新函數返回,替換被修飾的舊函數 def call_func(file_name, url): return func(file_name, url) return call_func return set_func
基礎的裝飾器定義為兩層函數嵌套,外層負責接受原函數(被添加裝飾器的函數)的引用,內層負責擴充原函數功能,最後外層函數將內層函數返回,返回的內層函數將替代原函數(被添加裝飾器的函數)
我們這裡的裝飾器函數在基礎裝飾器的層面上又多套了一層負責判定的函數,這樣我們的裝飾器就有了"分類功能"
測試代碼:
import re# 用於儲存"函數的引用" 與 "urlRegex"的索引值對route_dic = dict()# route用於匹配函數式def route(url): # set_func用於接收函數的引用 def set_func(func): # 預先載入,將Regex與函數綁定 route_dic[url] = func print("--為route_dic--添加了一個索引值對") # call_func用於建立新的函數,並將新函數返回,替換被修飾的舊函數 def call_func(file_name, url): return func(file_name, url) return call_func return set_func @route(r"[a-z]+\.html")def char(file_name, url=None): print("收到了'名稱為字元'的網頁") print("接收的檔案名稱為",file_name) print("匹配的正則url運算式為:",url) print("~"*20)@route(r"[\d]+\.html")def num(file_name, url=None): print("收到了'名稱為數字'的網頁") print("接收的檔案名稱為",file_name) print("匹配的正則url運算式為:",url) print("~"*20)def match_func(file_name): for url, call_func in route_dic.items(): ret = re.match(url, file_name) if ret: print("找到了頁面") call_func(file_name,url) else: passdef main(): print("route字典的值為", route_dic) # 測試"名稱為字元"的網頁 match_func("index.html") # 測試"名稱為數字"的網頁 match_func("123.html")if __name__ == "__main__": main()
運行結果
運行結果
如所示,123.html和index.html都被分配到了對應的函數執行!
更深一點:細心的人會發現,其實裝飾器函數比主程式要更早的執行
- 證據1:主程式在使用字典
route_dic之前,未調用過route函數,但route_dic中卻早已有了值,說明在main函數執行之前,route函數已經被調用過了;
- 證據2:從以上程式的執行結果來看,添加索引值對的操作在main函數執行之前,提前執行了兩次,正好對應兩個裝飾器
如果想在程式中加入惡意代碼,裝飾器函數是不錯的選擇!
- 一來裝飾器函數一般在主程式外定義,不易被發現;
- 二來裝飾器函數在主程式運行之前執行,可以在主程式運行之前,獲得系統分配的資源.
作者:_昭昭_
連結:http://www.jianshu.com/p/dee05d5dfb50
來源:簡書
著作權歸作者所有。商業轉載請聯絡作者獲得授權,非商業轉載請註明出處。