问题描述
我有一个代码库,我想使用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