具有 None 类型或 dict 的类成员的 Python 类型提示

发布于03月14日

我有None或dict的班级成员.

class A:
    def __init__(self, a: bool = False) -> None:
        self.has_a = a
        self.prop_a = {"a" : 1} if a else None

    def length_prop(self) -> int:
        length_prop = None
        if self.has_a:
            length_prop = len(self.prop_a)
        return length_prop

如果我用mypy判断打字:

test.py:10: error: Argument 1 to "len" has incompatible type "Optional[Dict[str, int]]"; expected "Sized"
                length_prop = len(self.prop_a)
                                  ^
test.py:11: error: Incompatible return value type (got "Optional[int]", expected "int")
            return length_prop

很容易理解，因为它不知道self.prop_a是否真的是dict.

推荐答案

有两个问题:

100 can't and shouldn't infer the relationship between 101 and 102 across the functions.您需要在length_prop函数中包含None的显式判断.这个错误不是虚假的——例如，如果一些外部代码修改了has_a而不是prop_a的值，会发生什么？

最简单的解决方案是go 掉has_a字段，因为它编码的所有信息都可以通过判断prop_a(例如prop_a is not None)来检索.
The return value of 100 can be 101，所以在length_prop方法中使用Optional[int]而不是int.

下面的代码片段解决了这两个问题.mypy未在此类定义中找到任何错误:

from typing import Optional

class A:
    def __init__(self, a: bool = False) -> None:
        self.prop_a = {"a" : 1} if a else None

    def length_prop(self) -> Optional[int]:
        length_prop = None
        if self.prop_a is not None:
            length_prop = len(self.prop_a)
        return length_prop