由于您的__str__
和__repr__
遵循相同的模式,您可以编写一个函数为你创建对象的字符串表示。这将需要一个对象,属性列表和str
或repr
作为参数:
def stringify(obj, attrs, strfunc):
values = []
# get each attribute's value and convert it to a string
for attr in attrs:
value = getattr(obj, attr)
values.append(strfunc(value))
# get the class name
clsname = type(obj).__name__
# put everything together
args = ', '.join(values)
return '{}({})'.format(clsname, args)
print(stringify(MyClass('foo'), ['a'], repr))
# output: MyClass('foo')
我建议把这个函数的一类,你再继承:
class Printable:
def __str__(self):
return self.__stringify(str)
def __repr__(self):
return self.__stringify(repr)
def __stringify(self, strfunc):
values = []
for attr in self._attributes:
value = getattr(self, attr)
values.append(strfunc(value))
clsname = type(self).__name__
args = ', '.join(values)
return '{}({})'.format(clsname, args)
class MyClass(Printable):
_attributes = ['a']
def __init__(self, a):
self.a = a
而且您甚至可以通过直接从__init__
函数的签名中获取属性完全自动完成:
import inspect
class AutoPrintable:
def __str__(self):
return self.__stringify(str)
def __repr__(self):
return self.__stringify(repr)
def __stringify(self, strfunc):
sig= inspect.signature(self.__init__)
values= []
for attr in sig.parameters:
value= getattr(self, attr)
values.append(strfunc(value))
clsname= type(self).__name__
args= ', '.join(values)
return '{}({})'.format(clsname, args)
class MyClass(AutoPrintable):
def __init__(self, a, b):
self.a = a
self.b = b
print(str(MyClass('foo', 'bar'))) # output: MyClass(foo, bar)
print(repr(MyClass('foo', 'bar'))) # output: MyClass('foo', 'bar')
'str'和'repr'是否真的需要返回两个不同的字符串? –
你可以说'__str__ = __repr__'(反之亦然)复制实现。 – jasonharper
他们没有必要返回不同的字符串,但通常更可取。也许这个例子很简单地表达这一点。 –