以编程方式扩展数据类'__repr__

Question

假设我有一个带有 set 方法的数据类。如何扩展 repr 方法，以便在调用 set 方法时它也会更新：

from dataclasses import dataclass
@dataclass
class State:
    A: int = 1
    B: int = 2
    def set(self, var, val):
        setattr(self, var, val)

例如：

In [2]: x = State()

In [3]: x
Out[3]: State(A=1, B=2)

In [4]: x.set("C", 3)

In [5]: x
Out[5]: State(A=1, B=2)

In [6]: x.C
Out[6]: 3

我想要的结果

In [7]: x
Out[7]: State(A=1, B=2, C=3)

Answer 1

dataclass 装饰器可让您快速轻松地构建具有特定字段的类，这些字段在您定义类时已预先确定。但是，您打算使用类的方式与数据类的用途不太匹配。您希望能够在类已经存在之后动态添加新字段，并让它们使用各种方法（如__init__、__repr__ 和大概__eq__）。这几乎消除了使用dataclass 的所有好处。相反，您应该编写自己的类来执行您希望它执行的操作。

这是一个快速而肮脏的版本：

class State:
    _defaults = {"A": 1, "B": 2}
    
    def __init__(self, **kwargs):
        self.__dict__.update(self._defaults)
        self.__dict__.update(kwargs)
        
    def __eq__(self, other):
        return self.__dict__ == other.__dict__ # you might want to add some type checking here
        
    def __repr__(self):
        kws = [f"{key}={value!r}" for key, value in self.__dict__.items()]
        return "{}({})".format(type(self).__name__, ", ".join(kws))

这与您从 types.SimpleNamespace 获得的内容非常相似，因此您可能只能使用它（虽然它不使用默认值）。

您可以将您的 set 方法添加到此框架中，但在我看来，这似乎是对您已经用来实现它的内置 setattr 函数的不必要重复。如果调用者需要动态设置属性，可以自己调用setattr。如果属性名是常量，他们可以使用普通的属性赋值语法来代替s.foo = "bar"。