大家好，今天在 Github 阅读 EdgeDB[1] 的代码，发现它在处理大量if…elif…else的时候，巧妙地使用了装饰器，方法设计精巧，分享给大家一下，欢迎收藏学习，喜欢点赞支持，技术交流见文末。

我们来看看这个方法具体是什么样的。假设我们要做一个功能，根据用户的等级判断他可以获得的折扣。常规的if … elif…写法是这样的：

def get_discount(level):if level == 1:"大量计算代码"discount = 0.1elif level == 2:"大量计算代码"discount = 0.2elif level == 3:discount = 0.3elif level == 4:discount = 0.4elif level == 5:discount = 0.5elif level == 6:discount = 3 + 2 - 5 * 0.1else:return '等级错误'return discount

大家都知道，这样大量的if … elif…代码非常难看，也很难维护。并且每个 if 的内部有很多代码。这个函数就会被拉得非常长。

更多原创文章、资料，可以微信扫描下面二维码——>添加csdn官方认证第一时间免费获取

有一些同学知道，可以使用字典来改写这个太长的 if 判断：

def parse_level_1():"大量计算代码"discount = 0.1return discountdef parse_level_2():"大量计算代码"discount = 0.2return discountdef parse_level_3():"大量计算代码"discount = 0.3return discountdef parse_level_4():"大量计算代码"discount = 0.4return discountdef parse_level_5():"大量计算代码"discount = 0.5return discountdef parse_level_6():"大量计算代码"discount = 3 + 2 - 5 * 0.1return discountdiscount_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.1return discount@get_discount.register(2)
def parse_level_2(level):"大量计算代码"discount = 0.2return discount@get_discount.register(3)
def parse_level_3(level):"大量计算代码"discount = 0.3return discount@get_discount.register(4)
def parse_level_4(level):"大量计算代码"discount = 0.4return discount@get_discount.register(5)
def parse_level_5(level):"大量计算代码"discount = 0.5return discount@get_discount.register(6)
def parse_level_1(level):"大量计算代码"discount = 3 + 2 - 5 * 0.1return discountdiscount = 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.2return discount

运行效果如下图所示：

它这个代码目前只能实现相等的查询。但其实只要对这个代码稍作修改，我们就能实现大于、小于、大于等于、小于等于、不等于、in等等判断。如果大家有兴趣的话，请在文章下面留言，我们明天就来说说怎么对这个代码进行改造，实现更多的逻辑判断。

最后大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！
朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费】