mypy处理含混合类型的变长元组时报错，该如何解决？

问题根源是mypy无法自动推断出列表中每个变长元组的精确类型——它会把所有元组统一推断为 tuple[int | str, ...]，导致i被识别为int | str（不符合hex()的参数要求），s被识别为list[int | str]（不符合str.join()的参数要求）。

下面是几种可行的解决方法：

方法1：使用Python 3.10+的模式匹配（最优雅） #

模式匹配可以精确匹配不同长度的元组，mypy能直接推断出每个分支的类型：

for item in [(1, 'a'), (2, 'b', 'c')]:
    match item:
        case i, s:
            print(hex(i), '_'.join([s]))
        case i, s1, s2:
            print(hex(i), '_'.join([s1, s2]))

方法2：显式类型断言 #

如果你无法使用模式匹配，可以通过cast告诉mypy变量的实际类型：

from typing import cast, Tuple, Union

# 先给列表标注精确的联合类型
tuples: list[Union[Tuple[int, str], Tuple[int, str, str]]] = [(1, 'a'), (2, 'b', 'c')]

for i, *s in tuples:
    # 断言i是int，s是字符串列表
    i_int = cast(int, i)
    s_strs = cast(list[str], s)
    print(hex(i_int), '_'.join(s_strs))

方法3：自定义类型守卫 #

通过类型守卫函数让mypy识别符合要求的元组结构，适合需要大量校验的场景：

from typing import TypeGuard, Tuple, Any

def is_valid_tuple(t: Tuple[Any, ...]) -> TypeGuard[Tuple[int, str, ...]]:
    """校验元组第一个元素是int，其余都是str"""
    if not t:
        return False
    if not isinstance(t[0], int):
        return False
    return all(isinstance(item, str) for item in t[1:])

tuples = [(1, 'a'), (2, 'b', 'c')]
for item in tuples:
    if is_valid_tuple(item):
        i, *s = item
        print(hex(i), '_'.join(s))

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