假设我有一些类似的代码
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成为生产依赖关系.