由於您的__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
他們沒有必要返回不同的字符串,但通常更可取。也許這個例子很簡單地表達這一點。 –