python - 在数据类的自定义 `__init__` 覆盖中调用生成的 `__init__`

Question

目前我有这样的事情:

@dataclass(frozen=True)
class MyClass:
  a: str
  b: str
  c: str
  d: Dict[str, str]

...除了 dict，一切都很好s 是可变的，所以我不能用我的类来键入另一个字典。
相反，我想要字段 d成为 FrozenSet[Tuple[str, str]] 之类的东西，但我仍然希望有人构建我的类的实例，以便能够在构造函数上传递字典，因为这更直观。
所以我想做类似的事情

@dataclass(frozen=True)
class MyClass:
  a: str
  b: str
  c: str
  d: FrozenSet[Tuple[str, str]] = field(init=False)

  def __init__(self, a, b, c, d: Dict[str, str]):
    self.original_generated_init(a, b, c)  # ???
    object.setattr(self, 'd', frozenset(d.items()))  # required because my dataclass is frozen

我如何实现这一目标？或者有没有更优雅的方式来实现同样的事情？

Answer 1

您可以使用 InitVar 并分配给 d在 __post_init__ :

@dataclass(frozen=True)
class MyClass:
  a: str
  b: str
  c: str
  d: FrozenSet[Tuple[str, str]] = field(init=False)
  d_init: InitVar[Dict[str, str]]

  def __post_init__(self, d_init):
    object.__setattr__(self, 'd', frozenset(d_init.items()))