包括格式化的iterable作为更大格式化字符串的一部分

时间:2011-08-16 02:17:18

标签: python python-3.x

最近在写一堂课时,我最初在以下几行中加入了__repr__方法:

return "{}({!r}, {!r}, {!r})".format(
                self.__class__.__name__,
                self.arg1,
                self.arg2,
                self.arg3)

重复“{!r}”代码片段感觉不对,如果我再向该类添加更多参数,则会很难维护。然而,对我而言,更加强大的替代方案也不会赢得任何优雅奖品。

以编程方式构建格式字符串:

fmt = "{}(%s)" % ", ".join(["{!r}"]*3)
return fmt.format(self.__class__.__name__,
                  self.arg1,
                  self.arg2,
                  self.arg3)

使用str.join单独格式化参数:

args = ", ".join(map(repr, [self.arg1, self.arg2, self.arg3]))
return "{}({})".format(self.__class__.__name__, args)

我目前使用最后一个示例实现了该类,但我对替代方法的建议感兴趣(因为我对上述任何选项都不是特别满意)。

更新 受Ned的回答启发,我现在已将以下实用程序功能添加到辅助模块中:

def format_iter(iterable, fmt='{!r}', sep=', '):
    return sep.join(fmt.format(x) for x in iterable)

>>> format_iter(range(10))
'0, 1, 2, 3, 4, 5, 6, 7, 8, 9'
>>> format_iter(range(10), sep='|')
'0|1|2|3|4|5|6|7|8|9'
>>> format_iter(range(10), fmt='{:04b}', sep='|')
'0000|0001|0010|0011|0100|0101|0110|0111|1000|1001'
>>> format_iter(range(10), fmt='{0.real}+{0.imag}j')
'0+0j, 1+0j, 2+0j, 3+0j, 4+0j, 5+0j, 6+0j, 7+0j, 8+0j, 9+0j'

UPDATE2: 我最后添加了第二个实用函数,几乎与agf的答案相同:

def call_repr(name, *args):
    return "{}({})".format(name, format_iter(args))

所以最初违规的__repr__函数现在看起来像:

def __repr__(self):
    return call_repr(self.__class__.__name__,
                     self.arg1,
                     self.arg2)

(是的,今天早些时候,其中一个原始的构造函数参数已经消失了。)

3 个答案:

答案 0 :(得分:8)

如果你不喜欢这些代码,我倾向于将其隐藏在一个函数中:

def repr_all(*args):
    return ", ".join(repr(a) for a in args)


def __repr__(self):
    args = repr_all(self.arg1, self.arg2, self.arg3)
    return "{}({})".format(self.__class__.__name__, args)

答案 1 :(得分:8)

如果这是一个你要重复的模式,我可能会使用:

# This is a static method or module-level function
def argrepr(name, *args):
    return '{}({})'.format(name, ', '.join(repr(arg) for arg in args))

def __repr__(self):
    return argrepr(self.__name__, self.arg1, self.arg2, self.arg3)

# This is a static method or module-level function
def argrepr(*args):
    return '(' + ', '.join(repr(arg) for arg in args) + ')'

def __repr__(self):
    return repr(self.__name__) + argrepr(self.arg1, self.arg2, self.arg3)

答案 2 :(得分:0)

我可以这样做:

def __repr__(self):
    return "{0}({1})".format(
        type(self).__name__,
        ", ".join(repr(arg) for arg in (self.arg1, self.arg2, self.arg3)))

或:

_init_getter = operator.attrgetter('arg1', 'arg2', 'arg3')
def __repr__(self):
    return "{0}({1})".format(
        type(self).__name__,
        ", ".join(repr(arg) for arg in self._init_getter(self)))