基类基于值类型参数化时的Python类型提示问题

现有如下Python代码，运行正常能输出预期结果，但Pylance会对res1.y和res2.z的访问抛出类型错误：

from typing import Type
from dataclasses import dataclass, asdict
import json

@dataclass
class ValueType1:
    x: int
    y: int

@dataclass
class ValueType2:
    x: int
    z: str

class FooBase:
    def __init__(self, value_cls: Type[ValueType1|ValueType2], name: str):
        self.name = name
        self.value_cls = value_cls
    def save(self, value: ValueType1|ValueType2):
        with open(self.name+'.json', 'w') as f:
            json.dump(asdict(value), f)
    def load(self) -> ValueType1|ValueType2:
        with open(self.name+'.json', 'r') as f:
            return self.value_cls(**json.load(f))
    
class Foo1(FooBase):
    def __init__(self):
        super().__init__(value_cls=ValueType1, name='1')
class Foo2(FooBase):
    def __init__(self):
        super().__init__(value_cls=ValueType2, name='2')

foo1 = Foo1()
foo2 = Foo2()
foo1.save(ValueType1(x=10, y=20))
foo2.save(ValueType2(x=10, z='a'))
res1 = foo1.load()
res2 = foo2.load()
print(res1.y)
print(res2.z)

Pylance针对res1.y的错误提示如下：

Type of "y" is unknownPylancereportUnknownMemberType
Type of "y" is partially unknown
  Type of "y" is "int | Unknown"PylancereportUnknownMemberType
Argument type is partially unknown
  Argument corresponds to parameter "values" in function "print"
  Argument type is "int | Unknown"PylancereportUnknownArgumentType
Cannot access member "y" for type "ValueType2"
  Member "y" is unknownPylancereportGeneralTypeIssues
(variable) y: int | Unknown

需要找到合适的类型提示方案，让类型检查工具明确识别res1为ValueType1类型、res2为ValueType2类型。

通过泛型类型变量，让FooBase的子类与对应的ValueType建立明确的类型关联，类型检查工具就能准确推断load方法的返回类型。

修改后的代码 #

from typing import Type, Generic, TypeVar
from dataclasses import dataclass, asdict
import json

# 定义类型变量，约束为ValueType1或ValueType2
T = TypeVar('T', ValueType1, ValueType2)

@dataclass
class ValueType1:
    x: int
    y: int

@dataclass
class ValueType2:
    x: int
    z: str

class FooBase(Generic[T]):
    def __init__(self, value_cls: Type[T], name: str):
        self.name = name
        self.value_cls: Type[T] = value_cls
    def save(self, value: T):
        with open(self.name+'.json', 'w') as f:
            json.dump(asdict(value), f)
    def load(self) -> T:
        with open(self.name+'.json', 'r') as f:
            return self.value_cls(**json.load(f))
    
class Foo1(FooBase[ValueType1]):
    def __init__(self):
        super().__init__(value_cls=ValueType1, name='1')

class Foo2(FooBase[ValueType2]):
    def __init__(self):
        super().__init__(value_cls=ValueType2, name='2')

foo1 = Foo1()
foo2 = Foo2()
foo1.save(ValueType1(x=10, y=20))
foo2.save(ValueType2(x=10, z='a'))
res1 = foo1.load()  # 类型检查器自动识别为ValueType1
res2 = foo2.load()  # 类型检查器自动识别为ValueType2
print(res1.y)
print(res2.z)

关键改动说明 #

1. 定义类型变量T：用TypeVar指定T只能是ValueType1或ValueType2，明确类型范围。
2. 泛型化FooBase：让FooBase继承Generic[T]，使其成为与T绑定的泛型类。
3. 精确类型注解：
   • 将__init__的value_cls和实例属性self.value_cls标注为Type[T]
   • save方法的参数value改为T，确保只能传入对应类型的实例
   • load方法返回类型改为T，直接关联到当前类绑定的具体类型
4. 子类指定具体类型：Foo1继承FooBase[ValueType1]，Foo2继承FooBase[ValueType2]，明确每个子类对应的ValueType类型。

如果希望更灵活的类型约束（比如后续新增ValueType子类），也可以创建一个基类，再用bound约束类型变量：

@dataclass
class BaseValueType:
    x: int

@dataclass
class ValueType1(BaseValueType):
    y: int

@dataclass
class ValueType2(BaseValueType):
    z: str

T = TypeVar('T', bound=BaseValueType)

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