summaryrefslogtreecommitdiffstats
path: root/Lib/pprint.py
diff options
context:
space:
mode:
authorSerhiy Storchaka <storchaka@gmail.com>2014-12-20 18:57:15 (GMT)
committerSerhiy Storchaka <storchaka@gmail.com>2014-12-20 18:57:15 (GMT)
commitfe3dc376fa73fce563be26417ea763b05115990b (patch)
tree286e2d24f64a15e2aecda42f8b43070a02a045af /Lib/pprint.py
parentf65d1d3b0241f79715df04192e21144672704fd7 (diff)
downloadcpython-fe3dc376fa73fce563be26417ea763b05115990b.zip
cpython-fe3dc376fa73fce563be26417ea763b05115990b.tar.gz
cpython-fe3dc376fa73fce563be26417ea763b05115990b.tar.bz2
Issue #19104: pprint now produces evaluable output for wrapped strings.
Diffstat (limited to 'Lib/pprint.py')
-rw-r--r--Lib/pprint.py58
1 files changed, 32 insertions, 26 deletions
diff --git a/Lib/pprint.py b/Lib/pprint.py
index 3be9c36..2cbffed 100644
--- a/Lib/pprint.py
+++ b/Lib/pprint.py
@@ -235,35 +235,41 @@ class PrettyPrinter:
return
if issubclass(typ, str) and len(object) > 0 and r is str.__repr__:
- def _str_parts(s):
- """
- Return a list of string literals comprising the repr()
- of the given string using literal concatenation.
- """
- lines = s.splitlines(True)
- for i, line in enumerate(lines):
- rep = repr(line)
- if len(rep) <= max_width:
- yield rep
- else:
- # A list of alternating (non-space, space) strings
- parts = re.split(r'(\s+)', line) + ['']
- current = ''
- for i in range(0, len(parts), 2):
- part = parts[i] + parts[i+1]
- candidate = current + part
- if len(repr(candidate)) > max_width:
- if current:
- yield repr(current)
- current = part
- else:
- current = candidate
- if current:
- yield repr(current)
- for i, rep in enumerate(_str_parts(object)):
+ chunks = []
+ lines = object.splitlines(True)
+ if level == 1:
+ indent += 1
+ max_width -= 2
+ for i, line in enumerate(lines):
+ rep = repr(line)
+ if len(rep) <= max_width:
+ chunks.append(rep)
+ else:
+ # A list of alternating (non-space, space) strings
+ parts = re.split(r'(\s+)', line) + ['']
+ current = ''
+ for i in range(0, len(parts), 2):
+ part = parts[i] + parts[i+1]
+ candidate = current + part
+ if len(repr(candidate)) > max_width:
+ if current:
+ chunks.append(repr(current))
+ current = part
+ else:
+ current = candidate
+ if current:
+ chunks.append(repr(current))
+ if len(chunks) == 1:
+ write(rep)
+ return
+ if level == 1:
+ write('(')
+ for i, rep in enumerate(chunks):
if i > 0:
write('\n' + ' '*indent)
write(rep)
+ if level == 1:
+ write(')')
return
write(rep)