我們來看看這個方法具體是什么樣的。

假設我們要做一個功能，根據用戶的等級判斷他可以獲得的折扣。

常規的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等等判斷。如果大家有興趣的話，請在文章下面留言，我們明天就來說說怎么對這個代碼進行改造，實現更多的邏輯判斷。

參考文獻

[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 裝飾器內容請搜索服務器之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持服務器之家！

原文鏈接：https://blog.csdn.net/weixin_38037405/article/details/121380740