巧妙使用Python装饰器处理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等等判断。如果大家有兴趣的话，请在文章下面留言，我们明天就来说说怎么对这个代码进行改造，实现更多的逻辑判断。
参考文献
[1] EdgeDB: https://github.com/edgedb/edgedb
[2] 源代码: https://github.com/edgedb/edgedb/blob/master/edb/common/value_dispatch.py
技术交流
欢迎转载、收藏、有所收获点赞支持一下！
到此这篇关于巧妙使用Python装饰器处理if...elif...else的文章就介绍到这了,更多相关Python 装饰器内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
