python优化if/else

在任何语言中，if/else都是常见的编程语法，而且因为python中没有switch/case，所以if/else使用频率就会更高。但是当if/else判断分支比较多，而且业务逻辑比较复杂的时候，整体代码就会显得非常臃肿。例如，下面根据不同手机品牌添加对应不同处理逻辑的程序，如果有新的手机品牌，就要不断的写elif分支。对于这种情况有没有更加优雅的写法呢？

def f(name):
    if name == 'Huawei':
        # 业务逻辑
        pass
    elif name == 'Apple':
        # 业务逻辑
        pass
    elif name == 'Xiaomi':
        # 业务逻辑
        pass
    elif name == 'Samsung':
        # 业务逻辑
        pass
    else:
        pass

1. map映射

既然基于不同的name参数，对应不同的业务逻辑，那么我们可以很容易想到，把业务逻辑分别独立抽为一个个业务函数，然后把name和业务逻辑函数保存为一个map关系，这样使用的时候只需要根据name从映射关系中取对应的回调函数就可以了。

def huawei():
    pass


def apple():
    pass


def xiaomi():
    pass


def samsung():
    pass


def other():
    pass


callback_dict = {'Huawei': huawei, 'Apple': apple, 'Xiaomi': xiaomi, 'Samsung': samsung}


def tool(name):
    callback_dict.get(name, other)()

2. 反射调用

上面的方法看起来挺好了，但是我们发现每次新增一个品牌的回调函数以后，都要同步改一下callback_dict，还不够优雅。能不能只需要定义回调函数，就能自动加到逻辑规则中，自动调用呢？我们可以采用反射机制来实现自动注入调用功能。

class Phone:
    def huawei(self):
        pass

    def apple(self):
        pass

    def xiaomi(self):
        pass

    def samsung(self):
        pass

    def other(self):
        pass


def tool(name):
    phone = Phone()
    getattr(phone, name.lower())()

如上，我们把所有的回调函数都封装为统一的类方法，这样新增手机品牌，只需要定义对应的类成员方法就可以了。

3. 子类调用

封装在一个类中虽然具有了自动注入的功能。但是从面向对象的角度看，仍然不够优雅。为了兼容不同的业务处理逻辑，应该把对应的品牌封装为不同的类更加合适，这样就可以在类中定义不同的业务逻辑方法。

class Phone:
    def f1(self): ...
    
    def f2(self): ...
    
    
class Huawei(Phone):
    def f1(self):
        pass
    
    def f2(self):
        pass


class Apple(Phone):
    def f1(self):
        pass

    def f2(self):
        pass


class Xiaomi(Phone):
    def f1(self):
        pass

    def f2(self):
        pass


class Samsung(Phone):
    def f1(self):
        pass

    def f2(self):
        pass
    
    
class Other(Phone):
    def f1(self):
        pass

    def f2(self):
        pass
    
    
def tool(name):
    for sc in Phone.__subclasses__():
        if name == sc.__name__:
            t = sc()
            t.f1()
            t.f2()