巧妙使用Python裝飾器處理if...elif...else
我們來看看這個方法具體是什么樣的。假設我們要做一個功能，根據用戶的等級判斷他可以獲得的折扣。常規(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 的內部有很多代碼。這個函數就會被拉得非常長。
有一些同學知道，可以使用字典來改寫這個太長的 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是怎么實現(xiàn)的呢？密碼就藏在這個開源項目EdgeDB的源代碼[2]中，核心代碼只有 20 多行：
并且，還能夠實現(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
技術交流
歡迎轉載、收藏、有所收獲點贊支持一下！
到此這篇關于巧妙使用Python裝飾器處理if...elif...else的文章就介紹到這了,更多相關Python 裝飾器內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
本文主要介紹了python中綁定方法與非綁定方法的實現(xiàn)示例，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-09-09
python實現(xiàn)批量監(jiān)聽頁面并發(fā)送郵件
這篇文章主要為大家詳細介紹了python如何實現(xiàn)自動化批量監(jiān)聽頁面并發(fā)送郵件,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2023-11-11
Python3開發(fā)監(jiān)控自動化觸發(fā)聲光報警
使用python制作一個自動監(jiān)控并觸發(fā)聲光報警是不是感覺很高端，很多人都會認為只是一件很難的事情，但實際很簡單就能實現(xiàn)。2023-07-07
手動安裝Anaconda環(huán)境變量的實現(xiàn)教程
這篇文章主要介紹了手動安裝Anaconda環(huán)境變量的實現(xiàn)教程，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-01-01
本文主要介紹了PyQt5+PyQt5Designer的安裝步驟，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-01-01
這篇文章主要介紹了Python導入模塊包原理及相關注意事項,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-03-03
今天小編就為大家分享一篇Python重新加載模塊的實現(xiàn)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-10-10
2018年Python值得關注的開源庫、工具和開發(fā)者(總結篇)
本文給大家總結了2018年Python值得關注的開源庫、工具和開發(fā)者,需要的朋友可以參考下2018-01-01
本文精心篩選了最能展現(xiàn) Python 語法之精妙的十個知識點，并附上詳細的實例代碼,需要的朋友可以參考下2020-01-01
這篇文章主要為大家詳細介紹了如何基于Python搭建局域網大文件分享傳輸工具,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2024-12-12