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

　　今天在Github阅读EdgeDB〔1〕的代码，发现它在处理大量if。。。elif。。。else判断的时候，使用了一个非常巧妙的装饰器。我们来看看这个方法具体是什么样的。
　　正好双十二快到了，假设我们要做一个功能，根据用户的等级判断他可以获得的折扣。常规的if。。。elif。。。写法是这样的：defgetdiscount（level）：iflevel1：大量计算代码discount0。1eliflevel2：大量计算代码discount0。2eliflevel3：discount0。3eliflevel4：discount0。4eliflevel5：discount0。5eliflevel6：discount3250。1else：return等级错误returndiscount
　　大家都知道，这样大量的if。。。elif。。。代码非常难看，也很难维护。并且每个if的内部有很多代码。这个函数就会被拉得非常长。
　　有一些同学知道，可以使用字典来改写这个太长的if判断：defparselevel1（）：大量计算代码discount0。1returndiscountdefparselevel2（）：大量计算代码discount0。2returndiscountdefparselevel3（）：大量计算代码discount0。3returndiscountdefparselevel4（）：大量计算代码discount0。4returndiscountdefparselevel5（）：大量计算代码discount0。5returndiscountdefparselevel6（）：大量计算代码discount3250。1returndiscountdiscountmap｛1：parselevel1，2：parselevel2，3：parselevel3，4：parselevel4，5：parselevel5，6：parselevel6，｝discountdiscountmap。get（level，等级错误）
　　但今天我学到的这个方法，比用字典更简单。我们先来看它的效果：valuedispatchdefgetdiscount（level）：return等级错误getdiscount。register（1）defparselevel1（level）：大量计算代码discount0。1returndiscountgetdiscount。register（2）defparselevel2（level）：大量计算代码discount0。2returndiscountgetdiscount。register（3）defparselevel3（level）：大量计算代码discount0。3returndiscountgetdiscount。register（4）defparselevel4（level）：大量计算代码discount0。4returndiscountgetdiscount。register（5）defparselevel5（level）：大量计算代码discount0。5returndiscountgetdiscount。register（6）defparselevel1（level）：大量计算代码discount3250。1returndiscountdiscountgetdiscount（3）print（f等级3的用户，获得的折扣是：｛discount｝）
　　运行效果如下图所示：
　　这样写，比用字典的方式更直观，比直接用if。。。elif。。。更简洁。
　　那么，这个装饰器valuedispatch是怎么实现的呢？密码就藏在这个开源项目EdgeDB的源代码〔2〕中，核心代码只有20多行：
　　并且，还能够实现或查询。例如用户等级为2或者3的时候，折扣都是0。2，那么代码可以写成：getdiscount。register（2）getdiscount。register（3）defparselevel2（level）：大量计算代码discount0。2returndiscount
　　运行效果如下图所示：
　　它这个代码目前只能实现相等的查询。但其实只要对这个代码稍作修改，我们就能实现大于、小于、大于等于、小于等于、不等于、in等等判断。如果大家有兴趣的话，请在文章下面留言，我们明天就来说说怎么对这个代码进行改造，实现更多的逻辑判断。