Python函数基于用户输入按需执行步骤 替代逐选项传布尔参数的优化方案

逐选项传布尔值、空值占位参数的写法，会随着可选操作数量增加产生大量重复代码：每新增一个可选操作，就需要同步修改函数签名、在所有调用分支补全对应空值参数、在函数内部新增if判断，扩展性极差。

方案1：以集合形式传入待触发操作，替代逐布尔参数 #

不再为每个独立操作单独拆分形参，直接把用户选择对应的待触发操作名拼接为集合传入，函数内部通过集合成员判断决定是否执行对应逻辑。
实现示例：

choice = input("Select option: ")
# 维护选项和触发操作的映射表，集中管理所有分支逻辑，不需要逐分支传空值
option_to_ops = {
    "1": {"apple"},
    "2": {"pear"},
    "3": set()
}
# 无论用户选哪个选项，只需要传一次集合参数即可
test_operation(user, enabled_ops=option_to_ops[choice])


def test_operation(user, enabled_ops):
    # 按需执行对应操作，不需要每个操作对应一个传入的布尔值
    if "apple" in enabled_ops:
        # 执行apple对应的逻辑
    if "pear" in enabled_ops:
        # 执行pear对应的逻辑
    # 兜底逻辑：没有触发任何指定操作时，执行双场景通用逻辑
    if not enabled_ops:
        # 执行两个操作都覆盖的通用逻辑

    # 所有场景通用的打印逻辑完全不需要改动
    for category in operations:
        print(f"\n{category}")
        for task in operations[category]:
            print(f"{task}: {operations[category][task]}")

这个方案的核心优势：

• 所有分支逻辑集中在映射表中维护，不需要逐分支给无关参数传空值占位
• 新增可选操作时不需要修改主函数的签名，避免为所有旧调用位置补全新参数的重复工作
• 顺带修正原代码的逻辑bug：原代码中if pear...else: do both的分支逻辑，会在apple为真、pear为假的场景下误跳过兜底逻辑，用集合判断可以严格匹配“无指定操作时执行双逻辑”的预期，不会出现分支错配

方案2：操作注册表+动态执行，彻底解耦操作和主流程 #

把所有可选操作封装为独立的可执行单元，预先注册到全局字典中，主函数完全不需要硬编码每个操作的if判断，只需要遍历传入的操作集合批量执行即可。
实现示例：

# 预先注册所有可选操作的独立逻辑，和主流程完全解耦
operation_registry = {
    "apple": lambda: 执行apple对应的操作逻辑,
    "pear": lambda: 执行pear对应的操作逻辑
}

def test_operation(user, enabled_ops):
    # 批量执行所有触发的操作，不需要逐个写if判断
    for op_name in enabled_ops:
        operation_registry[op_name]()
    
    # 兜底逻辑保持和之前一致
    if not enabled_ops:
        # 执行双场景通用逻辑

    # 通用打印逻辑不变

这个方案的核心优势：

• 调整单个操作的逻辑时，只需要修改注册表里的对应条目，不需要动主函数的判断逻辑
• 支持操作的灵活组合，比如多个用户选项可以复用同一组操作逻辑，不需要为单个选项写单独的调用分支
• 如果某个操作需要额外入参，只需要把传入的集合换成键为操作名、值为参数字典的结构即可，依然不需要给主函数堆砌零散的形参

内容的提问来源于stack exchange，提问作者user10404568