Python小技巧学起来！如何简化大量的ifelifelse代码？

　　今天在 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 的内部有很多代码。这个函数就会被拉得非常长。
　　有一些同学知道，可以使用字典来改写这个太长的 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等等判断。如果大家有兴趣的话，请在文章下面留言，我们明天就来说说怎么对这个代码进行改造，实现更多的逻辑判断。