微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

重构以满足mypy要求

如何解决重构以满足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

考虑

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。