Python如何根据指定键名返回对应类实例，替代大量if判断

Python原生的字典（dict）就是和TypeScript中Record<K,T>完全对等的结构，不需要依赖大量if分支就能实现键到实例/构造规则的映射，代码可维护性远高于分支判断写法。

方案1：每次查询返回新初始化实例 #

和最初的if逻辑完全等价，每次调用查询方法返回全新的类实例，只需要提前维护键对应的构造参数映射即可：

from typing import Dict, Tuple

class Ex:
    def __init__(self, var1: int, var2: int, var3: int):
        self.var1 = var1
        self.var2 = var2
        self.var3 = var3

# 键与类构造参数的映射表，新增配置只需要在这里加行
PARAM_MAP: Dict[str, Tuple[int, int, int]] = {
    "A": (1, 2, 3),
    "B": (2, 3, 4),
    "C": (4, 5, 6)
}

def find(key: str) -> Ex:
    if key not in PARAM_MAP:
        raise KeyError(f"未找到键{key}对应的实例配置")
    return Ex(*PARAM_MAP[key])

如果类初始化用关键字参数更清晰，可以把参数存成字典格式，解包传入：

PARAM_MAP: Dict[str, Dict[str, int]] = {
    "A": {"var1": 1, "var2": 2, "var3": 3},
    "B": {"var1": 2, "var2": 3, "var3": 4},
    "C": {"var1": 4, "var2": 5, "var3": 6}
}

def find(key: str) -> Ex:
    return Ex(**PARAM_MAP[key])

方案2：预初始化固定实例直接返回 #

和给出的TS Record示例逻辑完全一致，提前初始化好所有实例，查询时直接返回已有对象，适合实例不需要动态生成的场景：

from typing import Dict, Literal

# 用Literal做类型约束，等价于TS的联合类型键定义
PoolKey = Literal["A", "B", "C"]

class Pool:
    def __init__(self, var1: int, var2: int, var3: int):
        self.var1 = var1
        self.var2 = var2
        self.var3 = var3

# 以下字典完全等价于TS中的Record<MainnetPoolKey, Pool>定义
MAINNET_POOLS: Dict[PoolKey, Pool] = {
    "A": Pool(var1=86498781, var2=0, var3=686505742),
    "B": Pool(var1=686500029, var2=31566704, var3=86508050),
    "C": Pool(var1=86500844, var2=312769, var3=509463)
}

def find(key: PoolKey) -> Pool:
    return MAINNET_POOLS[key]

相比堆叠if的优势 #

• 配置与逻辑分离：所有键对应的实例规则统一维护在映射表中，新增、修改、删除配置不需要改动查询函数的逻辑
• 查询性能更高：字典基于哈希表实现，查询时间复杂度为O(1)，if分支需要逐行判断，时间复杂度为O(n)，键的数量越多性能差距越明显
• 冗余度极低：哪怕有上百个待匹配的键，查询函数的逻辑也不需要做任何改动
• 类型支持完善：配合Python内置的typing模块做类型标注后，IDE可以实现和TS一致的键名自动补全、类型校验能力

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