时间戳

首页 > 解决方案 > mypy 无法识别 Dictionary of Dictionary

python - mypy 无法识别 Dictionary of Dictionary

问题描述

我正在尝试使用 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]实际上是一个字符串。问题是我的类型别名：

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

或者它是一个错误？

配置是从 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}
}

标签： pythondictionarytype-hintingmypypython-typing

解决方案

如果您知道"start"key inconfig将始终具有 type 的值Dict[str, Union[int, float]]，那么一种解决方案可能是TypedDict在您的类型注释中使用 a 。ATypedDict允许您指定与每个键关联的值的预期类型。在下面的示例中，我指定total=False, 让 mypy 知道即使字典ConfigDict不包含类型定义中指定的所有"start"字段（您提到这是一个可选参数），它仍然可以被视为 a。

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 的评论，即dict 上的 Mypy 错误：“object”类型的值不可索引与您的问题的原因有关。

推荐阅读