如何为工厂方法添加类型提示？解决mypy抽象类实例化报错

这个问题我很熟悉！mypy之所以报错，是因为它没办法自动推断AlgorithmBase.__subclasses__()返回的子类都是可实例化的具体类——虽然你我都知道代码里不会实例化抽象基类，但mypy只会根据类型注解来判断。下面提供几种不需要用Union硬编码子类的优雅解决方案：

方案1：使用TypeVar绑定具体子类（最推荐） #

通过定义一个绑定到AlgorithmBase的TypeVar，我们可以告诉mypy：_algorithm_types里的类都是AlgorithmBase的具体子类，返回的实例也是这些具体子类的对象。

from typing import TypeVar, Type
import random

# 定义TypeVar，绑定到AlgorithmBase的非抽象子类
ConcreteAlgorithm = TypeVar('ConcreteAlgorithm', bound=AlgorithmBase)

class AlgorithmFactory:
    # 用cast告诉mypy，__subclasses__返回的都是可实例化的具体子类
    _algorithm_types: list[Type[ConcreteAlgorithm]] = cast(
        list[Type[ConcreteAlgorithm]],
        AlgorithmBase.__subclasses__()
    )
    
    def __init__(self):
        self._random_generator = random.Random()

    def select_random_algorithm(self) -> ConcreteAlgorithm:
        algorithm_class = self._random_generator.choice(self._algorithm_types)
        return algorithm_class()

这个方案的好处是类型注解更精确，mypy能正确识别返回的是具体子类实例，而且后续新增子类时不需要修改注解，完全符合工厂模式的扩展性。

方案2：用cast直接标记可实例化类型 #

如果你不想引入TypeVar，可以在选择子类后，用cast明确告诉mypy这个类是可实例化的AlgorithmBase子类：

from typing import cast, Type
import random

class AlgorithmFactory:
    _algorithm_types = AlgorithmBase.__subclasses__()
    
    def __init__(self):
        self._random_generator = random.Random()

    def select_random_algorithm(self) -> AlgorithmBase:
        algorithm_class = self._random_generator.choice(AlgorithmFactory._algorithm_types)
        # 用cast消除mypy对抽象类实例化的担忧
        concrete_class = cast(Type[AlgorithmBase], algorithm_class)
        return concrete_class()

这种方法更简洁，适合快速解决问题，核心是通过cast绕过mypy对抽象类实例化的检查（因为我们明确知道选中的都是具体子类）。

方案3：使用assert_type（Python 3.10+） #

如果你使用Python 3.10及以上版本，可以用typing.assert_type给mypy提供类型提示，这个操作在运行时不会执行，仅用于类型检查：

from typing import assert_type, Type
import random

class AlgorithmFactory:
    _algorithm_types = AlgorithmBase.__subclasses__()
    
    def __init__(self):
        self._random_generator = random.Random()

    def select_random_algorithm(self) -> AlgorithmBase:
        algorithm_class = self._random_generator.choice(AlgorithmFactory._algorithm_types)
        # 断言algorithm_class是可实例化的AlgorithmBase子类
        assert_type(algorithm_class, Type[AlgorithmBase])
        return algorithm_class()

为什么不用Union？ #

你提到想避免用Union指定具体子类，这非常合理——如果子类数量多或者后续会新增，Union会让代码变得臃肿且难以维护。上面的方案都不需要硬编码任何子类，完全依赖__subclasses__()的动态特性，同时满足mypy的类型检查要求。

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