[[439061]]

今天在 Github 閱讀EdgeDB[1]的代碼，發現它在處理大量if...elif...else判斷的時候，使用了一個非常巧妙的裝飾器。我們來看看這個方法具體是什么樣的。

正好今天是雙十一，假設我們要做一個功能，根據用戶的等級判斷他可以獲得的折扣。常規的if ... elif...寫法是這樣的： 

def get_discount(level):  
    if level == 1:  
        "大量計算代碼"  
        discount = 0.1  
    elif level == 2:  
        "大量計算代碼"  
        discount = 0.2  
    elif level == 3:  
        discount = 0.3  
    elif level == 4:  
        discount = 0.4  
    elif level == 5:  
        discount = 0.5  
    elif level == 6:  
        discount = 3 + 2 - 5 * 0.1  
    else:  
         return '等級錯誤'  
    return discount 

大家都知道，這樣大量的if ... elif...代碼非常難看，也很難維護。并且每個 if 的內部有很多代碼。這個函數就會被拉得非常長。

有一些同學知道，可以使用字典來改寫這個太長的 if 判斷： 

def parse_level_1():  
    "大量計算代碼"  
    discount = 0.1  
    return discount  
def parse_level_2():  
    "大量計算代碼"  
    discount = 0.2  
    return discount  
def parse_level_3():  
    "大量計算代碼"  
    discount = 0.3  
    return discount  
def parse_level_4():  
    "大量計算代碼"  
    discount = 0.4  
    return discount  
def parse_level_5():  
    "大量計算代碼"  
    discount = 0.5  
    return discount  
def parse_level_6():  
    "大量計算代碼"  
    discount = 3 + 2 - 5 * 0.1  
    return discount  
discount_map = {  
 1: parse_level_1,  
  2: parse_level_2,  
  3: parse_level_3,  
  4: parse_level_4,  
  5: parse_level_5,  
  6: parse_level_6,  
}  
discount = discount_map.get(level, '等級錯誤') 

但今天我學到的這個方法，比用字典更簡單。我們先來看它的效果： 

@value_dispatch  
def get_discount(level):  
    return '等級錯誤'  
@get_discount.register(1)  
def parse_level_1(level):  
    "大量計算代碼"  
    discount = 0.1  
    return discount  
@get_discount.register(2)  
def parse_level_2(level):  
    "大量計算代碼"  
    discount = 0.2  
    return discount  
@get_discount.register(3)  
def parse_level_3(level):  
    "大量計算代碼"  
    discount = 0.3  
    return discount  
@get_discount.register(4)  
def parse_level_4(level):  
    "大量計算代碼"  
    discount = 0.4  
    return discount  
@get_discount.register(5)  
def parse_level_5(level):  
    "大量計算代碼"  
    discount = 0.5  
    return discount  
@get_discount.register(6)  
def parse_level_1(level):  
    "大量計算代碼"  
    discount = 3 + 2 - 5 * 0.1  
    return discount  
discount = get_discount(3)  
print(f'等級3的用戶，獲得的折扣是：{discount}') 

運行效果如下圖所示：

這樣寫，比用字典的方式更直觀，比直接用if ... elif...更簡潔。

那么，這個裝飾器value_dispatch是怎么實現的呢？密碼就藏在這個開源項目EdgeDB的源代碼[2]中，核心代碼只有20多行：

并且，還能夠實現或查詢。例如用戶等級為2或者3的時候，折扣都是0.2，那么代碼可以寫成： 

@get_discount.register(2)  
@get_discount.register(3)  
def parse_level_2(level):  
    "大量計算代碼"  
    discount = 0.2  
    return discount 

運行效果如下圖所示：

它這個代碼目前只能實現相等的查詢。但其實只要對這個代碼稍作修改，我們就能實現大于、小于、大于等于、小于等于、不等于、in等等判斷。如果大家有興趣的話，請在文章下面留言，我們明天就來說說怎么對這個代碼進行改造，實現更多的邏輯判斷。