如果想在程序中加入惡意代碼,裝飾器是不錯的選擇!

• 一來裝飾器函數一般在主程序外定義,不易被發現;
• 二來裝飾器函數在主程序運行之前執行,可以在主程序運行之前,獲得系統分配的資源.

裝飾器的語法規則很有趣,可以在不改變原有函數代碼的前提下進行擴展,實現程序的友好叠代升級

其實裝飾器的功能不只是更改函數,如果你願意,裝飾器還能"智能"的選擇函數

在網站的後臺,有這樣一種需求:需要把前端頁面的請求分類處理:把形似(1345.html, 4567.html)等以數字命名的網頁邏輯交由一個函數處理;把形似(index.html, center.html)等以字母命名的網頁交由另一個函數處理;

直白的講,就是對不同頁面的請求進行分類處理.

對數據進行分類匹配,正則表達式是首選的方式!

我們先定義兩個負責處理不同邏輯的函數,並附帶"正則表達式"匹配規則的裝飾器

# 匹配"字符名稱網頁"的函數
@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):
        # 將正則表達式與函數綁定
        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

# 用於存儲"函數的引用" 與 "url正則表達式"的鍵值對
route_dic = dict()

# route用於匹配函數式
def route(url):
    # set_func用於接收函數的引用
    def set_func(func):
        # 預加載,將正則表達式與函數綁定
        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:
            pass

def 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函數執行之前,提前執行了兩次,正好對應兩個裝飾器

如果想在程序中加入惡意代碼,裝飾器函數是不錯的選擇!

• 一來裝飾器函數一般在主程序外定義,不易被發現;
• 二來裝飾器函數在主程序運行之前執行,可以在主程序運行之前,獲得系統分配的資源.

python3 裝飾器