mypy 不识别 Dictionary of Dictionary

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] 的结果实际上是一个 字符串 。问题是我的类型别名：

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

还是错误？

config 是从 JSON 文件加载的字典，其中唯一的可选参数是 “开始”。 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

如果您知道 config 中的 "start" 键将始终具有 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 不识别 Dictionary of Dictionary

mypy doesn't recognize Dictionary of Dictionary

python

dictionary

type-hinting

mypy

python-typing