巧妙使用Python裝飾器處理if...elif...else
我們來看看這個方法具體是什么樣的。假設(shè)我們要做一個功能，根據(jù)用戶的等級判斷他可以獲得的折扣。常規(guī)的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 的內(nèi)部有很多代碼。這個函數(shù)就會被拉得非常長。
有一些同學(xué)知道，可以使用字典來改寫這個太長的 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, '等級錯誤')
但今天我學(xué)到的這個方法，比用字典更簡單。我們先來看它的效果：
@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是怎么實現(xiàn)的呢？密碼就藏在這個開源項目EdgeDB的源代碼[2]中，核心代碼只有 20 多行：
并且，還能夠?qū)崿F(xiàn)或查詢。例如用戶等級為 2 或者 3 的時候，折扣都是 0.2，那么代碼可以寫成：
@get_discount.register(2) @get_discount.register(3) def parse_level_2(level): "大量計算代碼" discount = 0.2 return discount
運行效果如下圖所示：
它這個代碼目前只能實現(xiàn)相等的查詢。但其實只要對這個代碼稍作修改，我們就能實現(xiàn)大于、小于、大于等于、小于等于、不等于、in等等判斷。如果大家有興趣的話，請在文章下面留言，我們明天就來說說怎么對這個代碼進行改造，實現(xiàn)更多的邏輯判斷。
參考文獻
[1] EdgeDB: https://github.com/edgedb/edgedb
[2] 源代碼: https://github.com/edgedb/edgedb/blob/master/edb/common/value_dispatch.py
技術(shù)交流
歡迎轉(zhuǎn)載、收藏、有所收獲點贊支持一下！
到此這篇關(guān)于巧妙使用Python裝飾器處理if...elif...else的文章就介紹到這了,更多相關(guān)Python 裝飾器內(nèi)容請搜索本站以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持本站！
版權(quán)聲明：本站文章來源標注為YINGSOO的內(nèi)容版權(quán)均為本站所有，歡迎引用、轉(zhuǎn)載，請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站，禁止在非maisonbaluchon.cn所屬的服務(wù)器上建立鏡像，否則將依法追究法律責任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來，僅供學(xué)習(xí)參考，不代表本站立場，如有內(nèi)容涉嫌侵權(quán)，請聯(lián)系alex-e#qq.com處理。
關(guān)注官方微信