如何为 TypedDicts 列表中的每个字典添加一个键

Question

问题

(Python 3.9) 我继承了一些TypedDict ，如下：

from typing import TypedDict


class Base(TypedDict):
    key: str


class Child(Base):
   extended: str

我有一个Base指令列表，我想将其扩展为每个指令都具有相同值的Child指令。

base_list: list[Base] = [{'key': 'value'}, {'key': 'value2'}]

最后我想要以下内容：

child_list: list[Child] = [
  {'key': 'value', 'extended': 'sameval'}, 
  {'key': 'value2', 'extended': 'sameval'}
]

我正在使用 pylance/pyright。

关于如何优雅地/无需cast就可以完成此任务的任何建议？ 请注意，这是一个特定的打字问题，我可以扩展字典。

我试过的

列表理解

此响应表明：

child_list: list[Child] = [dict(base,extended='ex') for base in base_list]

但是，pylance 抱怨道：

Expression of type "list[dict[str, object]]" cannot be assigned to declared type "list[Child]"

我可以用cast解决这个问题，但感觉很脏：

a_ex: list[Child] = [cast(Child, dict(base,extended='ex')) for base in a]

NotRequired

我认为我可以通过将extended标记为NotRequired来绕过它，然后遍历列表以设置密钥。 任何Base object 都应该是一个有效的Child object，对吧？ 没有。

a_ex: list[Child] = a

Expression of type "list[Base]" cannot be assigned to declared type "list[Child]"
  "list[Base]" is incompatible with "list[Child]"
    TypeVar "_T@list" is invariant
      "extended" is missing from "Type[Base]"

我也玩过dict.update ，但遇到了与列表理解类似的问题。

Answer 1

为什么不明确地初始化每个Child ？

from pprint import pprint
from typing import TypedDict


class Base(TypedDict):
    key: str


class Child(Base):
    extended: str


base_list: list[Base] = [{"key": "value"}, {"key": "value2"}]

child_list: list[Child]

child_list = [Child(key=d["key"], extended="sameval") for d in base_list]

pprint(child_list)

Output：

[{'extended': 'sameval', 'key': 'value'},
 {'extended': 'sameval', 'key': 'value2'}]

这通过mypy --strict没有问题，如果 Pyright 对此有问题，我会感到惊讶。

需要说明的是，这两个也有效：（基于相同的原理）

child_list = [dict(key=d["key"], extended="sameval") for d in base_list]

child_list = [{"key": d["key"], "extended": "sameval"} for d in base_list]

---

您的第一次尝试不起作用，因为您不再处理字典文字或明确命名为 arguments ，而是在for循环中使用变量 -- base 。 我假设无法保证类型安全，因为字典是可变的，但如果可能的话，其他人可以在这里解决细节问题。

在任何情况下，为了以尽可能少的示例来证明该问题，请考虑以下内容：

from typing import TypedDict


class Base(TypedDict):
    key: str


b1: Base = {"key": "value"}  # safe
b2: Base = dict(key="value")  # safe
b3: Base = dict(b1)  # not safe

通过mypy运行它只会在最后一行出现以下错误：

error: Incompatible types in assignment (expression has type "Dict[str, object]", variable has type "Base") [assignment]

这类似于您在第一个列表理解示例中从 Pyright 收到的错误。 类型检查器只能推断出字典初始化的结果是dict[str, object] 。

---

您第二次使用NotRequired的尝试也不应该起作用。 Base与Child不一致。 有关详细信息，请参阅此帖子。