如何强制施法键入命令以避免分配中的不兼容类型

Question

我正在尝试在高效率函数（试图避免复制）中执行以下（简化的示例），而无需使用Naborequired（出于此后显示的原因）：

Mytd1 = TypedDict("Mytd1", {a: NotRequired[str], x: int})
Mytd_extra = TypedDict("Mytd_extra", {a: str, x: int, z: str})


def foo(y: Mytd1) -> Mytd_extra:
    # users input object should NOT contain z
    y = cast(Mytd_extra, y). # fails - my guess is `a` is the offender
    y["z"] = "hooray"
    if "a" not in y:
        y["a"] = "whatever"
    return y
 

Mypy抱怨演员阵容为“不兼容类型作业”，但这是我想做的。我该如何实现？我认为a是罪犯 - 在输出类型

i Do 不是 WANT上需要：

# kill Mytd1 
Mytd_extra = TypedDict("Mytd_extra", {x: int, z: NotRequired[str]})

因为这将允许客户端传递{x：5，z ：“ asdf”} foo ，但应禁止 - 它们只允许通过{x：5}传递。

我认为以下内容是可行的，但是我试图避免它以提高效率，因为此功能被称为多次：

def foo(y: Mytd1) -> Mytd_extra:
    new_var: Mytd_extra = {k: v for k:v in y.items()}
    new_var["z"] = "ohno"
    return new_var

Answer 1

您至少有两个完全有效的解决方案（假设Python 3.11不要打扰键入 vs typing_extensions，请在需要时调整导入）。问题是y是类型mytd1，因此您无法分配mytd_extra（cast的结果）。

在这里使用另一个变量

from typing import TypedDict, NotRequired, cast

Mytd1 = TypedDict("Mytd1", {'a': NotRequired[str], 'x': int})
Mytd_extra = TypedDict("Mytd_extra", {'a': str, 'x': int, 'z': str})


def foo(orig_y: Mytd1) -> Mytd_extra:
    # users input object should NOT contain z
    y = cast(Mytd_extra, orig_y)
    y["z"] = "hooray"
    if "a" not in y:
        y["a"] = "whatever"
    return y

是

允许重新定义（mypy flag）

保留您的原始代码，但使用- allow-redefinition（或添加ally_redefinition = true true config config config config config您使用）：

警告： allow_redefinition没有超级牛的功率，用例仅限于单个方案：当rhs中的表达式包含此变量时，将其分配到预定变量。它允许以下内容：

y = 'abc'
y = list(y)

但不是这样：

y = 'abc'
y = 1