Python中如何基于类变量指定类方法的返回类型？

我代码库采用数据类（dataclass）+ 执行类的配对结构，数据类用作数据收集器。执行类通过类变量关联对应的数据类，实例化数据类没问题，但没法给return_name方法指定正确的返回类型——它应该返回类变量对应的实例。

示例里把返回类型设为基类Name，导致调用时访问子类BiggerName的other_name属性会出现类型提示错误。直接用类变量_INDIVIDAL当返回类型会报错，想问Python 3.8及以上版本支不支持这种依赖类变量的类型注解？

示例代码 #

from abc import ABC
from dataclasses import dataclass
from typing import ClassVar

@dataclass
class Name(ABC):
    name: str


class RelatedName(ABC):
    _INDIVIDAL: ClassVar[Name]

    def return_name(self, **properties) -> Name:
        # There is a typing issue here too but you can ignore that for now
        return self._INDIVIDAL(**properties)  


@dataclass
class BiggerName(Name):
    other_name: str

class RelatedBiggerName(RelatedName):
    _INDIVIDAL: ClassVar[Name] = BiggerName


if __name__ == "__main__":
    biggie = RelatedBiggerName()
    biggiename = biggie.return_name(name="Alfred", other_name="Biggie").other_name
    print(biggiename)

解决方案 #

要实现这种依赖类变量的类型注解，得用**泛型（Generics）**结合TypeVar来关联执行类和对应的数据类，让类型检查器能正确推断子类的返回类型。Python 3.8及以上版本完全支持这种写法。

修改后的代码示例：

from abc import ABC
from dataclasses import dataclass
from typing import ClassVar, TypeVar, Generic

# 定义类型变量，绑定到Name及其子类
T = TypeVar('T', bound='Name')

@dataclass
class Name(ABC):
    name: str

class RelatedName(ABC, Generic[T]):
    # 类变量的类型设为对应数据类的类型
    _INDIVIDAL: ClassVar[type[T]]

    def return_name(self, **properties) -> T:
        return self._INDIVIDAL(**properties)

@dataclass
class BiggerName(Name):
    other_name: str

class RelatedBiggerName(RelatedName[BiggerName]):
    _INDIVIDAL: ClassVar[type[BiggerName]] = BiggerName

if __name__ == "__main__":
    biggie = RelatedBiggerName()
    # 类型检查器可识别返回值为BiggerName，访问other_name无类型错误
    biggiename = biggie.return_name(name="Alfred", other_name="Biggie").other_name
    print(biggiename)

关键说明 #

1. 用TypeVar定义T并绑定到Name基类，确保只能传入Name或其子类；
2. 让RelatedName继承Generic[T]，将执行类与对应数据类的类型绑定；
3. 类变量_INDIVIDAL的类型设为type[T]，明确它是T对应的类类型；
4. return_name的返回类型设为T，子类继承时会自动替换为具体的数据类类型，类型检查器能正确推断返回值的属性。

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