如何将类属性用作类内部方法的类型提示？

有两种可行的方式实现这个需求，让类型检查器（比如VS Code的IntelliSense）正确识别子类中method方法的返回类型：

方法1：使用泛型（兼容Python 3.9+） #

通过TypeVar和Generic定义泛型父类，将返回类型与子类指定的method_type绑定：

from typing import TypeVar, Generic

# 定义类型变量，代表子类中method_type的具体类型
T = TypeVar('T')

class Foo(Generic[T]):
    # 标注method_type为T的类型对象
    method_type: type[T]

    def method(self) -> T:
        # 示例实现：返回method_type的实例
        return self.method_type()

# 子类继承时指定具体类型
class FooStr(Foo[str]):
    method_type = str

foo_str = FooStr().method() # IntelliSense会识别为str类型

class FooInt(Foo[int]):
    method_type = int

foo_int = FooInt().method() # IntelliSense会识别为int类型

方法2：使用Self类型（Python 3.11+） #

利用Python 3.11新增的Self类型，直接引用当前类的method_type属性：

from __future__ import annotations
from typing import Self

class Foo:
    method_type: type

    def method(self) -> Self.method_type:
        # 示例实现：返回method_type的实例
        return self.method_type()

class FooStr(Foo):
    method_type = str

foo_str = FooStr().method() # IntelliSense会识别为str类型

class FooInt(Foo):
    method_type = int

foo_int = FooInt().method() # IntelliSense会识别为int类型

为什么直接写method_type不生效？ #

在父类Foo中直接写-> method_type时，类型检查器只会将其识别为type类型，无法关联到子类中具体的类型（比如str或int）。上述两种方法通过泛型或Self类型，让类型检查器能够跟踪子类对method_type的具体赋值，从而正确推断返回类型。

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