diff options
Diffstat (limited to 'Lib/traceback.py')
| -rw-r--r-- | Lib/traceback.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/Lib/traceback.py b/Lib/traceback.py index d3c581f..4f0dff9 100644 --- a/Lib/traceback.py +++ b/Lib/traceback.py @@ -476,12 +476,11 @@ class StackSummary(list): gets called for every frame to be printed in the stack summary. """ row = [] - if frame_summary.filename.startswith("<python-input"): - row.append(' File "<stdin>", line {}, in {}\n'.format( - frame_summary.lineno, frame_summary.name)) - else: - row.append(' File "{}", line {}, in {}\n'.format( - frame_summary.filename, frame_summary.lineno, frame_summary.name)) + filename = frame_summary.filename + if frame_summary.filename.startswith("<stdin>-"): + filename = "<stdin>" + row.append(' File "{}", line {}, in {}\n'.format( + filename, frame_summary.lineno, frame_summary.name)) if frame_summary.line: stripped_line = frame_summary.line.strip() row.append(' {}\n'.format(stripped_line)) |