Skip to content

Commit beed64f

Browse files
committed
fix: make timeit traceback can with color
Signed-off-by: yihong0618 <[email protected]>
1 parent 89b5571 commit beed64f

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

Lib/timeit.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,15 +150,16 @@ def print_exc(self, file=None):
150150
The optional file argument directs where the traceback is
151151
sent; it defaults to sys.stderr.
152152
"""
153-
import linecache, traceback
153+
import linecache, traceback, _colorize, sys
154154
if self.src is not None:
155155
linecache.cache[dummy_src_name] = (len(self.src),
156156
None,
157157
self.src.split("\n"),
158158
dummy_src_name)
159159
# else the source is already stored somewhere else
160160

161-
traceback.print_exc(file=file)
161+
traceback.print_exception(sys.exception(), file=file,
162+
colorize=_colorize.can_colorize(file=file))
162163

163164
def timeit(self, number=default_number):
164165
"""Time 'number' executions of the main statement.
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Make timeit error traceback can with corlor

0 commit comments

Comments
 (0)