Python多模式类初始化如何消除if分支 采用设计模式优化代码

你遇到的场景可以通过两种常用方案消除if分支，分别适配不同的复杂度需求：

方案1：映射表实现（轻量化首选） #

Python支持函数作为一等公民，直接用字典维护模式值和对应方法的映射关系即可，是该场景下性价比最高的实现：

class SomeClass():
    # 类级别的模式映射表，仅初始化一次
    _mode_handler_map = {
        1: lambda self: self.mode1(),
        2: lambda self: self.mode2(),
        3: lambda self: self.mode3(),
        # 按需补充到模式10即可
        10: lambda self: self.mode10()
    }

    def __init__(self, m):
        try:
            self._mode_handler_map[m](self)
        except KeyError:
            raise ValueError(f"非法模式值: {m}, 仅支持1~10")

    def mode1(self):
        # 原有mode1逻辑保持不变
        pass
    
    # mode2~mode10的方法无需修改

优势：

• 所有模式对应关系集中维护，新增模式仅需在映射表加一行配置，无需修改初始化逻辑，符合开闭原则
• 完全消除分支判断，代码可读性和维护性大幅提升
• 无额外依赖，性能开销几乎可以忽略

方案2：策略模式（适合复杂场景） #

如果每个模式的逻辑复杂度高、后续扩展频繁，可以用标准的策略模式做更规范的实现：

from abc import ABC, abstractmethod

# 抽象策略基类
class BaseModeStrategy(ABC):
    @abstractmethod
    def init_mode(self, instance):
        """初始化对应模式的逻辑"""
        pass

# 各个模式的具体策略实现
class Mode1Strategy(BaseModeStrategy):
    def init_mode(self, instance):
        # 原有mode1的逻辑
        print("模式1初始化完成")

class Mode2Strategy(BaseModeStrategy):
    def init_mode(self, instance):
        # 原有mode2的逻辑
        print("模式2初始化完成")

# 其他模式的策略类以此类推

class SomeClass():
    _strategy_map = {
        1: Mode1Strategy(),
        2: Mode2Strategy(),
        # 补全所有模式对应的策略实例
        10: Mode10Strategy()
    }

    def __init__(self, m):
        try:
            strategy = self._strategy_map[m]
            strategy.init_mode(self)
        except KeyError:
            raise ValueError(f"非法模式值: {m}, 仅支持1~10")

适用场景：

• 单个模式的初始化逻辑超过30行，单独抽类更易维护
• 不同模式之间有公共的抽象逻辑，可以放在基类复用
• 后续需要频繁新增/修改模式逻辑，需要严格遵守开闭原则

简化技巧（仅适合内部小项目） #

如果你的模式方法命名严格遵循mode+数字的规则，还可以用反射进一步省去手动维护映射表的成本：

class SomeClass():
    def __init__(self, m):
        try:
            getattr(self, f"mode{m}")()
        except AttributeError:
            raise ValueError(f"非法模式值: {m}, 仅支持1~10")

    def mode1(self):
        pass
    # 其他mode方法保持不变

注意该方案的方法名和模式值强绑定，重构方法名时容易出现隐式bug，不推荐对外发布的公共类使用。

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