python之switch语句，优化多个if语句

Posted 2021-03-06 fjc0000

　　python中并没有多分支的语句。像c语言中有switch语句，可以避免多个if的使用场合，简化代码。

　　python若想实现多分支的功能需要自己构建代码，涉及到装饰器的知识点。下面举个例子。

switch_dicts = {}


def deco(data):
    def wrapper(func):
        if data not in switch_dicts.keys():
            switch_dicts[data] = func

        def wrapper1(*args, **kwargs):
            return func(*args, **kwargs)

        return wrapper1

    return wrapper


@deco(1)
def case1(*args, **kwargs):
    print("case1")


@deco(2)
def case1(*args, **kwargs):
    print("case2")


@deco(3)
def case1(*args, **kwargs):
    print("case3")

#装饰器自动运行时，会自动将1，2，3装入字典中
print(switch_dicts)
调用字典中key为1所指向的函数
print(switch_dicts[1]())