##// END OF EJS Templates
Linting
martinRenou -
Show More
@@ -1,1122 +1,1124 b''
1 1 # -*- coding: utf-8 -*-
2 2 """
3 3 Verbose and colourful traceback formatting.
4 4
5 5 **ColorTB**
6 6
7 7 I've always found it a bit hard to visually parse tracebacks in Python. The
8 8 ColorTB class is a solution to that problem. It colors the different parts of a
9 9 traceback in a manner similar to what you would expect from a syntax-highlighting
10 10 text editor.
11 11
12 12 Installation instructions for ColorTB::
13 13
14 14 import sys,ultratb
15 15 sys.excepthook = ultratb.ColorTB()
16 16
17 17 **VerboseTB**
18 18
19 19 I've also included a port of Ka-Ping Yee's "cgitb.py" that produces all kinds
20 20 of useful info when a traceback occurs. Ping originally had it spit out HTML
21 21 and intended it for CGI programmers, but why should they have all the fun? I
22 22 altered it to spit out colored text to the terminal. It's a bit overwhelming,
23 23 but kind of neat, and maybe useful for long-running programs that you believe
24 24 are bug-free. If a crash *does* occur in that type of program you want details.
25 25 Give it a shot--you'll love it or you'll hate it.
26 26
27 27 .. note::
28 28
29 29 The Verbose mode prints the variables currently visible where the exception
30 30 happened (shortening their strings if too long). This can potentially be
31 31 very slow, if you happen to have a huge data structure whose string
32 32 representation is complex to compute. Your computer may appear to freeze for
33 33 a while with cpu usage at 100%. If this occurs, you can cancel the traceback
34 34 with Ctrl-C (maybe hitting it more than once).
35 35
36 36 If you encounter this kind of situation often, you may want to use the
37 37 Verbose_novars mode instead of the regular Verbose, which avoids formatting
38 38 variables (but otherwise includes the information and context given by
39 39 Verbose).
40 40
41 41 .. note::
42 42
43 43 The verbose mode print all variables in the stack, which means it can
44 44 potentially leak sensitive information like access keys, or unencrypted
45 45 password.
46 46
47 47 Installation instructions for VerboseTB::
48 48
49 49 import sys,ultratb
50 50 sys.excepthook = ultratb.VerboseTB()
51 51
52 52 Note: Much of the code in this module was lifted verbatim from the standard
53 53 library module 'traceback.py' and Ka-Ping Yee's 'cgitb.py'.
54 54
55 55 Color schemes
56 56 -------------
57 57
58 58 The colors are defined in the class TBTools through the use of the
59 59 ColorSchemeTable class. Currently the following exist:
60 60
61 61 - NoColor: allows all of this module to be used in any terminal (the color
62 62 escapes are just dummy blank strings).
63 63
64 64 - Linux: is meant to look good in a terminal like the Linux console (black
65 65 or very dark background).
66 66
67 67 - LightBG: similar to Linux but swaps dark/light colors to be more readable
68 68 in light background terminals.
69 69
70 70 - Neutral: a neutral color scheme that should be readable on both light and
71 71 dark background
72 72
73 73 You can implement other color schemes easily, the syntax is fairly
74 74 self-explanatory. Please send back new schemes you develop to the author for
75 75 possible inclusion in future releases.
76 76
77 77 Inheritance diagram:
78 78
79 79 .. inheritance-diagram:: IPython.core.ultratb
80 80 :parts: 3
81 81 """
82 82
83 83 #*****************************************************************************
84 84 # Copyright (C) 2001 Nathaniel Gray <n8gray@caltech.edu>
85 85 # Copyright (C) 2001-2004 Fernando Perez <fperez@colorado.edu>
86 86 #
87 87 # Distributed under the terms of the BSD License. The full license is in
88 88 # the file COPYING, distributed as part of this software.
89 89 #*****************************************************************************
90 90
91 91
92 92 import inspect
93 93 import linecache
94 94 import pydoc
95 95 import sys
96 96 import time
97 97 import traceback
98 98
99 99 import stack_data
100 100 from pygments.formatters.terminal256 import Terminal256Formatter
101 101 from pygments.styles import get_style_by_name
102 102
103 103 # IPython's own modules
104 104 from IPython import get_ipython
105 105 from IPython.core import debugger
106 106 from IPython.core.display_trap import DisplayTrap
107 107 from IPython.core.excolors import exception_colors
108 108 from IPython.utils import path as util_path
109 109 from IPython.utils import py3compat
110 110 from IPython.utils.terminal import get_terminal_size
111 111
112 112 import IPython.utils.colorable as colorable
113 113
114 114 # Globals
115 115 # amount of space to put line numbers before verbose tracebacks
116 116 INDENT_SIZE = 8
117 117
118 118 # Default color scheme. This is used, for example, by the traceback
119 119 # formatter. When running in an actual IPython instance, the user's rc.colors
120 120 # value is used, but having a module global makes this functionality available
121 121 # to users of ultratb who are NOT running inside ipython.
122 122 DEFAULT_SCHEME = 'NoColor'
123 123
124 124 # ---------------------------------------------------------------------------
125 125 # Code begins
126 126
127 127 # Helper function -- largely belongs to VerboseTB, but we need the same
128 128 # functionality to produce a pseudo verbose TB for SyntaxErrors, so that they
129 129 # can be recognized properly by ipython.el's py-traceback-line-re
130 130 # (SyntaxErrors have to be treated specially because they have no traceback)
131 131
132 132
133 133 def _format_traceback_lines(lines, Colors, has_colors, lvals):
134 134 """
135 135 Format tracebacks lines with pointing arrow, leading numbers...
136 136
137 137 Parameters
138 138 ----------
139 139 lines : list[Line]
140 140 Colors
141 141 ColorScheme used.
142 142 lvals : str
143 143 Values of local variables, already colored, to inject just after the error line.
144 144 """
145 145 numbers_width = INDENT_SIZE - 1
146 146 res = []
147 147
148 148 for stack_line in lines:
149 149 if stack_line is stack_data.LINE_GAP:
150 150 res.append('%s (...)%s\n' % (Colors.linenoEm, Colors.Normal))
151 151 continue
152 152
153 153 line = stack_line.render(pygmented=has_colors).rstrip('\n') + '\n'
154 154 lineno = stack_line.lineno
155 155 if stack_line.is_current:
156 156 # This is the line with the error
157 157 pad = numbers_width - len(str(lineno))
158 158 num = '%s%s' % (debugger.make_arrow(pad), str(lineno))
159 159 start_color = Colors.linenoEm
160 160 else:
161 161 num = '%*s' % (numbers_width, lineno)
162 162 start_color = Colors.lineno
163 163
164 164 line = '%s%s%s %s' % (start_color, num, Colors.Normal, line)
165 165
166 166 res.append(line)
167 167 if lvals and stack_line.is_current:
168 168 res.append(lvals + '\n')
169 169 return res
170 170
171 171
172 172 def _format_filename(file, ColorFilename, ColorNormal):
173 173 """
174 174 Format filename lines with `In [n]` if it's the nth code cell or `File *.py` if it's a module.
175 175
176 176 Parameters
177 177 ----------
178 178 file : str
179 179 ColorFilename
180 180 ColorScheme's filename coloring to be used.
181 181 ColorNormal
182 182 ColorScheme's normal coloring to be used.
183 183 """
184 184 ipinst = get_ipython()
185 185
186 186 if ipinst is not None and file in ipinst.compile._filename_map:
187 187 file = "[%s]" % ipinst.compile._filename_map[file]
188 188 tpl_link = "In %s%%s%s" % (ColorFilename, ColorNormal)
189 189 else:
190 190 file = util_path.compress_user(
191 191 py3compat.cast_unicode(file, util_path.fs_encoding)
192 192 )
193 193 tpl_link = "File %s%%s%s" % (ColorFilename, ColorNormal)
194 194
195 195 return tpl_link % file
196 196
197 197 #---------------------------------------------------------------------------
198 198 # Module classes
199 199 class TBTools(colorable.Colorable):
200 200 """Basic tools used by all traceback printer classes."""
201 201
202 202 # Number of frames to skip when reporting tracebacks
203 203 tb_offset = 0
204 204
205 205 def __init__(self, color_scheme='NoColor', call_pdb=False, ostream=None, parent=None, config=None):
206 206 # Whether to call the interactive pdb debugger after printing
207 207 # tracebacks or not
208 208 super(TBTools, self).__init__(parent=parent, config=config)
209 209 self.call_pdb = call_pdb
210 210
211 211 # Output stream to write to. Note that we store the original value in
212 212 # a private attribute and then make the public ostream a property, so
213 213 # that we can delay accessing sys.stdout until runtime. The way
214 214 # things are written now, the sys.stdout object is dynamically managed
215 215 # so a reference to it should NEVER be stored statically. This
216 216 # property approach confines this detail to a single location, and all
217 217 # subclasses can simply access self.ostream for writing.
218 218 self._ostream = ostream
219 219
220 220 # Create color table
221 221 self.color_scheme_table = exception_colors()
222 222
223 223 self.set_colors(color_scheme)
224 224 self.old_scheme = color_scheme # save initial value for toggles
225 225
226 226 if call_pdb:
227 227 self.pdb = debugger.Pdb()
228 228 else:
229 229 self.pdb = None
230 230
231 231 def _get_ostream(self):
232 232 """Output stream that exceptions are written to.
233 233
234 234 Valid values are:
235 235
236 236 - None: the default, which means that IPython will dynamically resolve
237 237 to sys.stdout. This ensures compatibility with most tools, including
238 238 Windows (where plain stdout doesn't recognize ANSI escapes).
239 239
240 240 - Any object with 'write' and 'flush' attributes.
241 241 """
242 242 return sys.stdout if self._ostream is None else self._ostream
243 243
244 244 def _set_ostream(self, val):
245 245 assert val is None or (hasattr(val, 'write') and hasattr(val, 'flush'))
246 246 self._ostream = val
247 247
248 248 ostream = property(_get_ostream, _set_ostream)
249 249
250 250 def get_parts_of_chained_exception(self, evalue):
251 251 def get_chained_exception(exception_value):
252 252 cause = getattr(exception_value, '__cause__', None)
253 253 if cause:
254 254 return cause
255 255 if getattr(exception_value, '__suppress_context__', False):
256 256 return None
257 257 return getattr(exception_value, '__context__', None)
258 258
259 259 chained_evalue = get_chained_exception(evalue)
260 260
261 261 if chained_evalue:
262 262 return chained_evalue.__class__, chained_evalue, chained_evalue.__traceback__
263 263
264 264 def prepare_chained_exception_message(self, cause):
265 265 direct_cause = "\nThe above exception was the direct cause of the following exception:\n"
266 266 exception_during_handling = "\nDuring handling of the above exception, another exception occurred:\n"
267 267
268 268 if cause:
269 269 message = [[direct_cause]]
270 270 else:
271 271 message = [[exception_during_handling]]
272 272 return message
273 273
274 274 @property
275 275 def has_colors(self):
276 276 return self.color_scheme_table.active_scheme_name.lower() != "nocolor"
277 277
278 278 def set_colors(self, *args, **kw):
279 279 """Shorthand access to the color table scheme selector method."""
280 280
281 281 # Set own color table
282 282 self.color_scheme_table.set_active_scheme(*args, **kw)
283 283 # for convenience, set Colors to the active scheme
284 284 self.Colors = self.color_scheme_table.active_colors
285 285 # Also set colors of debugger
286 286 if hasattr(self, 'pdb') and self.pdb is not None:
287 287 self.pdb.set_colors(*args, **kw)
288 288
289 289 def color_toggle(self):
290 290 """Toggle between the currently active color scheme and NoColor."""
291 291
292 292 if self.color_scheme_table.active_scheme_name == 'NoColor':
293 293 self.color_scheme_table.set_active_scheme(self.old_scheme)
294 294 self.Colors = self.color_scheme_table.active_colors
295 295 else:
296 296 self.old_scheme = self.color_scheme_table.active_scheme_name
297 297 self.color_scheme_table.set_active_scheme('NoColor')
298 298 self.Colors = self.color_scheme_table.active_colors
299 299
300 300 def stb2text(self, stb):
301 301 """Convert a structured traceback (a list) to a string."""
302 302 return '\n'.join(stb)
303 303
304 304 def text(self, etype, value, tb, tb_offset=None, context=5):
305 305 """Return formatted traceback.
306 306
307 307 Subclasses may override this if they add extra arguments.
308 308 """
309 309 tb_list = self.structured_traceback(etype, value, tb,
310 310 tb_offset, context)
311 311 return self.stb2text(tb_list)
312 312
313 313 def structured_traceback(self, etype, evalue, tb, tb_offset=None,
314 314 context=5, mode=None):
315 315 """Return a list of traceback frames.
316 316
317 317 Must be implemented by each class.
318 318 """
319 319 raise NotImplementedError()
320 320
321 321
322 322 #---------------------------------------------------------------------------
323 323 class ListTB(TBTools):
324 324 """Print traceback information from a traceback list, with optional color.
325 325
326 326 Calling requires 3 arguments: (etype, evalue, elist)
327 327 as would be obtained by::
328 328
329 329 etype, evalue, tb = sys.exc_info()
330 330 if tb:
331 331 elist = traceback.extract_tb(tb)
332 332 else:
333 333 elist = None
334 334
335 335 It can thus be used by programs which need to process the traceback before
336 336 printing (such as console replacements based on the code module from the
337 337 standard library).
338 338
339 339 Because they are meant to be called without a full traceback (only a
340 340 list), instances of this class can't call the interactive pdb debugger."""
341 341
342 342 def __init__(self, color_scheme='NoColor', call_pdb=False, ostream=None, parent=None, config=None):
343 343 TBTools.__init__(self, color_scheme=color_scheme, call_pdb=call_pdb,
344 344 ostream=ostream, parent=parent,config=config)
345 345
346 346 def __call__(self, etype, value, elist):
347 347 self.ostream.flush()
348 348 self.ostream.write(self.text(etype, value, elist))
349 349 self.ostream.write('\n')
350 350
351 351 def _extract_tb(self, tb):
352 352 if tb:
353 353 return traceback.extract_tb(tb)
354 354 else:
355 355 return None
356 356
357 357 def structured_traceback(self, etype, evalue, etb=None, tb_offset=None,
358 358 context=5):
359 359 """Return a color formatted string with the traceback info.
360 360
361 361 Parameters
362 362 ----------
363 363 etype : exception type
364 364 Type of the exception raised.
365 365 evalue : object
366 366 Data stored in the exception
367 367 etb : object
368 368 If list: List of frames, see class docstring for details.
369 369 If Traceback: Traceback of the exception.
370 370 tb_offset : int, optional
371 371 Number of frames in the traceback to skip. If not given, the
372 372 instance evalue is used (set in constructor).
373 373 context : int, optional
374 374 Number of lines of context information to print.
375 375
376 376 Returns
377 377 -------
378 378 String with formatted exception.
379 379 """
380 380 # This is a workaround to get chained_exc_ids in recursive calls
381 381 # etb should not be a tuple if structured_traceback is not recursive
382 382 if isinstance(etb, tuple):
383 383 etb, chained_exc_ids = etb
384 384 else:
385 385 chained_exc_ids = set()
386 386
387 387 if isinstance(etb, list):
388 388 elist = etb
389 389 elif etb is not None:
390 390 elist = self._extract_tb(etb)
391 391 else:
392 392 elist = []
393 393 tb_offset = self.tb_offset if tb_offset is None else tb_offset
394 394 Colors = self.Colors
395 395 out_list = []
396 396 if elist:
397 397
398 398 if tb_offset and len(elist) > tb_offset:
399 399 elist = elist[tb_offset:]
400 400
401 401 out_list.append('Traceback %s(most recent call last)%s:' %
402 402 (Colors.normalEm, Colors.Normal) + '\n')
403 403 out_list.extend(self._format_list(elist))
404 404 # The exception info should be a single entry in the list.
405 405 lines = ''.join(self._format_exception_only(etype, evalue))
406 406 out_list.append(lines)
407 407
408 408 exception = self.get_parts_of_chained_exception(evalue)
409 409
410 410 if exception and not id(exception[1]) in chained_exc_ids:
411 411 chained_exception_message = self.prepare_chained_exception_message(
412 412 evalue.__cause__)[0]
413 413 etype, evalue, etb = exception
414 414 # Trace exception to avoid infinite 'cause' loop
415 415 chained_exc_ids.add(id(exception[1]))
416 416 chained_exceptions_tb_offset = 0
417 417 out_list = (
418 418 self.structured_traceback(
419 419 etype, evalue, (etb, chained_exc_ids),
420 420 chained_exceptions_tb_offset, context)
421 421 + chained_exception_message
422 422 + out_list)
423 423
424 424 return out_list
425 425
426 426 def _format_list(self, extracted_list):
427 427 """Format a list of traceback entry tuples for printing.
428 428
429 429 Given a list of tuples as returned by extract_tb() or
430 430 extract_stack(), return a list of strings ready for printing.
431 431 Each string in the resulting list corresponds to the item with the
432 432 same index in the argument list. Each string ends in a newline;
433 433 the strings may contain internal newlines as well, for those items
434 434 whose source text line is not None.
435 435
436 436 Lifted almost verbatim from traceback.py
437 437 """
438 438
439 439 Colors = self.Colors
440 440 list = []
441 441 for filename, lineno, name, line in extracted_list[:-1]:
442 442 item = " %s, line %s%d%s, in %s%s%s\n" % (
443 443 _format_filename(filename, Colors.filename, Colors.Normal),
444 444 Colors.lineno,
445 445 lineno,
446 446 Colors.Normal,
447 447 Colors.name,
448 448 name,
449 449 Colors.Normal,
450 450 )
451 451 if line:
452 452 item += ' %s\n' % line.strip()
453 453 list.append(item)
454 454 # Emphasize the last entry
455 455 filename, lineno, name, line = extracted_list[-1]
456 456 item = "%s %s, line %s%d%s, in %s%s%s%s\n" % (
457 457 Colors.normalEm,
458 458 _format_filename(filename, Colors.filenameEm, Colors.normalEm),
459 459 Colors.linenoEm,
460 460 lineno,
461 461 Colors.normalEm,
462 462 Colors.nameEm,
463 463 name,
464 464 Colors.normalEm,
465 465 Colors.Normal,
466 466 )
467 467 if line:
468 468 item += '%s %s%s\n' % (Colors.line, line.strip(),
469 469 Colors.Normal)
470 470 list.append(item)
471 471 return list
472 472
473 473 def _format_exception_only(self, etype, value):
474 474 """Format the exception part of a traceback.
475 475
476 476 The arguments are the exception type and value such as given by
477 477 sys.exc_info()[:2]. The return value is a list of strings, each ending
478 478 in a newline. Normally, the list contains a single string; however,
479 479 for SyntaxError exceptions, it contains several lines that (when
480 480 printed) display detailed information about where the syntax error
481 481 occurred. The message indicating which exception occurred is the
482 482 always last string in the list.
483 483
484 484 Also lifted nearly verbatim from traceback.py
485 485 """
486 486 have_filedata = False
487 487 Colors = self.Colors
488 488 list = []
489 489 stype = py3compat.cast_unicode(Colors.excName + etype.__name__ + Colors.Normal)
490 490 if value is None:
491 491 # Not sure if this can still happen in Python 2.6 and above
492 492 list.append(stype + '\n')
493 493 else:
494 494 if issubclass(etype, SyntaxError):
495 495 have_filedata = True
496 496 if not value.filename: value.filename = "<string>"
497 497 if value.lineno:
498 498 lineno = value.lineno
499 499 textline = linecache.getline(value.filename, value.lineno)
500 500 else:
501 501 lineno = "unknown"
502 502 textline = ""
503 503 list.append(
504 504 "%s %s, line %s%s%s\n"
505 505 % (
506 506 Colors.normalEm,
507 _format_filename(value.filename, Colors.filenameEm, Colors.normalEm),
507 _format_filename(
508 value.filename, Colors.filenameEm, Colors.normalEm
509 ),
508 510 Colors.linenoEm,
509 511 lineno,
510 512 Colors.Normal,
511 513 )
512 514 )
513 515 if textline == "":
514 516 textline = py3compat.cast_unicode(value.text, "utf-8")
515 517
516 518 if textline is not None:
517 519 i = 0
518 520 while i < len(textline) and textline[i].isspace():
519 521 i += 1
520 522 list.append('%s %s%s\n' % (Colors.line,
521 523 textline.strip(),
522 524 Colors.Normal))
523 525 if value.offset is not None:
524 526 s = ' '
525 527 for c in textline[i:value.offset - 1]:
526 528 if c.isspace():
527 529 s += c
528 530 else:
529 531 s += ' '
530 532 list.append('%s%s^%s\n' % (Colors.caret, s,
531 533 Colors.Normal))
532 534
533 535 try:
534 536 s = value.msg
535 537 except Exception:
536 538 s = self._some_str(value)
537 539 if s:
538 540 list.append('%s%s:%s %s\n' % (stype, Colors.excName,
539 541 Colors.Normal, s))
540 542 else:
541 543 list.append('%s\n' % stype)
542 544
543 545 # sync with user hooks
544 546 if have_filedata:
545 547 ipinst = get_ipython()
546 548 if ipinst is not None:
547 549 ipinst.hooks.synchronize_with_editor(value.filename, value.lineno, 0)
548 550
549 551 return list
550 552
551 553 def get_exception_only(self, etype, value):
552 554 """Only print the exception type and message, without a traceback.
553 555
554 556 Parameters
555 557 ----------
556 558 etype : exception type
557 559 evalue : exception value
558 560 """
559 561 return ListTB.structured_traceback(self, etype, value)
560 562
561 563 def show_exception_only(self, etype, evalue):
562 564 """Only print the exception type and message, without a traceback.
563 565
564 566 Parameters
565 567 ----------
566 568 etype : exception type
567 569 evalue : exception value
568 570 """
569 571 # This method needs to use __call__ from *this* class, not the one from
570 572 # a subclass whose signature or behavior may be different
571 573 ostream = self.ostream
572 574 ostream.flush()
573 575 ostream.write('\n'.join(self.get_exception_only(etype, evalue)))
574 576 ostream.flush()
575 577
576 578 def _some_str(self, value):
577 579 # Lifted from traceback.py
578 580 try:
579 581 return py3compat.cast_unicode(str(value))
580 582 except:
581 583 return u'<unprintable %s object>' % type(value).__name__
582 584
583 585
584 586 #----------------------------------------------------------------------------
585 587 class VerboseTB(TBTools):
586 588 """A port of Ka-Ping Yee's cgitb.py module that outputs color text instead
587 589 of HTML. Requires inspect and pydoc. Crazy, man.
588 590
589 591 Modified version which optionally strips the topmost entries from the
590 592 traceback, to be used with alternate interpreters (because their own code
591 593 would appear in the traceback)."""
592 594
593 595 def __init__(self, color_scheme='Linux', call_pdb=False, ostream=None,
594 596 tb_offset=0, long_header=False, include_vars=True,
595 597 check_cache=None, debugger_cls = None,
596 598 parent=None, config=None):
597 599 """Specify traceback offset, headers and color scheme.
598 600
599 601 Define how many frames to drop from the tracebacks. Calling it with
600 602 tb_offset=1 allows use of this handler in interpreters which will have
601 603 their own code at the top of the traceback (VerboseTB will first
602 604 remove that frame before printing the traceback info)."""
603 605 TBTools.__init__(self, color_scheme=color_scheme, call_pdb=call_pdb,
604 606 ostream=ostream, parent=parent, config=config)
605 607 self.tb_offset = tb_offset
606 608 self.long_header = long_header
607 609 self.include_vars = include_vars
608 610 # By default we use linecache.checkcache, but the user can provide a
609 611 # different check_cache implementation. This is used by the IPython
610 612 # kernel to provide tracebacks for interactive code that is cached,
611 613 # by a compiler instance that flushes the linecache but preserves its
612 614 # own code cache.
613 615 if check_cache is None:
614 616 check_cache = linecache.checkcache
615 617 self.check_cache = check_cache
616 618
617 619 self.debugger_cls = debugger_cls or debugger.Pdb
618 620 self.skip_hidden = True
619 621
620 622 def format_record(self, frame_info):
621 623 """Format a single stack frame"""
622 624 Colors = self.Colors # just a shorthand + quicker name lookup
623 625 ColorsNormal = Colors.Normal # used a lot
624 626
625 627 if isinstance(frame_info, stack_data.RepeatedFrames):
626 628 return ' %s[... skipping similar frames: %s]%s\n' % (
627 629 Colors.excName, frame_info.description, ColorsNormal)
628 630
629 631 indent = ' ' * INDENT_SIZE
630 632 em_normal = '%s\n%s%s' % (Colors.valEm, indent, ColorsNormal)
631 633 tpl_call = 'in %s%%s%s%%s%s' % (Colors.vName, Colors.valEm,
632 634 ColorsNormal)
633 635 tpl_call_fail = 'in %s%%s%s(***failed resolving arguments***)%s' % \
634 636 (Colors.vName, Colors.valEm, ColorsNormal)
635 637 tpl_name_val = '%%s %s= %%s%s' % (Colors.valEm, ColorsNormal)
636 638
637 639 link = _format_filename(frame_info.filename, Colors.filenameEm, ColorsNormal)
638 640 args, varargs, varkw, locals_ = inspect.getargvalues(frame_info.frame)
639 641
640 642 func = frame_info.executing.code_qualname()
641 643 if func == '<module>':
642 644 call = tpl_call % (func, '')
643 645 else:
644 646 # Decide whether to include variable details or not
645 647 var_repr = eqrepr if self.include_vars else nullrepr
646 648 try:
647 649 call = tpl_call % (func, inspect.formatargvalues(args,
648 650 varargs, varkw,
649 651 locals_, formatvalue=var_repr))
650 652 except KeyError:
651 653 # This happens in situations like errors inside generator
652 654 # expressions, where local variables are listed in the
653 655 # line, but can't be extracted from the frame. I'm not
654 656 # 100% sure this isn't actually a bug in inspect itself,
655 657 # but since there's no info for us to compute with, the
656 658 # best we can do is report the failure and move on. Here
657 659 # we must *not* call any traceback construction again,
658 660 # because that would mess up use of %debug later on. So we
659 661 # simply report the failure and move on. The only
660 662 # limitation will be that this frame won't have locals
661 663 # listed in the call signature. Quite subtle problem...
662 664 # I can't think of a good way to validate this in a unit
663 665 # test, but running a script consisting of:
664 666 # dict( (k,v.strip()) for (k,v) in range(10) )
665 667 # will illustrate the error, if this exception catch is
666 668 # disabled.
667 669 call = tpl_call_fail % func
668 670
669 671 lvals = ''
670 672 lvals_list = []
671 673 if self.include_vars:
672 674 for var in frame_info.variables_in_executing_piece:
673 675 lvals_list.append(tpl_name_val % (var.name, repr(var.value)))
674 676 if lvals_list:
675 677 lvals = '%s%s' % (indent, em_normal.join(lvals_list))
676 678
677 679 result = "%s, %s\n" % (link, call)
678 680
679 681 result += ''.join(_format_traceback_lines(frame_info.lines, Colors, self.has_colors, lvals))
680 682 return result
681 683
682 684 def prepare_header(self, etype, long_version=False):
683 685 colors = self.Colors # just a shorthand + quicker name lookup
684 686 colorsnormal = colors.Normal # used a lot
685 687 exc = '%s%s%s' % (colors.excName, etype, colorsnormal)
686 688 width = min(75, get_terminal_size()[0])
687 689 if long_version:
688 690 # Header with the exception type, python version, and date
689 691 pyver = 'Python ' + sys.version.split()[0] + ': ' + sys.executable
690 692 date = time.ctime(time.time())
691 693
692 694 head = '%s%s%s\n%s%s%s\n%s' % (colors.topline, '-' * width, colorsnormal,
693 695 exc, ' ' * (width - len(str(etype)) - len(pyver)),
694 696 pyver, date.rjust(width) )
695 697 head += "\nA problem occurred executing Python code. Here is the sequence of function" \
696 698 "\ncalls leading up to the error, with the most recent (innermost) call last."
697 699 else:
698 700 # Simplified header
699 701 head = '%s%s' % (exc, 'Traceback (most recent call last)'. \
700 702 rjust(width - len(str(etype))) )
701 703
702 704 return head
703 705
704 706 def format_exception(self, etype, evalue):
705 707 colors = self.Colors # just a shorthand + quicker name lookup
706 708 colorsnormal = colors.Normal # used a lot
707 709 # Get (safely) a string form of the exception info
708 710 try:
709 711 etype_str, evalue_str = map(str, (etype, evalue))
710 712 except:
711 713 # User exception is improperly defined.
712 714 etype, evalue = str, sys.exc_info()[:2]
713 715 etype_str, evalue_str = map(str, (etype, evalue))
714 716 # ... and format it
715 717 return ['%s%s%s: %s' % (colors.excName, etype_str,
716 718 colorsnormal, py3compat.cast_unicode(evalue_str))]
717 719
718 720 def format_exception_as_a_whole(self, etype, evalue, etb, number_of_lines_of_context, tb_offset):
719 721 """Formats the header, traceback and exception message for a single exception.
720 722
721 723 This may be called multiple times by Python 3 exception chaining
722 724 (PEP 3134).
723 725 """
724 726 # some locals
725 727 orig_etype = etype
726 728 try:
727 729 etype = etype.__name__
728 730 except AttributeError:
729 731 pass
730 732
731 733 tb_offset = self.tb_offset if tb_offset is None else tb_offset
732 734 head = self.prepare_header(etype, self.long_header)
733 735 records = self.get_records(etb, number_of_lines_of_context, tb_offset)
734 736
735 737 frames = []
736 738 skipped = 0
737 739 lastrecord = len(records) - 1
738 740 for i, r in enumerate(records):
739 741 if not isinstance(r, stack_data.RepeatedFrames) and self.skip_hidden:
740 742 if r.frame.f_locals.get("__tracebackhide__", 0) and i != lastrecord:
741 743 skipped += 1
742 744 continue
743 745 if skipped:
744 746 Colors = self.Colors # just a shorthand + quicker name lookup
745 747 ColorsNormal = Colors.Normal # used a lot
746 748 frames.append(
747 749 " %s[... skipping hidden %s frame]%s\n"
748 750 % (Colors.excName, skipped, ColorsNormal)
749 751 )
750 752 skipped = 0
751 753 frames.append(self.format_record(r))
752 754 if skipped:
753 755 Colors = self.Colors # just a shorthand + quicker name lookup
754 756 ColorsNormal = Colors.Normal # used a lot
755 757 frames.append(
756 758 " %s[... skipping hidden %s frame]%s\n"
757 759 % (Colors.excName, skipped, ColorsNormal)
758 760 )
759 761
760 762 formatted_exception = self.format_exception(etype, evalue)
761 763 if records:
762 764 frame_info = records[-1]
763 765 ipinst = get_ipython()
764 766 if ipinst is not None:
765 767 ipinst.hooks.synchronize_with_editor(frame_info.filename, frame_info.lineno, 0)
766 768
767 769 return [[head] + frames + [''.join(formatted_exception[0])]]
768 770
769 771 def get_records(self, etb, number_of_lines_of_context, tb_offset):
770 772 context = number_of_lines_of_context - 1
771 773 after = context // 2
772 774 before = context - after
773 775 if self.has_colors:
774 776 style = get_style_by_name('default')
775 777 style = stack_data.style_with_executing_node(style, 'bg:#00005f')
776 778 formatter = Terminal256Formatter(style=style)
777 779 else:
778 780 formatter = None
779 781 options = stack_data.Options(
780 782 before=before,
781 783 after=after,
782 784 pygments_formatter=formatter,
783 785 )
784 786 return list(stack_data.FrameInfo.stack_data(etb, options=options))[tb_offset:]
785 787
786 788 def structured_traceback(self, etype, evalue, etb, tb_offset=None,
787 789 number_of_lines_of_context=5):
788 790 """Return a nice text document describing the traceback."""
789 791
790 792 formatted_exception = self.format_exception_as_a_whole(etype, evalue, etb, number_of_lines_of_context,
791 793 tb_offset)
792 794
793 795 colors = self.Colors # just a shorthand + quicker name lookup
794 796 colorsnormal = colors.Normal # used a lot
795 797 head = '%s%s%s' % (colors.topline, '-' * min(75, get_terminal_size()[0]), colorsnormal)
796 798 structured_traceback_parts = [head]
797 799 chained_exceptions_tb_offset = 0
798 800 lines_of_context = 3
799 801 formatted_exceptions = formatted_exception
800 802 exception = self.get_parts_of_chained_exception(evalue)
801 803 if exception:
802 804 formatted_exceptions += self.prepare_chained_exception_message(evalue.__cause__)
803 805 etype, evalue, etb = exception
804 806 else:
805 807 evalue = None
806 808 chained_exc_ids = set()
807 809 while evalue:
808 810 formatted_exceptions += self.format_exception_as_a_whole(etype, evalue, etb, lines_of_context,
809 811 chained_exceptions_tb_offset)
810 812 exception = self.get_parts_of_chained_exception(evalue)
811 813
812 814 if exception and not id(exception[1]) in chained_exc_ids:
813 815 chained_exc_ids.add(id(exception[1])) # trace exception to avoid infinite 'cause' loop
814 816 formatted_exceptions += self.prepare_chained_exception_message(evalue.__cause__)
815 817 etype, evalue, etb = exception
816 818 else:
817 819 evalue = None
818 820
819 821 # we want to see exceptions in a reversed order:
820 822 # the first exception should be on top
821 823 for formatted_exception in reversed(formatted_exceptions):
822 824 structured_traceback_parts += formatted_exception
823 825
824 826 return structured_traceback_parts
825 827
826 828 def debugger(self, force=False):
827 829 """Call up the pdb debugger if desired, always clean up the tb
828 830 reference.
829 831
830 832 Keywords:
831 833
832 834 - force(False): by default, this routine checks the instance call_pdb
833 835 flag and does not actually invoke the debugger if the flag is false.
834 836 The 'force' option forces the debugger to activate even if the flag
835 837 is false.
836 838
837 839 If the call_pdb flag is set, the pdb interactive debugger is
838 840 invoked. In all cases, the self.tb reference to the current traceback
839 841 is deleted to prevent lingering references which hamper memory
840 842 management.
841 843
842 844 Note that each call to pdb() does an 'import readline', so if your app
843 845 requires a special setup for the readline completers, you'll have to
844 846 fix that by hand after invoking the exception handler."""
845 847
846 848 if force or self.call_pdb:
847 849 if self.pdb is None:
848 850 self.pdb = self.debugger_cls()
849 851 # the system displayhook may have changed, restore the original
850 852 # for pdb
851 853 display_trap = DisplayTrap(hook=sys.__displayhook__)
852 854 with display_trap:
853 855 self.pdb.reset()
854 856 # Find the right frame so we don't pop up inside ipython itself
855 857 if hasattr(self, 'tb') and self.tb is not None:
856 858 etb = self.tb
857 859 else:
858 860 etb = self.tb = sys.last_traceback
859 861 while self.tb is not None and self.tb.tb_next is not None:
860 862 self.tb = self.tb.tb_next
861 863 if etb and etb.tb_next:
862 864 etb = etb.tb_next
863 865 self.pdb.botframe = etb.tb_frame
864 866 self.pdb.interaction(None, etb)
865 867
866 868 if hasattr(self, 'tb'):
867 869 del self.tb
868 870
869 871 def handler(self, info=None):
870 872 (etype, evalue, etb) = info or sys.exc_info()
871 873 self.tb = etb
872 874 ostream = self.ostream
873 875 ostream.flush()
874 876 ostream.write(self.text(etype, evalue, etb))
875 877 ostream.write('\n')
876 878 ostream.flush()
877 879
878 880 # Changed so an instance can just be called as VerboseTB_inst() and print
879 881 # out the right info on its own.
880 882 def __call__(self, etype=None, evalue=None, etb=None):
881 883 """This hook can replace sys.excepthook (for Python 2.1 or higher)."""
882 884 if etb is None:
883 885 self.handler()
884 886 else:
885 887 self.handler((etype, evalue, etb))
886 888 try:
887 889 self.debugger()
888 890 except KeyboardInterrupt:
889 891 print("\nKeyboardInterrupt")
890 892
891 893
892 894 #----------------------------------------------------------------------------
893 895 class FormattedTB(VerboseTB, ListTB):
894 896 """Subclass ListTB but allow calling with a traceback.
895 897
896 898 It can thus be used as a sys.excepthook for Python > 2.1.
897 899
898 900 Also adds 'Context' and 'Verbose' modes, not available in ListTB.
899 901
900 902 Allows a tb_offset to be specified. This is useful for situations where
901 903 one needs to remove a number of topmost frames from the traceback (such as
902 904 occurs with python programs that themselves execute other python code,
903 905 like Python shells). """
904 906
905 907 def __init__(self, mode='Plain', color_scheme='Linux', call_pdb=False,
906 908 ostream=None,
907 909 tb_offset=0, long_header=False, include_vars=False,
908 910 check_cache=None, debugger_cls=None,
909 911 parent=None, config=None):
910 912
911 913 # NEVER change the order of this list. Put new modes at the end:
912 914 self.valid_modes = ['Plain', 'Context', 'Verbose', 'Minimal']
913 915 self.verbose_modes = self.valid_modes[1:3]
914 916
915 917 VerboseTB.__init__(self, color_scheme=color_scheme, call_pdb=call_pdb,
916 918 ostream=ostream, tb_offset=tb_offset,
917 919 long_header=long_header, include_vars=include_vars,
918 920 check_cache=check_cache, debugger_cls=debugger_cls,
919 921 parent=parent, config=config)
920 922
921 923 # Different types of tracebacks are joined with different separators to
922 924 # form a single string. They are taken from this dict
923 925 self._join_chars = dict(Plain='', Context='\n', Verbose='\n',
924 926 Minimal='')
925 927 # set_mode also sets the tb_join_char attribute
926 928 self.set_mode(mode)
927 929
928 930 def structured_traceback(self, etype, value, tb, tb_offset=None, number_of_lines_of_context=5):
929 931 tb_offset = self.tb_offset if tb_offset is None else tb_offset
930 932 mode = self.mode
931 933 if mode in self.verbose_modes:
932 934 # Verbose modes need a full traceback
933 935 return VerboseTB.structured_traceback(
934 936 self, etype, value, tb, tb_offset, number_of_lines_of_context
935 937 )
936 938 elif mode == 'Minimal':
937 939 return ListTB.get_exception_only(self, etype, value)
938 940 else:
939 941 # We must check the source cache because otherwise we can print
940 942 # out-of-date source code.
941 943 self.check_cache()
942 944 # Now we can extract and format the exception
943 945 return ListTB.structured_traceback(
944 946 self, etype, value, tb, tb_offset, number_of_lines_of_context
945 947 )
946 948
947 949 def stb2text(self, stb):
948 950 """Convert a structured traceback (a list) to a string."""
949 951 return self.tb_join_char.join(stb)
950 952
951 953
952 954 def set_mode(self, mode=None):
953 955 """Switch to the desired mode.
954 956
955 957 If mode is not specified, cycles through the available modes."""
956 958
957 959 if not mode:
958 960 new_idx = (self.valid_modes.index(self.mode) + 1 ) % \
959 961 len(self.valid_modes)
960 962 self.mode = self.valid_modes[new_idx]
961 963 elif mode not in self.valid_modes:
962 964 raise ValueError('Unrecognized mode in FormattedTB: <' + mode + '>\n'
963 965 'Valid modes: ' + str(self.valid_modes))
964 966 else:
965 967 self.mode = mode
966 968 # include variable details only in 'Verbose' mode
967 969 self.include_vars = (self.mode == self.valid_modes[2])
968 970 # Set the join character for generating text tracebacks
969 971 self.tb_join_char = self._join_chars[self.mode]
970 972
971 973 # some convenient shortcuts
972 974 def plain(self):
973 975 self.set_mode(self.valid_modes[0])
974 976
975 977 def context(self):
976 978 self.set_mode(self.valid_modes[1])
977 979
978 980 def verbose(self):
979 981 self.set_mode(self.valid_modes[2])
980 982
981 983 def minimal(self):
982 984 self.set_mode(self.valid_modes[3])
983 985
984 986
985 987 #----------------------------------------------------------------------------
986 988 class AutoFormattedTB(FormattedTB):
987 989 """A traceback printer which can be called on the fly.
988 990
989 991 It will find out about exceptions by itself.
990 992
991 993 A brief example::
992 994
993 995 AutoTB = AutoFormattedTB(mode = 'Verbose',color_scheme='Linux')
994 996 try:
995 997 ...
996 998 except:
997 999 AutoTB() # or AutoTB(out=logfile) where logfile is an open file object
998 1000 """
999 1001
1000 1002 def __call__(self, etype=None, evalue=None, etb=None,
1001 1003 out=None, tb_offset=None):
1002 1004 """Print out a formatted exception traceback.
1003 1005
1004 1006 Optional arguments:
1005 1007 - out: an open file-like object to direct output to.
1006 1008
1007 1009 - tb_offset: the number of frames to skip over in the stack, on a
1008 1010 per-call basis (this overrides temporarily the instance's tb_offset
1009 1011 given at initialization time."""
1010 1012
1011 1013 if out is None:
1012 1014 out = self.ostream
1013 1015 out.flush()
1014 1016 out.write(self.text(etype, evalue, etb, tb_offset))
1015 1017 out.write('\n')
1016 1018 out.flush()
1017 1019 # FIXME: we should remove the auto pdb behavior from here and leave
1018 1020 # that to the clients.
1019 1021 try:
1020 1022 self.debugger()
1021 1023 except KeyboardInterrupt:
1022 1024 print("\nKeyboardInterrupt")
1023 1025
1024 1026 def structured_traceback(self, etype=None, value=None, tb=None,
1025 1027 tb_offset=None, number_of_lines_of_context=5):
1026 1028 if etype is None:
1027 1029 etype, value, tb = sys.exc_info()
1028 1030 if isinstance(tb, tuple):
1029 1031 # tb is a tuple if this is a chained exception.
1030 1032 self.tb = tb[0]
1031 1033 else:
1032 1034 self.tb = tb
1033 1035 return FormattedTB.structured_traceback(
1034 1036 self, etype, value, tb, tb_offset, number_of_lines_of_context)
1035 1037
1036 1038
1037 1039 #---------------------------------------------------------------------------
1038 1040
1039 1041 # A simple class to preserve Nathan's original functionality.
1040 1042 class ColorTB(FormattedTB):
1041 1043 """Shorthand to initialize a FormattedTB in Linux colors mode."""
1042 1044
1043 1045 def __init__(self, color_scheme='Linux', call_pdb=0, **kwargs):
1044 1046 FormattedTB.__init__(self, color_scheme=color_scheme,
1045 1047 call_pdb=call_pdb, **kwargs)
1046 1048
1047 1049
1048 1050 class SyntaxTB(ListTB):
1049 1051 """Extension which holds some state: the last exception value"""
1050 1052
1051 1053 def __init__(self, color_scheme='NoColor', parent=None, config=None):
1052 1054 ListTB.__init__(self, color_scheme, parent=parent, config=config)
1053 1055 self.last_syntax_error = None
1054 1056
1055 1057 def __call__(self, etype, value, elist):
1056 1058 self.last_syntax_error = value
1057 1059
1058 1060 ListTB.__call__(self, etype, value, elist)
1059 1061
1060 1062 def structured_traceback(self, etype, value, elist, tb_offset=None,
1061 1063 context=5):
1062 1064 # If the source file has been edited, the line in the syntax error can
1063 1065 # be wrong (retrieved from an outdated cache). This replaces it with
1064 1066 # the current value.
1065 1067 if isinstance(value, SyntaxError) \
1066 1068 and isinstance(value.filename, str) \
1067 1069 and isinstance(value.lineno, int):
1068 1070 linecache.checkcache(value.filename)
1069 1071 newtext = linecache.getline(value.filename, value.lineno)
1070 1072 if newtext:
1071 1073 value.text = newtext
1072 1074 self.last_syntax_error = value
1073 1075 return super(SyntaxTB, self).structured_traceback(etype, value, elist,
1074 1076 tb_offset=tb_offset, context=context)
1075 1077
1076 1078 def clear_err_state(self):
1077 1079 """Return the current error state and clear it"""
1078 1080 e = self.last_syntax_error
1079 1081 self.last_syntax_error = None
1080 1082 return e
1081 1083
1082 1084 def stb2text(self, stb):
1083 1085 """Convert a structured traceback (a list) to a string."""
1084 1086 return ''.join(stb)
1085 1087
1086 1088
1087 1089 # some internal-use functions
1088 1090 def text_repr(value):
1089 1091 """Hopefully pretty robust repr equivalent."""
1090 1092 # this is pretty horrible but should always return *something*
1091 1093 try:
1092 1094 return pydoc.text.repr(value)
1093 1095 except KeyboardInterrupt:
1094 1096 raise
1095 1097 except:
1096 1098 try:
1097 1099 return repr(value)
1098 1100 except KeyboardInterrupt:
1099 1101 raise
1100 1102 except:
1101 1103 try:
1102 1104 # all still in an except block so we catch
1103 1105 # getattr raising
1104 1106 name = getattr(value, '__name__', None)
1105 1107 if name:
1106 1108 # ick, recursion
1107 1109 return text_repr(name)
1108 1110 klass = getattr(value, '__class__', None)
1109 1111 if klass:
1110 1112 return '%s instance' % text_repr(klass)
1111 1113 except KeyboardInterrupt:
1112 1114 raise
1113 1115 except:
1114 1116 return 'UNRECOVERABLE REPR FAILURE'
1115 1117
1116 1118
1117 1119 def eqrepr(value, repr=text_repr):
1118 1120 return '=%s' % repr(value)
1119 1121
1120 1122
1121 1123 def nullrepr(value, repr=text_repr):
1122 1124 return ''
General Comments 0
You need to be logged in to leave comments. Login now