Python中如何用函数封装过于复杂的逻辑判断
如果条件分支里的表达式过于复杂,出现了太多的 not/and/or,那么这段代码的可读性就会大打折扣,比如下面这段代码:
# 如果活动还在开放,并且活动剩余名额大于 10,为所有性别为女性,或者级别大于 3
# 的活跃用户发放 10000 个金币
if activity.is_active and activity.remaining > 10 and \
user.is_active and (user.sex == 'female' or user.level > 3):
user.add_coins(10000)
return
对于这样的代码,我们可以考虑将具体的分支逻辑封装成函数或者方法,来达到简化代码的目的:
if activity.allow_new_user() and user.match_activity_condition():
user.add_coins(10000)
return
事实上,将代码改写后,之前的注释文字其实也可以去掉了。因为后面这段代码已经达到了自说明的目的。至于具体的 什么样的用户满足活动条件? 这种问题,就应由具体的 match_activity_condition() 方法来回答了。
Hint: 恰当的封装不光直接改善了代码的可读性,事实上,如果上面的活动判断逻辑在代码中出现了不止一次的话,封装更是必须的。不然重复代码会极大的破坏这段逻辑的可维护性。
Python中如何用函数封装过于复杂的逻辑判断
According to PEP8, long lines should be placed in parentheses. When using parentheses, the lines can be broken up without using backslashes.
在Python里,如果遇到一长串if-elif-else或者嵌套很深的逻辑判断,确实该考虑用函数来封装了。这么干能让代码更干净、好测试,也更容易复用。
核心思路就是把判断条件本身抽象成一个独立的、有明确返回值的函数。这个返回值通常是布尔值(True/False),但也可以是其他能明确表示逻辑结果的数据。
给你看个典型的例子。假设我们要根据用户的年龄和VIP状态来判断他能不能进入某个区域,原来的代码可能是一堆判断:
# 原来的复杂逻辑判断
age = 25
is_vip = True
has_invitation = False
if age >= 18:
if is_vip:
print("允许进入VIP区")
else:
if has_invitation:
print("允许进入普通区")
else:
print("不允许进入")
else:
print("未成年人禁止进入")
我们可以把核心的判断逻辑抽出来,放到一个函数里:
def can_enter_area(age, is_vip, has_invitation):
"""判断用户是否可以进入区域"""
# 先检查年龄
if age < 18:
return "未成年人禁止进入"
# 成年人再根据VIP状态和邀请函判断
if is_vip:
return "允许进入VIP区"
elif has_invitation:
return "允许进入普通区"
else:
return "不允许进入"
# 使用函数
result = can_enter_area(25, True, False)
print(result) # 输出:允许进入VIP区
更进一步,如果逻辑更复杂,可以考虑拆分成多个小函数,每个函数负责一个具体的子判断:
def is_adult(age):
"""判断是否成年"""
return age >= 18
def has_vip_access(is_vip):
"""判断是否有VIP权限"""
return is_vip
def has_invitation_access(has_invitation):
"""判断是否有邀请函权限"""
return has_invitation
def check_access(age, is_vip, has_invitation):
"""综合判断访问权限"""
if not is_adult(age):
return "未成年人禁止进入"
if has_vip_access(is_vip):
return "允许进入VIP区"
if has_invitation_access(has_invitation):
return "允许进入普通区"
return "不允许进入"
# 使用
print(check_access(25, True, False)) # 允许进入VIP区
print(check_access(20, False, True)) # 允许进入普通区
print(check_access(16, False, False)) # 未成年人禁止进入
对于更复杂的业务规则,可以考虑使用策略模式或者规则引擎,但大多数情况下,拆分成函数就够用了。
总结:把判断条件打包成函数,调用时传参数就行。
如果没有需要重用,避免过早优化
这种问题只要整个 team 统一用一个 formatter 和配置 就好, 比如 black
编程基础第一节就没必要科普了吧?
二楼说得对
谢谢楼主不辞辛劳从火星赶来告诉大家大清忘了。

