在python字典中为mypy指定键

发布时间：2020-12-17 17:36:42 所属栏目：Python 来源：网络整理

导读：假设我有一些类似的代码 def get_x(d: dict) - int: d["x"] 但是,我想告诉mypy d仅应包含某些键(例如,仅包含“ x”键).这样,如果我在尝试引用无效键d的代码中犯了一个错误,那么mypy将触发错误. 我的问题是：这可能吗？ 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成为生产依赖关系.

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!