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