重构以满足mypy要求

问题描述

我有一个代码库,我想使用mypy进行验证。在当前设计中,一个类可能具有非原始成员,这可以在__init__之后进行设置,这是很常见的。

因此,在__init__中,该成员被初始化为None,其类型也相应地变为Optional

问题在于,现在MyPy要求我每次使用成员时都要检查该成员是否不是None

作为一种快速解决方案,我可以在所有相关范围内添加assert self._member is not None # MyPy,但这似乎是一个非常糟糕的做法。

一个想法是添加一个@property并在其中进行断言,但这似乎是一个巨大的开销。

是否有更自然/正确的设计可以克服此问题?

编辑:

为澄清起见,我的代码带有完整的类型注释。

成员定义为self._member: Optional[MemberType]

解决方法

如果在__init__之外分配了实例变量,则该类的实例可能处于未设置该变量的状态。老实说,在使用它的功能中应该考虑到这一点。但是,如果我们相信或确保在此状态下不会调用这些函数,则可以按以下方式注释此实例成员:

class Foo:
    var_1: int  # variant 1

    def __init__(self) -> None:
        self.var_2: int  # variant 2
       

这些语句不会创建实变量,仅annotations 实例变量会被mypy

考虑