在python字典中为mypy指定键

假设我有一些类似的代码

def get_x(d: dict) -> int:
    d["x"]

但是,我想告诉mypy d仅应包含某些键(例如,仅包含“ x”键).这样,如果我在尝试引用无效键d的代码中犯了一个错误,那么mypy将触发错误.

我的问题是：

>这可能吗？ Mypy可以验证字典密钥吗？
>如果是,该如何完成？如果否,是否有首选的解决方法？

最佳答案

尝试在AWS API Gateway / Lambda集成中进行一些输入时,也会遇到此问题.

正如@ Michael0x2a在评论中指出的,TypedDict似乎是一种解决方法,特别是因为它在传递给函数之前不需要自己进行任何转换(就像您提到的那样,需要API响应).

from mypy_extensions import TypedDict

DictWithOnlyX = TypedDict('DictWithOnlyX',{"x": str})

# error: Extra key 'y' for TypedDict "DictWithOnlyX@58"
dx1: DictWithOnlyX = { "y": 123 }

# error: Incompatible types (expression has type "int",#        TypedDict item "x" has type "str")
dx2: DictWithOnlyX = { "x": 123 }

# GOOD!
dx3: DictWithOnlyX = { "x": "123" }

还应注意,如果在生产和开发/测试之间隔离依赖关系,则这会使mypy成为生产依赖关系.

原文地址：https://www.jb51.cc/python/533096.html