Question

我有以下测试程序：

    from random import choice
    d = { }
    def data(length):
        alphabet = 'abcdefghijklmnopqrstuvwxyz'
        res = ''
        for _ in xrange(length):
            res += choice(alphabet)
        return res
    # Create the test data
    for cnt in xrange(10):
        key = 'key-%d' % (cnt)
        d[key] = data(30)
    def pprint_shorted(d, max_length):
        import pprint
        pp = pprint.PrettyPrinter(indent=4)
        pp.pprint(d)
    pprint_shorted(d, 10)

目前输出类似于：

{   'key-0': 'brnneqgetvanmggyayppxevwcnxvue',
    'key-1': 'qjzrklrdkykililenwcyhaexuylgub',
    'key-2': 'ayddiaxhvgxpszutnjdwlgojqaluhr',
    'key-3': 'rmjpzxrmbogezorigkycqhpsctinzq',
    'key-4': 'botfczymszkzwuiecyarknnrvwavnr',
    'key-5': 'norifblhtvfnwblcyeipjmteznylfy',
    'key-6': 'tiiubgdwxnogdmbafvnujbwpfdopjl',
    'key-7': 'badgwbrrqunivylutbxqkaeuctrykt',
    'key-8': 'wulrfkqfqqecxmscayzdbatyispwtu',
    'key-9': 'gzlwfvjrevlyvbmrvuisnyhhbbwtdd'}

在我的生产数据中，有时字符串确实长（数千个字符，例如来自base64编码的附件），我不希望填充我的日志。我想要像：

{   'key-0': 'brnneqgetv...',
    'key-1': 'qjzrklrdky...',
    'key-2': 'ayddiaxhvg...',
    'key-3': 'rmjpzxrmbo...',
    'key-4': 'botfczymsz...',
    'key-5': 'norifblhtv...',
    'key-6': 'tiiubgdwxn...',
    'key-7': 'badgwbrrqu...',
    'key-8': 'wulrfkqfqq...',
    'key-9': 'gzlwfvjrev...'}

即，dict中的字符串值，长度为＆gt;必须用省略号替换max_length。在pretty print中是否有任何内置支持，或者我是否必须通过手动移动并缩短字符串来创建dict的副本？

Answer 1

您可以继承PrettyPrinter并覆盖方法_format：

import pprint

class P(pprint.PrettyPrinter):
  def _format(self, object, *args, **kwargs):
    if isinstance(object, basestring):
      if len(object) > 20:
        object = object[:20] + '...'
    return pprint.PrettyPrinter._format(self, object, *args, **kwargs)

P().pprint(range(3))
P().pprint('x' * 1000)

打印：

[0, 1, 2]
'xxxxxxxxxxxxxxxxxxxx...'

使用漂亮的打印进行转储时自动缩短长字符串

1 个答案: