mypy 无法识别字典中的字典

Question

我正在尝试使用 mypy 检查我的类型注释，但此错误不断发生：

Script.py:201: error: Item "Dict[str, Union[float, int]]" of "Union[Dict[str, Union[float, int]], str, float, int, bool]" has no 
attribute "endswith"

我的代码如下所示：

from typing import Counter, Dict, Iterable, List, NoReturn, Optional, Set, Tuple, Union

TYPE_NUMBER         = Union[float, int]
TYPE_CONFIGURATION  = Dict[str, Union[Dict[str, TYPE_NUMBER], str, float, int, bool]]

def check_configuration(config: TYPE_CONFIGURATION) -> Union[bool, NoReturn]:
    database = 'database'
    assert isinstance(config[database], str)
    assert config[database].endswith('.prdb')

正常调用 python 运行得很好。所以我知道

config[database1]

的结果实际上是一个 string。问题是我的类型别名吗：

TYPE_CONFIGURATION  = Dict[str, Union[Dict[str, TYPE_NUMBER], str, float, int, bool]]

还是一个错误？

config是从JSON文件加载的字典，其中唯一的可选参数是“start”。 JSON 文件如下所示：

{
    "database" : "bla/bla/bla/file.csv",
    "distance" : 800,
    "t"        : false,
    "output"   : "bla/bla/bla/file-out.csv",

    "start"    : {"1": 1343.786, "2": 1356.523}
}

Answer 1

如果您知道

"start"

中的

config

键始终具有

Dict[str, Union[int, float]]

类型的值，那么一种解决方案可能是在类型注释中使用

TypedDict

。

TypedDict

允许您指定与每个键关联的值的预期类型。在下面的示例中，我指定了

total=False

，让 mypy 知道字典仍然可以被视为

ConfigDict

，即使它不包含 all 在类型定义中指定的字段（您提到过

"start"

是可选参数）。

from typing import Dict, Union, TypedDict, NoReturn
    
TYPE_NUMBER = Union[float, int]
    
class ConfigDict(TypedDict, total=False):
    database: str
    distance: int
    t: bool
    start: Dict[str, TYPE_NUMBER]
    
def check_configuration(config: ConfigDict) -> Union[bool, NoReturn]:
    assert isinstance(config['database'], str)
    assert config['database'].endswith('.prdb')
    return True

TypedDict

的子类实际上并不是类型，它们只是附加了一些额外信息的字典，以便于类型检查器。所以你可以像这样实例化它们：

config = ConfigDict(
    database="bla/bla/bla/file.csv",
    distance=800,
    t=False,
    output="bla/bla/bla/file-out.csv",
    start={"1": 1343.786, "2": 1356.523}
)

或者像这样：

config: ConfigDict = {
    "database" : "bla/bla/bla/file.csv",
    "distance" : 800,
    "t"        : False,
    "output"   : "bla/bla/bla/file-out.csv",

    "start"    : {"1": 1343.786, "2": 1356.523}
}

—这两种语法产生的结果是相同的，对于它们两者，

type(config)

将是

dict

而不是

ConfigDict

或

TypedDict

（

isinstance

确实不工作）。

我同意@MarcelWilson的评论，即Mypy error on dict of dict: Value of type "object" is not indexable与您的问题原因相关。