巧妙使用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...
代碼非常難看,也很難維護(hù)。并且每個 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}')
運(yùn)行效果如下圖所示:
這樣寫,比用字典的方式更直觀,比直接用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
運(yùn)行效果如下圖所示:
它這個代碼目前只能實現(xiàn)相等的查詢。但其實只要對這個代碼稍作修改,我們就能實現(xiàn)大于、小于、大于等于、小于等于、不等于、in
等等判斷。如果大家有興趣的話,請在文章下面留言,我們明天就來說說怎么對這個代碼進(jìn)行改造,實現(xiàn)更多的邏輯判斷。
參考文獻(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)文章希望大家以后多多支持腳本之家!
相關(guān)文章
Python?Pexpect庫自動化交互式進(jìn)程控制的expect_list方法解析
Pexpect是一個Python庫,為自動化和交互式進(jìn)程控制提供了豐富的功能,而expect_list方法是其功能強(qiáng)大且靈活的一部分,將詳細(xì)探討如何使用這一方法,并提供多個示例來說明其應(yīng)用場景和功能2024-01-01Python字節(jié)單位轉(zhuǎn)換(將字節(jié)轉(zhuǎn)換為K M G T)
這篇文章主要介紹了Python字節(jié)單位轉(zhuǎn)換(將字節(jié)轉(zhuǎn)換為K M G T),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03解決Building?wheel?for?wrapt?(setup.py)?...?error的問題
這篇文章主要介紹了解決Building?wheel?for?wrapt?(setup.py)?...?error的問題,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教2022-12-12python爬取網(wǎng)頁數(shù)據(jù)到保存到csv
大家好,本篇文章主要講的是python爬取網(wǎng)頁數(shù)據(jù)到保存到csv,感興趣的同學(xué)趕快來看一看吧,對你有幫助的話記得收藏一下,方便下次瀏覽2022-01-01