面试题：Python if语句在复杂场景下的优化运用

Question

在一个大型的Python项目中，有一个函数需要处理多种复杂业务逻辑，每种逻辑由不同的条件决定（假设条件数量较多且相互关联）。目前使用了大量嵌套的if语句，导致代码可读性和维护性较差。请阐述如何优化这部分代码结构，提高代码的可维护性和扩展性，并给出优化示例代码。

Answer 1

1. 优化思路
   • 使用字典映射：将条件和对应的处理逻辑用字典进行映射，这样可以通过条件直接获取对应的处理函数，避免大量嵌套if语句。
   • 函数封装：将每种业务逻辑封装成独立的函数，使代码结构更清晰，便于理解和维护。
2. 优化示例代码

# 假设这些是不同业务逻辑的处理函数
def handle_logic1():
    print("处理逻辑1")


def handle_logic2():
    print("处理逻辑2")


def handle_logic3():
    print("处理逻辑3")


# 构建条件与处理函数的映射字典
condition_handler = {
    'condition1': handle_logic1,
    'condition2': handle_logic2,
    'condition3': handle_logic3
}


def complex_handler(condition):
    if condition in condition_handler:
        handler = condition_handler[condition]
        handler()
    else:
        print("未找到对应的处理逻辑")


# 调用示例
complex_handler('condition2')

在上述代码中：

• 首先定义了不同业务逻辑的处理函数handle_logic1、handle_logic2和handle_logic3。
• 然后创建了一个字典condition_handler，将条件（这里用字符串模拟）和对应的处理函数进行映射。
• complex_handler函数接收一个条件，通过字典映射找到对应的处理函数并执行。如果条件不存在于字典中，则输出提示信息。这样代码结构更清晰，当需要添加新的业务逻辑时，只需添加新的处理函数并更新映射字典即可，提高了可维护性和扩展性。