如何让Python 3.7数据类(dataclass)变量支持多种数据类型？

问题根源 #

你用tuple or str作为类型注解是错误的——or是Python逻辑运算符，这里会被解析为返回第一个真值类型tuple，所以Pylance会认为index_range必须是tuple类型，而默认值"all"是字符串，自然触发类型不匹配报错。

双类型支持的正确写法 #

在Python 3.7中，需要使用typing.Union来声明字段支持多种类型，示例代码如下：

from dataclasses import dataclass
from typing import Union

@dataclass
class C:
    index_range: Union[tuple, str] = "all"

这样index_range既可以接收tuple类型值，也可以接收str类型值，默认值"all"完全符合类型注解要求，Pylance不会再报错。

三种及以上类型的支持 #

完全支持三种或更多类型，只需在Union中添加对应的类型即可。比如要支持tuple、None、str三种类型：

from dataclasses import dataclass
from typing import Union, Optional  # Optional是Union[T, None]的语法糖

@dataclass
class C:
    # 写法1：直接用Union包含所有类型
    index_range: Union[tuple, str, None] = "all"
    # 写法2：用Optional简化None的声明（效果和上面一致）
    # index_range: Union[tuple, Optional[str]] = "all"

后续逻辑处理建议 #

可以通过isinstance()判断字段实际类型，执行对应业务逻辑：

c = C()
if isinstance(c.index_range, str) and c.index_range == "all":
    # 处理全索引逻辑
    pass
elif isinstance(c.index_range, tuple):
    # 处理指定范围逻辑
    pass
elif c.index_range is None:
    # 处理None值逻辑
    pass

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