一日一技：使用装饰器简化大量 if…elif…代码

今天在 Github 阅读EdgeDB[1]的日技代码，发现它在处理大量if...elif...else判断的使用时候，使用了一个非常巧妙的装饰装饰器。我们来看看这个方法具体是器简什么样的。

正好今天是日技双十一，假设我们要做一个功能，使用根据用户的装饰等级判断他可以获得的折扣。常规的器简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 的云南idc服务商装饰内部有很多代码。这个函数就会被拉得非常长。器简

有一些同学知道，日技可以使用字典来改写这个太长的使用 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

很赞哦!（2）