让Python函数在基于字符串输入的不同模式下执行的干净方法是什么

def mode_a_handler(input_str):
    return f"处理模式A: {input_str}"

def mode_b_handler(input_str):
    return f"处理模式B: {input_str}"

def default_handler(input_str):
    return f"未知模式: {input_str}"

# 创建一个字典，将模式映射到相应的处理函数
mode_handlers = {
    "A": mode_a_handler,
    "B": mode_b_handler
}

def execute_based_on_mode(mode, input_str):
    # 获取对应的处理函数，如果没有找到则使用默认处理函数
    handler = mode_handlers.get(mode, default_handler)
    # 执行处理函数并返回结果
    return handler(input_str)

# 测试代码
print(execute_based_on_mode("A", "测试字符串1"))  # 输出: 处理模式A: 测试字符串1
print(execute_based_on_mode("B", "测试字符串2"))  # 输出: 处理模式B: 测试字符串2
print(execute_based_on_mode("C", "测试字符串3"))  # 输出: 未知模式: 测试字符串3