如何为调用类方法构造实例的泛型类方法添加正确类型提示？

核心思路 #

通过绑定到基类的自引用类型变量，让Mypy识别子类的具体类型，同时允许调用子类特有的方法，解决类型不兼容和属性不存在的双重错误。

修改后的代码示例（兼容Python 3.8+） #

from abc import ABC, abstractmethod
from typing import TypeVar, Type

# 定义自引用类型变量，约束为A的子类
T = TypeVar('T', bound='A')

class A(ABC):
    @classmethod
    @abstractmethod
    def create(cls: Type[T], data: dict) -> T:
        """抽象创建方法，返回子类实例"""
        pass

class B(A):
    @classmethod
    @abstractmethod
    def create_b(cls, data: dict) -> 'B':
        """子类特有的创建逻辑"""
        pass

    @classmethod
    def create(cls: Type['B'], data: dict) -> 'B':
        # Mypy可识别cls为B的类型，确认存在create_b方法
        return cls.create_b(data)

Python 3.11+简化写法 #

如果使用Python 3.11及以上版本，可直接用标准库的Self类型，代码更简洁直观：

from abc import ABC, abstractmethod
from typing import Self

class A(ABC):
    @classmethod
    @abstractmethod
    def create(cls, data: dict) -> Self:
        pass

class B(A):
    @classmethod
    @abstractmethod
    def create_b(cls, data: dict) -> Self:
        pass

    @classmethod
    def create(cls, data: dict) -> Self:
        return cls.create_b(data)

关键修改说明 #

• 自引用TypeVar：TypeVar('T', bound='A')让Mypy明确T是A的子类，子类实现create时，T会被自动推断为对应子类（如B），此时Type[T]就是子类的类型，自然包含子类特有的方法。
• 子类显式标注类型：在B的create方法中指定cls: Type['B']和返回类型'B'，消除Mypy对“是否存在create_b”的疑问。
• Self类型优势：Python 3.11引入的Self会自动指代当前类的实例类型，无需手动定义TypeVar，代码更简洁。

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