Dataclasses和TypedDicts之间的类型检查转换

我有一堆@dataclasses 和一堆对应的TypedDicts,我想促进它们之间的平滑和类型检查转换。

例如,考虑

from dataclasses import dataclass
from typing_extensions import TypedDict

@dataclass
class Foo:
    bar: int
    baz: int
   
    @property
    def qux(self) -> int:
        return bar + baz

class SerializedFoo(TypedDict):
    bar: int
    baz: int
    qux: int

要创建序列化函数,我可以编写类似

def serialize(foo: Foo) -> SerializedFoo:
    return SerializedFoo(
        bar=foo.bar,
        baz=foo.baz,
        qux=foo.qux,
    )

但是对于很多类型这样做变得乏味,而且每次更新类型时,我还必须更新序列化功能。

我也可以做类似的事情

def serialize(foo: Foo) -> SerializedFoo:
    return SerializedFoo(**dataclasses.asdict(foo))

但这不会进行类型检查;mypy抱怨它Expected keyword arguments, {...}, or dict(...) in TypedDict constructor

从理论上讲,足够聪明的类型检查器应该有可能知道数据类具有初始化类型字典所需的属性,但是当然,使用asdict使得这在实践中是不可能的。

有没有更好的方法将 a 转换dataclassTypedDict具有相应字段的 a ,这让我都可以让类型检查器在出现问题时告诉我,而不必在转换中输入每个字段?

以上是Dataclasses和TypedDicts之间的类型检查转换的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>