如何向工厂方法添加提示？

Question

我正在寻找一种方法来注释工厂函数的返回类型。

它返回“AlgorithmBase”的随机子节点。

class AlgorithmFactory:
    _algorithm_types = AlgorithmBase.__subclasses__()

    def select_random_algorithm(self) -> AlgorithmBase:
        # Select random algorithm
        algorithm_class = self._random_generator.choice(AlgorithmFactory._algorithm_types)
        algorithm = algorithm_class()
        return algorithm

我从 mypy 收到错误：

我得到的错误是：

Cannot instantiate abstract class 'AlgorithmBase' with abstract attributes 'get_constraints' and 'satisfy_constraints'

这段代码中没有办法实例化'AlgorithmBase'类，如何让mypy理解？

我想避免在返回类型中指定带有“Union”的实际子类。有什么建议吗？

Answer 1

这里的问题不是返回类型，而是“_algorithm_types”。 mypy 无法理解它是什么类型，所以它假设它像返回类型并得到错误。

以下代码解决了这个问题：

_algorithm_types: List[Type[AlgorithmBase]] = AlgorithmBase.__subclasses__()

Answer 2

据我所知，这应该可行，但您的一个或多个 AlgorithmBase 子类似乎没有实现这两个抽象方法。

为

运行 MyPy

import abc

class AlgorithmBase(abc.ABC):
    @abc.abstractmethod
    def get_constraints(self):
        raise NotImplementedError

    @abc.abstractmethod
    def satisfy_constraints(self):
        raise NotImplementedError


class SomeAlgorithm(AlgorithmBase):
    pass


class AlgorithmFactory:
    def get(self) -> AlgorithmBase:
        algorithm = SomeAlgorithm()
        return algorithm

产生与您得到的相同的错误，并且一旦实现方法，它就会运行而不会出现任何错误。

import abc

class AlgorithmBase(abc.ABC):
    @abc.abstractmethod
    def get_constraints(self):
        raise NotImplementedError

    @abc.abstractmethod
    def satisfy_constraints(self):
        raise NotImplementedError


class SomeAlgorithm(AlgorithmBase):
    def get_constraints(self):
        pass

    def satisfy_constraints(self):
        pass


class AlgorithmFactory:
    def get(self) -> AlgorithmBase:
        algorithm = SomeAlgorithm()
        return algorithm