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等等判断。如果大家有兴趣的话,请在文章下面留言,我们明天就来说说怎么对这个代码进行改造,实现更多的逻辑判断。
蓝天野圆满归去追思会上众人回忆他不做手术不化疗的人生选择从6月8日去世,人们似乎一直在等待这样一场告别仪式,7月5日,北京人艺召开蓝天野追思会,爱他的人们终于坐在一起,用朴素真挚的语言追忆他描摹他留住他张和平他给我写过三封信,两封是毛笔
花式避暑新玩法人均200承包一夏的吃喝玩乐在泰山北麓的秘境城堡中,藏着一个新潮浪漫的宝藏野奢休闲主题乐园,绝对不输上海迪士尼和广州长隆。人均200,一站式游乐度假,集合了你能想到的所有浪漫玩法!不管是家庭出游姐妹拍照情侣约
胃不好的人吃什么养胃?首先有请主食登场什么是主食?这个我个人的理解是身体摄入最多,胃吸收消化最多,而且获取容易的食物。是不是很多朋友就想到了大米?但是在这里不推荐大米,知道为什么吗?让我来跟大家解释解释
学习不好是专注力问题?不要再被骗了,带你了解专注力的本质孩子专注力有没有问题,只要看一点,孩子做自己喜欢的事,注意力是否集中?如果比较专心,那说明孩子的专注力是正常的。如果注意力涣散无法集中,那就说明有问题。专注力有问题不是训练可以解决
问界M7发布2小时订单破万辆,经济不好为啥买车还这么猛7月4日,由赛力斯汽车和华为联合设计的增程式电动SUV问界M7正式发布,同时开始接受预订。从2小时订单破万辆的成绩看,这又是一个爆款车型,而且凶悍程度远超6月21日刚刚发布的理想L
有哪些句子治愈最近心情不好的你?1慢品人间烟火色,闲观万事岁月长。2生活总是来来往往,千万别等来日方长。3借一把清风吹开阴霾,接一碗烈酒谈笑风生。4要珍惜的人很多你在其中名列前茅。5愿你内心山河壮阔,始终相信人间
望丛祠的古韵荷塘上周五早上,下过雨,天气比较凉快,才有了出门的勇气。与女儿约好去望丛祠拍荷花,按导航的停车场过去,结果还是像去年12月去过的那样,门口的停车区域整改,还是拦着进不去。这都过去半年时
国之大道G219新藏线,盘它之前你应该了解清楚的10个问题自驾人越玩越野,如今,G318逐渐不香了,而盘G219的人越来越多。我刚刚走完新藏线,趁着感受还新鲜,赶快给大家说道说道。1G219概况国之大道G219,有新旧两个概念。老的G21
驼奶怎么喝比较好?在现阶段疫情常态化的大环境下,人们越来越重视日常饮食健康。中国居民膳食指南建议每人每天要摄取300g乳制品,而奶作为乳制品的初始原料,营养丰富,为机体提供所需的营养和能量。因此,乳
又到了吃西瓜的季节,安心吃西瓜,这么做炎炎夏日,吃西瓜是不少人消暑的首选中国人有多爱吃西瓜?国家统计局专门就西瓜算过一笔账如果按一个西瓜10斤计算,中国人每年要吃掉160亿个西瓜666你是怎么吃西瓜呢?勺子挖着吃?还是
认识胆固醇四不用在意鸡蛋中的胆固醇问你一个问题,在你吃鸡蛋的时候,是不是会想起,有人说过蛋黄是高胆固醇食物,尽量不要吃?如果你信了,害怕吃了蛋黄会影响体内的胆固醇,并且在吃鸡蛋的时候把蛋黄扔了。真是暴殄天物啊。我的