前言:
当前各位老铁们对“pythonelseif函数”大约比较注重,同学们都需要学习一些“pythonelseif函数”的相关文章。那么小编同时在网络上搜集了一些关于“pythonelseif函数””的相关知识,希望我们能喜欢,兄弟们快快来学习一下吧!今天在 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等等判断。如果大家有兴趣的话,请在文章下面留言,我们明天就来说说怎么对这个代码进行改造,实现更多的逻辑判断。
标签: #pythonelseif函数