python - 有没有办法在 python 中自动生成 __str__() 实现？

Question

厌倦了为我的类手动实现字符串表示，我想知道是否有一种 Python 的方法可以自动执行此操作。

我想要一个涵盖类的所有属性和类名的输出。这是一个例子:

class Foo(object):
    attribute_1 = None
    attribute_2 = None
    def __init__(self, value_1, value_2):
         self.attribute_1 = value_1
         self.attribute_2 = value_2

导致:

bar = Foo("baz", "ping")
print(str(bar)) # desired: Foo(attribute_1=baz, attribute_2=ping)

在使用 Project Lombok @ToString 后想到了这个问题在一些 Java 项目中。

Answer 1

您可以使用 vars 迭代实例属性, dir , ...:

def auto_str(cls):
    def __str__(self):
        return '%s(%s)' % (
            type(self).__name__,
            ', '.join('%s=%s' % item for item in vars(self).items())
        )
    cls.__str__ = __str__
    return cls

@auto_str
class Foo(object):
    def __init__(self, value_1, value_2):
        self.attribute_1 = value_1
        self.attribute_2 = value_2

应用:

>>> str(Foo('bar', 'ping'))
'Foo(attribute_2=ping, attribute_1=bar)'