Python - 如何计算存在多少 If/Else 语句

Question

我想计算我的函数中有多少个 If/Else 语句。

我的代码如下所示：

def countdown(type):
    if type == 1:
        //code
    elif type == 2:
       //code
    else:
        print(f"You have reached the end of the script. "
              f"The maximum type of countdowns are: {x}")
        exit(1)

x 在哪里，应该有 if 查询的数量（If/Else）。在这种情况下，有 3 个查询。如果我在此函数中创建另一个 if/else 查询，它应该服务于，我不必更改脚本底部的警告。

这可能吗？

我正在使用 Python 3.10

Answer 1

不要使用if..else，而是使用字典或列表：

types = {
    1: ...,
    2: ...
}

try:
    types[type]
except KeyError:
    print(f"You have reached the end of the script. "
          f"The maximum type of countdowns are: {len(types)}")
    exit(1)

究竟将什么放入字典中作为值取决于...您能否概括该算法，以便您只需将值放入字典而不是实际代码？伟大的。否则，将函数放入字典中：

types = {1: lambda: ..., 2: some_func, 3: self.some_method}

...

types[type]()

Answer 2

由于您使用的是 Python 3.10，因此您可以使用新的 match 运算符。一个例子：

def countdown(type):
    match type:
        case 1:
            # code
        case 2:
            # code
        case _:
            print(f"You have reached the end of the script. "
                  f"The maximum type of countdowns are: {x}")
            exit(1)

对我来说，这是一个比dict 更具可读性的解决方案。

如何计算选项的数量，让我们考虑一下n 不同且逻辑分离的选项。在这种情况下，我建议您使用enum：

from enum import IntEnum
class CountdownOption(IntEnum):
    FIRST = 1
    SECOND = 2
    # ...

# ...

def countdown(type):
    match type:
        case CountdownOption.FIRST:
            # code
        case CountdownOption.SECOND:
            # code
        case _:
            print(f"You have reached the end of the script. "
                  f"The maximum type of countdowns are: {len(CountdownOption)}")
            exit(1)