##// END OF EJS Templates
Don't attempt to tokenize binary files for tracebacks....
Thomas Kluyver -
Show More
@@ -1,1232 +1,1237 b''
1 1 # -*- coding: utf-8 -*-
2 2 """
3 3 ultratb.py -- Spice up your tracebacks!
4 4
5 5 * ColorTB
6 6 I've always found it a bit hard to visually parse tracebacks in Python. The
7 7 ColorTB class is a solution to that problem. It colors the different parts of a
8 8 traceback in a manner similar to what you would expect from a syntax-highlighting
9 9 text editor.
10 10
11 11 Installation instructions for ColorTB:
12 12 import sys,ultratb
13 13 sys.excepthook = ultratb.ColorTB()
14 14
15 15 * VerboseTB
16 16 I've also included a port of Ka-Ping Yee's "cgitb.py" that produces all kinds
17 17 of useful info when a traceback occurs. Ping originally had it spit out HTML
18 18 and intended it for CGI programmers, but why should they have all the fun? I
19 19 altered it to spit out colored text to the terminal. It's a bit overwhelming,
20 20 but kind of neat, and maybe useful for long-running programs that you believe
21 21 are bug-free. If a crash *does* occur in that type of program you want details.
22 22 Give it a shot--you'll love it or you'll hate it.
23 23
24 24 Note:
25 25
26 26 The Verbose mode prints the variables currently visible where the exception
27 27 happened (shortening their strings if too long). This can potentially be
28 28 very slow, if you happen to have a huge data structure whose string
29 29 representation is complex to compute. Your computer may appear to freeze for
30 30 a while with cpu usage at 100%. If this occurs, you can cancel the traceback
31 31 with Ctrl-C (maybe hitting it more than once).
32 32
33 33 If you encounter this kind of situation often, you may want to use the
34 34 Verbose_novars mode instead of the regular Verbose, which avoids formatting
35 35 variables (but otherwise includes the information and context given by
36 36 Verbose).
37 37
38 38
39 39 Installation instructions for ColorTB:
40 40 import sys,ultratb
41 41 sys.excepthook = ultratb.VerboseTB()
42 42
43 43 Note: Much of the code in this module was lifted verbatim from the standard
44 44 library module 'traceback.py' and Ka-Ping Yee's 'cgitb.py'.
45 45
46 46 * Color schemes
47 47 The colors are defined in the class TBTools through the use of the
48 48 ColorSchemeTable class. Currently the following exist:
49 49
50 50 - NoColor: allows all of this module to be used in any terminal (the color
51 51 escapes are just dummy blank strings).
52 52
53 53 - Linux: is meant to look good in a terminal like the Linux console (black
54 54 or very dark background).
55 55
56 56 - LightBG: similar to Linux but swaps dark/light colors to be more readable
57 57 in light background terminals.
58 58
59 59 You can implement other color schemes easily, the syntax is fairly
60 60 self-explanatory. Please send back new schemes you develop to the author for
61 61 possible inclusion in future releases.
62 62 """
63 63
64 64 #*****************************************************************************
65 65 # Copyright (C) 2001 Nathaniel Gray <n8gray@caltech.edu>
66 66 # Copyright (C) 2001-2004 Fernando Perez <fperez@colorado.edu>
67 67 #
68 68 # Distributed under the terms of the BSD License. The full license is in
69 69 # the file COPYING, distributed as part of this software.
70 70 #*****************************************************************************
71 71
72 72 from __future__ import with_statement
73 73
74 74 import inspect
75 75 import keyword
76 76 import linecache
77 77 import os
78 78 import pydoc
79 79 import re
80 80 import sys
81 81 import time
82 82 import tokenize
83 83 import traceback
84 84 import types
85 85
86 86 try: # Python 2
87 87 generate_tokens = tokenize.generate_tokens
88 88 except AttributeError: # Python 3
89 89 generate_tokens = tokenize.tokenize
90 90
91 91 # For purposes of monkeypatching inspect to fix a bug in it.
92 92 from inspect import getsourcefile, getfile, getmodule,\
93 93 ismodule, isclass, ismethod, isfunction, istraceback, isframe, iscode
94 94
95 95 # IPython's own modules
96 96 # Modified pdb which doesn't damage IPython's readline handling
97 97 from IPython.core import debugger, ipapi
98 98 from IPython.core.display_trap import DisplayTrap
99 99 from IPython.core.excolors import exception_colors
100 100 from IPython.utils import PyColorize
101 101 from IPython.utils import io
102 102 from IPython.utils import py3compat
103 103 from IPython.utils import pyfile
104 104 from IPython.utils.data import uniq_stable
105 105 from IPython.utils.warn import info, error
106 106
107 107 # Globals
108 108 # amount of space to put line numbers before verbose tracebacks
109 109 INDENT_SIZE = 8
110 110
111 111 # Default color scheme. This is used, for example, by the traceback
112 112 # formatter. When running in an actual IPython instance, the user's rc.colors
113 113 # value is used, but havinga module global makes this functionality available
114 114 # to users of ultratb who are NOT running inside ipython.
115 115 DEFAULT_SCHEME = 'NoColor'
116 116
117 117 #---------------------------------------------------------------------------
118 118 # Code begins
119 119
120 120 # Utility functions
121 121 def inspect_error():
122 122 """Print a message about internal inspect errors.
123 123
124 124 These are unfortunately quite common."""
125 125
126 126 error('Internal Python error in the inspect module.\n'
127 127 'Below is the traceback from this internal error.\n')
128 128
129 129
130 130 # N.B. This function is a monkeypatch we are currently not applying.
131 131 # It was written some time ago, to fix an apparent Python bug with
132 132 # codeobj.co_firstlineno . Unfortunately, we don't know under what conditions
133 133 # the bug occurred, so we can't tell if it has been fixed. If it reappears, we
134 134 # will apply the monkeypatch again. Also, note that findsource() is not called
135 135 # by our code at this time - we don't know if it was when the monkeypatch was
136 136 # written, or if the monkeypatch is needed for some other code (like a debugger).
137 137 # For the discussion about not applying it, see gh-1229. TK, Jan 2011.
138 138 def findsource(object):
139 139 """Return the entire source file and starting line number for an object.
140 140
141 141 The argument may be a module, class, method, function, traceback, frame,
142 142 or code object. The source code is returned as a list of all the lines
143 143 in the file and the line number indexes a line in that list. An IOError
144 144 is raised if the source code cannot be retrieved.
145 145
146 146 FIXED version with which we monkeypatch the stdlib to work around a bug."""
147 147
148 148 file = getsourcefile(object) or getfile(object)
149 149 # If the object is a frame, then trying to get the globals dict from its
150 150 # module won't work. Instead, the frame object itself has the globals
151 151 # dictionary.
152 152 globals_dict = None
153 153 if inspect.isframe(object):
154 154 # XXX: can this ever be false?
155 155 globals_dict = object.f_globals
156 156 else:
157 157 module = getmodule(object, file)
158 158 if module:
159 159 globals_dict = module.__dict__
160 160 lines = linecache.getlines(file, globals_dict)
161 161 if not lines:
162 162 raise IOError('could not get source code')
163 163
164 164 if ismodule(object):
165 165 return lines, 0
166 166
167 167 if isclass(object):
168 168 name = object.__name__
169 169 pat = re.compile(r'^(\s*)class\s*' + name + r'\b')
170 170 # make some effort to find the best matching class definition:
171 171 # use the one with the least indentation, which is the one
172 172 # that's most probably not inside a function definition.
173 173 candidates = []
174 174 for i in range(len(lines)):
175 175 match = pat.match(lines[i])
176 176 if match:
177 177 # if it's at toplevel, it's already the best one
178 178 if lines[i][0] == 'c':
179 179 return lines, i
180 180 # else add whitespace to candidate list
181 181 candidates.append((match.group(1), i))
182 182 if candidates:
183 183 # this will sort by whitespace, and by line number,
184 184 # less whitespace first
185 185 candidates.sort()
186 186 return lines, candidates[0][1]
187 187 else:
188 188 raise IOError('could not find class definition')
189 189
190 190 if ismethod(object):
191 191 object = object.im_func
192 192 if isfunction(object):
193 193 object = object.func_code
194 194 if istraceback(object):
195 195 object = object.tb_frame
196 196 if isframe(object):
197 197 object = object.f_code
198 198 if iscode(object):
199 199 if not hasattr(object, 'co_firstlineno'):
200 200 raise IOError('could not find function definition')
201 201 pat = re.compile(r'^(\s*def\s)|(.*(?<!\w)lambda(:|\s))|^(\s*@)')
202 202 pmatch = pat.match
203 203 # fperez - fix: sometimes, co_firstlineno can give a number larger than
204 204 # the length of lines, which causes an error. Safeguard against that.
205 205 lnum = min(object.co_firstlineno,len(lines))-1
206 206 while lnum > 0:
207 207 if pmatch(lines[lnum]): break
208 208 lnum -= 1
209 209
210 210 return lines, lnum
211 211 raise IOError('could not find code object')
212 212
213 213 # Not applying the monkeypatch - see above the function for details. TK, Jan 2012
214 214 # Monkeypatch inspect to apply our bugfix. This code only works with py25
215 215 #if sys.version_info[:2] >= (2,5):
216 216 # inspect.findsource = findsource
217 217
218 218 def fix_frame_records_filenames(records):
219 219 """Try to fix the filenames in each record from inspect.getinnerframes().
220 220
221 221 Particularly, modules loaded from within zip files have useless filenames
222 222 attached to their code object, and inspect.getinnerframes() just uses it.
223 223 """
224 224 fixed_records = []
225 225 for frame, filename, line_no, func_name, lines, index in records:
226 226 # Look inside the frame's globals dictionary for __file__, which should
227 227 # be better.
228 228 better_fn = frame.f_globals.get('__file__', None)
229 229 if isinstance(better_fn, str):
230 230 # Check the type just in case someone did something weird with
231 231 # __file__. It might also be None if the error occurred during
232 232 # import.
233 233 filename = better_fn
234 234 fixed_records.append((frame, filename, line_no, func_name, lines, index))
235 235 return fixed_records
236 236
237 237
238 238 def _fixed_getinnerframes(etb, context=1,tb_offset=0):
239 239 import linecache
240 240 LNUM_POS, LINES_POS, INDEX_POS = 2, 4, 5
241 241
242 242 records = fix_frame_records_filenames(inspect.getinnerframes(etb, context))
243 243
244 244 # If the error is at the console, don't build any context, since it would
245 245 # otherwise produce 5 blank lines printed out (there is no file at the
246 246 # console)
247 247 rec_check = records[tb_offset:]
248 248 try:
249 249 rname = rec_check[0][1]
250 250 if rname == '<ipython console>' or rname.endswith('<string>'):
251 251 return rec_check
252 252 except IndexError:
253 253 pass
254 254
255 255 aux = traceback.extract_tb(etb)
256 256 assert len(records) == len(aux)
257 257 for i, (file, lnum, _, _) in zip(range(len(records)), aux):
258 258 maybeStart = lnum-1 - context//2
259 259 start = max(maybeStart, 0)
260 260 end = start + context
261 261 lines = linecache.getlines(file)[start:end]
262 262 buf = list(records[i])
263 263 buf[LNUM_POS] = lnum
264 264 buf[INDEX_POS] = lnum - 1 - start
265 265 buf[LINES_POS] = lines
266 266 records[i] = tuple(buf)
267 267 return records[tb_offset:]
268 268
269 269 # Helper function -- largely belongs to VerboseTB, but we need the same
270 270 # functionality to produce a pseudo verbose TB for SyntaxErrors, so that they
271 271 # can be recognized properly by ipython.el's py-traceback-line-re
272 272 # (SyntaxErrors have to be treated specially because they have no traceback)
273 273
274 274 _parser = PyColorize.Parser()
275 275
276 276 def _format_traceback_lines(lnum, index, lines, Colors, lvals=None,scheme=None):
277 277 numbers_width = INDENT_SIZE - 1
278 278 res = []
279 279 i = lnum - index
280 280
281 281 # This lets us get fully syntax-highlighted tracebacks.
282 282 if scheme is None:
283 283 ipinst = ipapi.get()
284 284 if ipinst is not None:
285 285 scheme = ipinst.colors
286 286 else:
287 287 scheme = DEFAULT_SCHEME
288 288
289 289 _line_format = _parser.format2
290 290
291 291 for line in lines:
292 292 # FIXME: we need to ensure the source is a pure string at this point,
293 293 # else the coloring code makes a royal mess. This is in need of a
294 294 # serious refactoring, so that all of the ultratb and PyColorize code
295 295 # is unicode-safe. So for now this is rather an ugly hack, but
296 296 # necessary to at least have readable tracebacks. Improvements welcome!
297 297 line = py3compat.cast_bytes_py2(line, 'utf-8')
298 298
299 299 new_line, err = _line_format(line, 'str', scheme)
300 300 if not err: line = new_line
301 301
302 302 if i == lnum:
303 303 # This is the line with the error
304 304 pad = numbers_width - len(str(i))
305 305 if pad >= 3:
306 306 marker = '-'*(pad-3) + '-> '
307 307 elif pad == 2:
308 308 marker = '> '
309 309 elif pad == 1:
310 310 marker = '>'
311 311 else:
312 312 marker = ''
313 313 num = marker + str(i)
314 314 line = '%s%s%s %s%s' %(Colors.linenoEm, num,
315 315 Colors.line, line, Colors.Normal)
316 316 else:
317 317 num = '%*s' % (numbers_width,i)
318 318 line = '%s%s%s %s' %(Colors.lineno, num,
319 319 Colors.Normal, line)
320 320
321 321 res.append(line)
322 322 if lvals and i == lnum:
323 323 res.append(lvals + '\n')
324 324 i = i + 1
325 325 return res
326 326
327 327
328 328 #---------------------------------------------------------------------------
329 329 # Module classes
330 330 class TBTools(object):
331 331 """Basic tools used by all traceback printer classes."""
332 332
333 333 # Number of frames to skip when reporting tracebacks
334 334 tb_offset = 0
335 335
336 336 def __init__(self, color_scheme='NoColor', call_pdb=False, ostream=None):
337 337 # Whether to call the interactive pdb debugger after printing
338 338 # tracebacks or not
339 339 self.call_pdb = call_pdb
340 340
341 341 # Output stream to write to. Note that we store the original value in
342 342 # a private attribute and then make the public ostream a property, so
343 343 # that we can delay accessing io.stdout until runtime. The way
344 344 # things are written now, the io.stdout object is dynamically managed
345 345 # so a reference to it should NEVER be stored statically. This
346 346 # property approach confines this detail to a single location, and all
347 347 # subclasses can simply access self.ostream for writing.
348 348 self._ostream = ostream
349 349
350 350 # Create color table
351 351 self.color_scheme_table = exception_colors()
352 352
353 353 self.set_colors(color_scheme)
354 354 self.old_scheme = color_scheme # save initial value for toggles
355 355
356 356 if call_pdb:
357 357 self.pdb = debugger.Pdb(self.color_scheme_table.active_scheme_name)
358 358 else:
359 359 self.pdb = None
360 360
361 361 def _get_ostream(self):
362 362 """Output stream that exceptions are written to.
363 363
364 364 Valid values are:
365 365
366 366 - None: the default, which means that IPython will dynamically resolve
367 367 to io.stdout. This ensures compatibility with most tools, including
368 368 Windows (where plain stdout doesn't recognize ANSI escapes).
369 369
370 370 - Any object with 'write' and 'flush' attributes.
371 371 """
372 372 return io.stdout if self._ostream is None else self._ostream
373 373
374 374 def _set_ostream(self, val):
375 375 assert val is None or (hasattr(val, 'write') and hasattr(val, 'flush'))
376 376 self._ostream = val
377 377
378 378 ostream = property(_get_ostream, _set_ostream)
379 379
380 380 def set_colors(self,*args,**kw):
381 381 """Shorthand access to the color table scheme selector method."""
382 382
383 383 # Set own color table
384 384 self.color_scheme_table.set_active_scheme(*args,**kw)
385 385 # for convenience, set Colors to the active scheme
386 386 self.Colors = self.color_scheme_table.active_colors
387 387 # Also set colors of debugger
388 388 if hasattr(self,'pdb') and self.pdb is not None:
389 389 self.pdb.set_colors(*args,**kw)
390 390
391 391 def color_toggle(self):
392 392 """Toggle between the currently active color scheme and NoColor."""
393 393
394 394 if self.color_scheme_table.active_scheme_name == 'NoColor':
395 395 self.color_scheme_table.set_active_scheme(self.old_scheme)
396 396 self.Colors = self.color_scheme_table.active_colors
397 397 else:
398 398 self.old_scheme = self.color_scheme_table.active_scheme_name
399 399 self.color_scheme_table.set_active_scheme('NoColor')
400 400 self.Colors = self.color_scheme_table.active_colors
401 401
402 402 def stb2text(self, stb):
403 403 """Convert a structured traceback (a list) to a string."""
404 404 return '\n'.join(stb)
405 405
406 406 def text(self, etype, value, tb, tb_offset=None, context=5):
407 407 """Return formatted traceback.
408 408
409 409 Subclasses may override this if they add extra arguments.
410 410 """
411 411 tb_list = self.structured_traceback(etype, value, tb,
412 412 tb_offset, context)
413 413 return self.stb2text(tb_list)
414 414
415 415 def structured_traceback(self, etype, evalue, tb, tb_offset=None,
416 416 context=5, mode=None):
417 417 """Return a list of traceback frames.
418 418
419 419 Must be implemented by each class.
420 420 """
421 421 raise NotImplementedError()
422 422
423 423
424 424 #---------------------------------------------------------------------------
425 425 class ListTB(TBTools):
426 426 """Print traceback information from a traceback list, with optional color.
427 427
428 428 Calling: requires 3 arguments:
429 429 (etype, evalue, elist)
430 430 as would be obtained by:
431 431 etype, evalue, tb = sys.exc_info()
432 432 if tb:
433 433 elist = traceback.extract_tb(tb)
434 434 else:
435 435 elist = None
436 436
437 437 It can thus be used by programs which need to process the traceback before
438 438 printing (such as console replacements based on the code module from the
439 439 standard library).
440 440
441 441 Because they are meant to be called without a full traceback (only a
442 442 list), instances of this class can't call the interactive pdb debugger."""
443 443
444 444 def __init__(self,color_scheme = 'NoColor', call_pdb=False, ostream=None):
445 445 TBTools.__init__(self, color_scheme=color_scheme, call_pdb=call_pdb,
446 446 ostream=ostream)
447 447
448 448 def __call__(self, etype, value, elist):
449 449 self.ostream.flush()
450 450 self.ostream.write(self.text(etype, value, elist))
451 451 self.ostream.write('\n')
452 452
453 453 def structured_traceback(self, etype, value, elist, tb_offset=None,
454 454 context=5):
455 455 """Return a color formatted string with the traceback info.
456 456
457 457 Parameters
458 458 ----------
459 459 etype : exception type
460 460 Type of the exception raised.
461 461
462 462 value : object
463 463 Data stored in the exception
464 464
465 465 elist : list
466 466 List of frames, see class docstring for details.
467 467
468 468 tb_offset : int, optional
469 469 Number of frames in the traceback to skip. If not given, the
470 470 instance value is used (set in constructor).
471 471
472 472 context : int, optional
473 473 Number of lines of context information to print.
474 474
475 475 Returns
476 476 -------
477 477 String with formatted exception.
478 478 """
479 479 tb_offset = self.tb_offset if tb_offset is None else tb_offset
480 480 Colors = self.Colors
481 481 out_list = []
482 482 if elist:
483 483
484 484 if tb_offset and len(elist) > tb_offset:
485 485 elist = elist[tb_offset:]
486 486
487 487 out_list.append('Traceback %s(most recent call last)%s:' %
488 488 (Colors.normalEm, Colors.Normal) + '\n')
489 489 out_list.extend(self._format_list(elist))
490 490 # The exception info should be a single entry in the list.
491 491 lines = ''.join(self._format_exception_only(etype, value))
492 492 out_list.append(lines)
493 493
494 494 # Note: this code originally read:
495 495
496 496 ## for line in lines[:-1]:
497 497 ## out_list.append(" "+line)
498 498 ## out_list.append(lines[-1])
499 499
500 500 # This means it was indenting everything but the last line by a little
501 501 # bit. I've disabled this for now, but if we see ugliness somewhre we
502 502 # can restore it.
503 503
504 504 return out_list
505 505
506 506 def _format_list(self, extracted_list):
507 507 """Format a list of traceback entry tuples for printing.
508 508
509 509 Given a list of tuples as returned by extract_tb() or
510 510 extract_stack(), return a list of strings ready for printing.
511 511 Each string in the resulting list corresponds to the item with the
512 512 same index in the argument list. Each string ends in a newline;
513 513 the strings may contain internal newlines as well, for those items
514 514 whose source text line is not None.
515 515
516 516 Lifted almost verbatim from traceback.py
517 517 """
518 518
519 519 Colors = self.Colors
520 520 list = []
521 521 for filename, lineno, name, line in extracted_list[:-1]:
522 522 item = ' File %s"%s"%s, line %s%d%s, in %s%s%s\n' % \
523 523 (Colors.filename, filename, Colors.Normal,
524 524 Colors.lineno, lineno, Colors.Normal,
525 525 Colors.name, name, Colors.Normal)
526 526 if line:
527 527 item += ' %s\n' % line.strip()
528 528 list.append(item)
529 529 # Emphasize the last entry
530 530 filename, lineno, name, line = extracted_list[-1]
531 531 item = '%s File %s"%s"%s, line %s%d%s, in %s%s%s%s\n' % \
532 532 (Colors.normalEm,
533 533 Colors.filenameEm, filename, Colors.normalEm,
534 534 Colors.linenoEm, lineno, Colors.normalEm,
535 535 Colors.nameEm, name, Colors.normalEm,
536 536 Colors.Normal)
537 537 if line:
538 538 item += '%s %s%s\n' % (Colors.line, line.strip(),
539 539 Colors.Normal)
540 540 list.append(item)
541 541 #from pprint import pformat; print 'LISTTB', pformat(list) # dbg
542 542 return list
543 543
544 544 def _format_exception_only(self, etype, value):
545 545 """Format the exception part of a traceback.
546 546
547 547 The arguments are the exception type and value such as given by
548 548 sys.exc_info()[:2]. The return value is a list of strings, each ending
549 549 in a newline. Normally, the list contains a single string; however,
550 550 for SyntaxError exceptions, it contains several lines that (when
551 551 printed) display detailed information about where the syntax error
552 552 occurred. The message indicating which exception occurred is the
553 553 always last string in the list.
554 554
555 555 Also lifted nearly verbatim from traceback.py
556 556 """
557 557
558 558 have_filedata = False
559 559 Colors = self.Colors
560 560 list = []
561 561 stype = Colors.excName + etype.__name__ + Colors.Normal
562 562 if value is None:
563 563 # Not sure if this can still happen in Python 2.6 and above
564 564 list.append( str(stype) + '\n')
565 565 else:
566 566 if etype is SyntaxError:
567 567 have_filedata = True
568 568 #print 'filename is',filename # dbg
569 569 if not value.filename: value.filename = "<string>"
570 570 list.append('%s File %s"%s"%s, line %s%d%s\n' % \
571 571 (Colors.normalEm,
572 572 Colors.filenameEm, value.filename, Colors.normalEm,
573 573 Colors.linenoEm, value.lineno, Colors.Normal ))
574 574 if value.text is not None:
575 575 i = 0
576 576 while i < len(value.text) and value.text[i].isspace():
577 577 i += 1
578 578 list.append('%s %s%s\n' % (Colors.line,
579 579 value.text.strip(),
580 580 Colors.Normal))
581 581 if value.offset is not None:
582 582 s = ' '
583 583 for c in value.text[i:value.offset-1]:
584 584 if c.isspace():
585 585 s += c
586 586 else:
587 587 s += ' '
588 588 list.append('%s%s^%s\n' % (Colors.caret, s,
589 589 Colors.Normal) )
590 590
591 591 try:
592 592 s = value.msg
593 593 except Exception:
594 594 s = self._some_str(value)
595 595 if s:
596 596 list.append('%s%s:%s %s\n' % (str(stype), Colors.excName,
597 597 Colors.Normal, s))
598 598 else:
599 599 list.append('%s\n' % str(stype))
600 600
601 601 # sync with user hooks
602 602 if have_filedata:
603 603 ipinst = ipapi.get()
604 604 if ipinst is not None:
605 605 ipinst.hooks.synchronize_with_editor(value.filename, value.lineno, 0)
606 606
607 607 return list
608 608
609 609 def get_exception_only(self, etype, value):
610 610 """Only print the exception type and message, without a traceback.
611 611
612 612 Parameters
613 613 ----------
614 614 etype : exception type
615 615 value : exception value
616 616 """
617 617 return ListTB.structured_traceback(self, etype, value, [])
618 618
619 619
620 620 def show_exception_only(self, etype, evalue):
621 621 """Only print the exception type and message, without a traceback.
622 622
623 623 Parameters
624 624 ----------
625 625 etype : exception type
626 626 value : exception value
627 627 """
628 628 # This method needs to use __call__ from *this* class, not the one from
629 629 # a subclass whose signature or behavior may be different
630 630 ostream = self.ostream
631 631 ostream.flush()
632 632 ostream.write('\n'.join(self.get_exception_only(etype, evalue)))
633 633 ostream.flush()
634 634
635 635 def _some_str(self, value):
636 636 # Lifted from traceback.py
637 637 try:
638 638 return str(value)
639 639 except:
640 640 return '<unprintable %s object>' % type(value).__name__
641 641
642 642 #----------------------------------------------------------------------------
643 643 class VerboseTB(TBTools):
644 644 """A port of Ka-Ping Yee's cgitb.py module that outputs color text instead
645 645 of HTML. Requires inspect and pydoc. Crazy, man.
646 646
647 647 Modified version which optionally strips the topmost entries from the
648 648 traceback, to be used with alternate interpreters (because their own code
649 649 would appear in the traceback)."""
650 650
651 651 def __init__(self,color_scheme = 'Linux', call_pdb=False, ostream=None,
652 652 tb_offset=0, long_header=False, include_vars=True,
653 653 check_cache=None):
654 654 """Specify traceback offset, headers and color scheme.
655 655
656 656 Define how many frames to drop from the tracebacks. Calling it with
657 657 tb_offset=1 allows use of this handler in interpreters which will have
658 658 their own code at the top of the traceback (VerboseTB will first
659 659 remove that frame before printing the traceback info)."""
660 660 TBTools.__init__(self, color_scheme=color_scheme, call_pdb=call_pdb,
661 661 ostream=ostream)
662 662 self.tb_offset = tb_offset
663 663 self.long_header = long_header
664 664 self.include_vars = include_vars
665 665 # By default we use linecache.checkcache, but the user can provide a
666 666 # different check_cache implementation. This is used by the IPython
667 667 # kernel to provide tracebacks for interactive code that is cached,
668 668 # by a compiler instance that flushes the linecache but preserves its
669 669 # own code cache.
670 670 if check_cache is None:
671 671 check_cache = linecache.checkcache
672 672 self.check_cache = check_cache
673 673
674 674 def structured_traceback(self, etype, evalue, etb, tb_offset=None,
675 675 context=5):
676 676 """Return a nice text document describing the traceback."""
677 677
678 678 tb_offset = self.tb_offset if tb_offset is None else tb_offset
679 679
680 680 # some locals
681 681 try:
682 682 etype = etype.__name__
683 683 except AttributeError:
684 684 pass
685 685 Colors = self.Colors # just a shorthand + quicker name lookup
686 686 ColorsNormal = Colors.Normal # used a lot
687 687 col_scheme = self.color_scheme_table.active_scheme_name
688 688 indent = ' '*INDENT_SIZE
689 689 em_normal = '%s\n%s%s' % (Colors.valEm, indent,ColorsNormal)
690 690 undefined = '%sundefined%s' % (Colors.em, ColorsNormal)
691 691 exc = '%s%s%s' % (Colors.excName,etype,ColorsNormal)
692 692
693 693 # some internal-use functions
694 694 def text_repr(value):
695 695 """Hopefully pretty robust repr equivalent."""
696 696 # this is pretty horrible but should always return *something*
697 697 try:
698 698 return pydoc.text.repr(value)
699 699 except KeyboardInterrupt:
700 700 raise
701 701 except:
702 702 try:
703 703 return repr(value)
704 704 except KeyboardInterrupt:
705 705 raise
706 706 except:
707 707 try:
708 708 # all still in an except block so we catch
709 709 # getattr raising
710 710 name = getattr(value, '__name__', None)
711 711 if name:
712 712 # ick, recursion
713 713 return text_repr(name)
714 714 klass = getattr(value, '__class__', None)
715 715 if klass:
716 716 return '%s instance' % text_repr(klass)
717 717 except KeyboardInterrupt:
718 718 raise
719 719 except:
720 720 return 'UNRECOVERABLE REPR FAILURE'
721 721 def eqrepr(value, repr=text_repr): return '=%s' % repr(value)
722 722 def nullrepr(value, repr=text_repr): return ''
723 723
724 724 # meat of the code begins
725 725 try:
726 726 etype = etype.__name__
727 727 except AttributeError:
728 728 pass
729 729
730 730 if self.long_header:
731 731 # Header with the exception type, python version, and date
732 732 pyver = 'Python ' + sys.version.split()[0] + ': ' + sys.executable
733 733 date = time.ctime(time.time())
734 734
735 735 head = '%s%s%s\n%s%s%s\n%s' % (Colors.topline, '-'*75, ColorsNormal,
736 736 exc, ' '*(75-len(str(etype))-len(pyver)),
737 737 pyver, date.rjust(75) )
738 738 head += "\nA problem occured executing Python code. Here is the sequence of function"\
739 739 "\ncalls leading up to the error, with the most recent (innermost) call last."
740 740 else:
741 741 # Simplified header
742 742 head = '%s%s%s\n%s%s' % (Colors.topline, '-'*75, ColorsNormal,exc,
743 743 'Traceback (most recent call last)'.\
744 744 rjust(75 - len(str(etype)) ) )
745 745 frames = []
746 746 # Flush cache before calling inspect. This helps alleviate some of the
747 747 # problems with python 2.3's inspect.py.
748 748 ##self.check_cache()
749 749 # Drop topmost frames if requested
750 750 try:
751 751 # Try the default getinnerframes and Alex's: Alex's fixes some
752 752 # problems, but it generates empty tracebacks for console errors
753 753 # (5 blanks lines) where none should be returned.
754 754 #records = inspect.getinnerframes(etb, context)[tb_offset:]
755 755 #print 'python records:', records # dbg
756 756 records = _fixed_getinnerframes(etb, context, tb_offset)
757 757 #print 'alex records:', records # dbg
758 758 except:
759 759
760 760 # FIXME: I've been getting many crash reports from python 2.3
761 761 # users, traceable to inspect.py. If I can find a small test-case
762 762 # to reproduce this, I should either write a better workaround or
763 763 # file a bug report against inspect (if that's the real problem).
764 764 # So far, I haven't been able to find an isolated example to
765 765 # reproduce the problem.
766 766 inspect_error()
767 767 traceback.print_exc(file=self.ostream)
768 768 info('\nUnfortunately, your original traceback can not be constructed.\n')
769 769 return ''
770 770
771 771 # build some color string templates outside these nested loops
772 772 tpl_link = '%s%%s%s' % (Colors.filenameEm,ColorsNormal)
773 773 tpl_call = 'in %s%%s%s%%s%s' % (Colors.vName, Colors.valEm,
774 774 ColorsNormal)
775 775 tpl_call_fail = 'in %s%%s%s(***failed resolving arguments***)%s' % \
776 776 (Colors.vName, Colors.valEm, ColorsNormal)
777 777 tpl_local_var = '%s%%s%s' % (Colors.vName, ColorsNormal)
778 778 tpl_global_var = '%sglobal%s %s%%s%s' % (Colors.em, ColorsNormal,
779 779 Colors.vName, ColorsNormal)
780 780 tpl_name_val = '%%s %s= %%s%s' % (Colors.valEm, ColorsNormal)
781 781 tpl_line = '%s%%s%s %%s' % (Colors.lineno, ColorsNormal)
782 782 tpl_line_em = '%s%%s%s %%s%s' % (Colors.linenoEm,Colors.line,
783 783 ColorsNormal)
784 784
785 785 # now, loop over all records printing context and info
786 786 abspath = os.path.abspath
787 787 for frame, file, lnum, func, lines, index in records:
788 788 #print '*** record:',file,lnum,func,lines,index # dbg
789 789 try:
790 790 file = file and abspath(file) or '?'
791 791 except OSError:
792 792 # if file is '<console>' or something not in the filesystem,
793 793 # the abspath call will throw an OSError. Just ignore it and
794 794 # keep the original file string.
795 795 pass
796 796 link = tpl_link % file
797 797 args, varargs, varkw, locals = inspect.getargvalues(frame)
798 798
799 799 if func == '?':
800 800 call = ''
801 801 else:
802 802 # Decide whether to include variable details or not
803 803 var_repr = self.include_vars and eqrepr or nullrepr
804 804 try:
805 805 call = tpl_call % (func,inspect.formatargvalues(args,
806 806 varargs, varkw,
807 807 locals,formatvalue=var_repr))
808 808 except KeyError:
809 809 # This happens in situations like errors inside generator
810 810 # expressions, where local variables are listed in the
811 811 # line, but can't be extracted from the frame. I'm not
812 812 # 100% sure this isn't actually a bug in inspect itself,
813 813 # but since there's no info for us to compute with, the
814 814 # best we can do is report the failure and move on. Here
815 815 # we must *not* call any traceback construction again,
816 816 # because that would mess up use of %debug later on. So we
817 817 # simply report the failure and move on. The only
818 818 # limitation will be that this frame won't have locals
819 819 # listed in the call signature. Quite subtle problem...
820 820 # I can't think of a good way to validate this in a unit
821 821 # test, but running a script consisting of:
822 822 # dict( (k,v.strip()) for (k,v) in range(10) )
823 823 # will illustrate the error, if this exception catch is
824 824 # disabled.
825 825 call = tpl_call_fail % func
826
827 # Don't attempt to tokenize binary files.
828 if file.endswith(('.so', '.pyd', '.dll')):
829 frames.append('%s %s\n' % (link,call))
830 continue
826 831
827 832 def linereader(file=file, lnum=[lnum], getline=linecache.getline):
828 833 if file.endswith(('.pyc','.pyo')):
829 834 file = pyfile.source_from_cache(file)
830 835 line = getline(file, lnum[0])
831 836 lnum[0] += 1
832 837 return line
833 838
834 839 # Build the list of names on this line of code where the exception
835 840 # occurred.
836 841 try:
837 842 names = []
838 843 name_cont = False
839 844
840 845 for token_type, token, start, end, line in generate_tokens(linereader):
841 846 # build composite names
842 847 if token_type == tokenize.NAME and token not in keyword.kwlist:
843 848 if name_cont:
844 849 # Continuation of a dotted name
845 850 try:
846 851 names[-1].append(token)
847 852 except IndexError:
848 853 names.append([token])
849 854 name_cont = False
850 855 else:
851 856 # Regular new names. We append everything, the caller
852 857 # will be responsible for pruning the list later. It's
853 858 # very tricky to try to prune as we go, b/c composite
854 859 # names can fool us. The pruning at the end is easy
855 860 # to do (or the caller can print a list with repeated
856 861 # names if so desired.
857 862 names.append([token])
858 863 elif token == '.':
859 864 name_cont = True
860 865 elif token_type == tokenize.NEWLINE:
861 866 break
862 867
863 868 except (IndexError, UnicodeDecodeError):
864 869 # signals exit of tokenizer
865 870 pass
866 871 except tokenize.TokenError,msg:
867 872 _m = ("An unexpected error occurred while tokenizing input\n"
868 873 "The following traceback may be corrupted or invalid\n"
869 874 "The error message is: %s\n" % msg)
870 875 error(_m)
871 876
872 877 # Join composite names (e.g. "dict.fromkeys")
873 878 names = ['.'.join(n) for n in names]
874 879 # prune names list of duplicates, but keep the right order
875 880 unique_names = uniq_stable(names)
876 881
877 882 # Start loop over vars
878 883 lvals = []
879 884 if self.include_vars:
880 885 for name_full in unique_names:
881 886 name_base = name_full.split('.',1)[0]
882 887 if name_base in frame.f_code.co_varnames:
883 888 if locals.has_key(name_base):
884 889 try:
885 890 value = repr(eval(name_full,locals))
886 891 except:
887 892 value = undefined
888 893 else:
889 894 value = undefined
890 895 name = tpl_local_var % name_full
891 896 else:
892 897 if frame.f_globals.has_key(name_base):
893 898 try:
894 899 value = repr(eval(name_full,frame.f_globals))
895 900 except:
896 901 value = undefined
897 902 else:
898 903 value = undefined
899 904 name = tpl_global_var % name_full
900 905 lvals.append(tpl_name_val % (name,value))
901 906 if lvals:
902 907 lvals = '%s%s' % (indent,em_normal.join(lvals))
903 908 else:
904 909 lvals = ''
905 910
906 911 level = '%s %s\n' % (link,call)
907 912
908 913 if index is None:
909 914 frames.append(level)
910 915 else:
911 916 frames.append('%s%s' % (level,''.join(
912 917 _format_traceback_lines(lnum,index,lines,Colors,lvals,
913 918 col_scheme))))
914 919
915 920 # Get (safely) a string form of the exception info
916 921 try:
917 922 etype_str,evalue_str = map(str,(etype,evalue))
918 923 except:
919 924 # User exception is improperly defined.
920 925 etype,evalue = str,sys.exc_info()[:2]
921 926 etype_str,evalue_str = map(str,(etype,evalue))
922 927 # ... and format it
923 928 exception = ['%s%s%s: %s' % (Colors.excName, etype_str,
924 929 ColorsNormal, evalue_str)]
925 930 if (not py3compat.PY3) and type(evalue) is types.InstanceType:
926 931 try:
927 932 names = [w for w in dir(evalue) if isinstance(w, basestring)]
928 933 except:
929 934 # Every now and then, an object with funny inernals blows up
930 935 # when dir() is called on it. We do the best we can to report
931 936 # the problem and continue
932 937 _m = '%sException reporting error (object with broken dir())%s:'
933 938 exception.append(_m % (Colors.excName,ColorsNormal))
934 939 etype_str,evalue_str = map(str,sys.exc_info()[:2])
935 940 exception.append('%s%s%s: %s' % (Colors.excName,etype_str,
936 941 ColorsNormal, evalue_str))
937 942 names = []
938 943 for name in names:
939 944 value = text_repr(getattr(evalue, name))
940 945 exception.append('\n%s%s = %s' % (indent, name, value))
941 946
942 947 # vds: >>
943 948 if records:
944 949 filepath, lnum = records[-1][1:3]
945 950 #print "file:", str(file), "linenb", str(lnum) # dbg
946 951 filepath = os.path.abspath(filepath)
947 952 ipinst = ipapi.get()
948 953 if ipinst is not None:
949 954 ipinst.hooks.synchronize_with_editor(filepath, lnum, 0)
950 955 # vds: <<
951 956
952 957 # return all our info assembled as a single string
953 958 # return '%s\n\n%s\n%s' % (head,'\n'.join(frames),''.join(exception[0]) )
954 959 return [head] + frames + [''.join(exception[0])]
955 960
956 961 def debugger(self,force=False):
957 962 """Call up the pdb debugger if desired, always clean up the tb
958 963 reference.
959 964
960 965 Keywords:
961 966
962 967 - force(False): by default, this routine checks the instance call_pdb
963 968 flag and does not actually invoke the debugger if the flag is false.
964 969 The 'force' option forces the debugger to activate even if the flag
965 970 is false.
966 971
967 972 If the call_pdb flag is set, the pdb interactive debugger is
968 973 invoked. In all cases, the self.tb reference to the current traceback
969 974 is deleted to prevent lingering references which hamper memory
970 975 management.
971 976
972 977 Note that each call to pdb() does an 'import readline', so if your app
973 978 requires a special setup for the readline completers, you'll have to
974 979 fix that by hand after invoking the exception handler."""
975 980
976 981 if force or self.call_pdb:
977 982 if self.pdb is None:
978 983 self.pdb = debugger.Pdb(
979 984 self.color_scheme_table.active_scheme_name)
980 985 # the system displayhook may have changed, restore the original
981 986 # for pdb
982 987 display_trap = DisplayTrap(hook=sys.__displayhook__)
983 988 with display_trap:
984 989 self.pdb.reset()
985 990 # Find the right frame so we don't pop up inside ipython itself
986 991 if hasattr(self,'tb') and self.tb is not None:
987 992 etb = self.tb
988 993 else:
989 994 etb = self.tb = sys.last_traceback
990 995 while self.tb is not None and self.tb.tb_next is not None:
991 996 self.tb = self.tb.tb_next
992 997 if etb and etb.tb_next:
993 998 etb = etb.tb_next
994 999 self.pdb.botframe = etb.tb_frame
995 1000 self.pdb.interaction(self.tb.tb_frame, self.tb)
996 1001
997 1002 if hasattr(self,'tb'):
998 1003 del self.tb
999 1004
1000 1005 def handler(self, info=None):
1001 1006 (etype, evalue, etb) = info or sys.exc_info()
1002 1007 self.tb = etb
1003 1008 ostream = self.ostream
1004 1009 ostream.flush()
1005 1010 ostream.write(self.text(etype, evalue, etb))
1006 1011 ostream.write('\n')
1007 1012 ostream.flush()
1008 1013
1009 1014 # Changed so an instance can just be called as VerboseTB_inst() and print
1010 1015 # out the right info on its own.
1011 1016 def __call__(self, etype=None, evalue=None, etb=None):
1012 1017 """This hook can replace sys.excepthook (for Python 2.1 or higher)."""
1013 1018 if etb is None:
1014 1019 self.handler()
1015 1020 else:
1016 1021 self.handler((etype, evalue, etb))
1017 1022 try:
1018 1023 self.debugger()
1019 1024 except KeyboardInterrupt:
1020 1025 print "\nKeyboardInterrupt"
1021 1026
1022 1027 #----------------------------------------------------------------------------
1023 1028 class FormattedTB(VerboseTB, ListTB):
1024 1029 """Subclass ListTB but allow calling with a traceback.
1025 1030
1026 1031 It can thus be used as a sys.excepthook for Python > 2.1.
1027 1032
1028 1033 Also adds 'Context' and 'Verbose' modes, not available in ListTB.
1029 1034
1030 1035 Allows a tb_offset to be specified. This is useful for situations where
1031 1036 one needs to remove a number of topmost frames from the traceback (such as
1032 1037 occurs with python programs that themselves execute other python code,
1033 1038 like Python shells). """
1034 1039
1035 1040 def __init__(self, mode='Plain', color_scheme='Linux', call_pdb=False,
1036 1041 ostream=None,
1037 1042 tb_offset=0, long_header=False, include_vars=False,
1038 1043 check_cache=None):
1039 1044
1040 1045 # NEVER change the order of this list. Put new modes at the end:
1041 1046 self.valid_modes = ['Plain','Context','Verbose']
1042 1047 self.verbose_modes = self.valid_modes[1:3]
1043 1048
1044 1049 VerboseTB.__init__(self, color_scheme=color_scheme, call_pdb=call_pdb,
1045 1050 ostream=ostream, tb_offset=tb_offset,
1046 1051 long_header=long_header, include_vars=include_vars,
1047 1052 check_cache=check_cache)
1048 1053
1049 1054 # Different types of tracebacks are joined with different separators to
1050 1055 # form a single string. They are taken from this dict
1051 1056 self._join_chars = dict(Plain='', Context='\n', Verbose='\n')
1052 1057 # set_mode also sets the tb_join_char attribute
1053 1058 self.set_mode(mode)
1054 1059
1055 1060 def _extract_tb(self,tb):
1056 1061 if tb:
1057 1062 return traceback.extract_tb(tb)
1058 1063 else:
1059 1064 return None
1060 1065
1061 1066 def structured_traceback(self, etype, value, tb, tb_offset=None, context=5):
1062 1067 tb_offset = self.tb_offset if tb_offset is None else tb_offset
1063 1068 mode = self.mode
1064 1069 if mode in self.verbose_modes:
1065 1070 # Verbose modes need a full traceback
1066 1071 return VerboseTB.structured_traceback(
1067 1072 self, etype, value, tb, tb_offset, context
1068 1073 )
1069 1074 else:
1070 1075 # We must check the source cache because otherwise we can print
1071 1076 # out-of-date source code.
1072 1077 self.check_cache()
1073 1078 # Now we can extract and format the exception
1074 1079 elist = self._extract_tb(tb)
1075 1080 return ListTB.structured_traceback(
1076 1081 self, etype, value, elist, tb_offset, context
1077 1082 )
1078 1083
1079 1084 def stb2text(self, stb):
1080 1085 """Convert a structured traceback (a list) to a string."""
1081 1086 return self.tb_join_char.join(stb)
1082 1087
1083 1088
1084 1089 def set_mode(self,mode=None):
1085 1090 """Switch to the desired mode.
1086 1091
1087 1092 If mode is not specified, cycles through the available modes."""
1088 1093
1089 1094 if not mode:
1090 1095 new_idx = ( self.valid_modes.index(self.mode) + 1 ) % \
1091 1096 len(self.valid_modes)
1092 1097 self.mode = self.valid_modes[new_idx]
1093 1098 elif mode not in self.valid_modes:
1094 1099 raise ValueError, 'Unrecognized mode in FormattedTB: <'+mode+'>\n'\
1095 1100 'Valid modes: '+str(self.valid_modes)
1096 1101 else:
1097 1102 self.mode = mode
1098 1103 # include variable details only in 'Verbose' mode
1099 1104 self.include_vars = (self.mode == self.valid_modes[2])
1100 1105 # Set the join character for generating text tracebacks
1101 1106 self.tb_join_char = self._join_chars[self.mode]
1102 1107
1103 1108 # some convenient shorcuts
1104 1109 def plain(self):
1105 1110 self.set_mode(self.valid_modes[0])
1106 1111
1107 1112 def context(self):
1108 1113 self.set_mode(self.valid_modes[1])
1109 1114
1110 1115 def verbose(self):
1111 1116 self.set_mode(self.valid_modes[2])
1112 1117
1113 1118 #----------------------------------------------------------------------------
1114 1119 class AutoFormattedTB(FormattedTB):
1115 1120 """A traceback printer which can be called on the fly.
1116 1121
1117 1122 It will find out about exceptions by itself.
1118 1123
1119 1124 A brief example:
1120 1125
1121 1126 AutoTB = AutoFormattedTB(mode = 'Verbose',color_scheme='Linux')
1122 1127 try:
1123 1128 ...
1124 1129 except:
1125 1130 AutoTB() # or AutoTB(out=logfile) where logfile is an open file object
1126 1131 """
1127 1132
1128 1133 def __call__(self,etype=None,evalue=None,etb=None,
1129 1134 out=None,tb_offset=None):
1130 1135 """Print out a formatted exception traceback.
1131 1136
1132 1137 Optional arguments:
1133 1138 - out: an open file-like object to direct output to.
1134 1139
1135 1140 - tb_offset: the number of frames to skip over in the stack, on a
1136 1141 per-call basis (this overrides temporarily the instance's tb_offset
1137 1142 given at initialization time. """
1138 1143
1139 1144
1140 1145 if out is None:
1141 1146 out = self.ostream
1142 1147 out.flush()
1143 1148 out.write(self.text(etype, evalue, etb, tb_offset))
1144 1149 out.write('\n')
1145 1150 out.flush()
1146 1151 # FIXME: we should remove the auto pdb behavior from here and leave
1147 1152 # that to the clients.
1148 1153 try:
1149 1154 self.debugger()
1150 1155 except KeyboardInterrupt:
1151 1156 print "\nKeyboardInterrupt"
1152 1157
1153 1158 def structured_traceback(self, etype=None, value=None, tb=None,
1154 1159 tb_offset=None, context=5):
1155 1160 if etype is None:
1156 1161 etype,value,tb = sys.exc_info()
1157 1162 self.tb = tb
1158 1163 return FormattedTB.structured_traceback(
1159 1164 self, etype, value, tb, tb_offset, context)
1160 1165
1161 1166 #---------------------------------------------------------------------------
1162 1167
1163 1168 # A simple class to preserve Nathan's original functionality.
1164 1169 class ColorTB(FormattedTB):
1165 1170 """Shorthand to initialize a FormattedTB in Linux colors mode."""
1166 1171 def __init__(self,color_scheme='Linux',call_pdb=0):
1167 1172 FormattedTB.__init__(self,color_scheme=color_scheme,
1168 1173 call_pdb=call_pdb)
1169 1174
1170 1175
1171 1176 class SyntaxTB(ListTB):
1172 1177 """Extension which holds some state: the last exception value"""
1173 1178
1174 1179 def __init__(self,color_scheme = 'NoColor'):
1175 1180 ListTB.__init__(self,color_scheme)
1176 1181 self.last_syntax_error = None
1177 1182
1178 1183 def __call__(self, etype, value, elist):
1179 1184 self.last_syntax_error = value
1180 1185 ListTB.__call__(self,etype,value,elist)
1181 1186
1182 1187 def clear_err_state(self):
1183 1188 """Return the current error state and clear it"""
1184 1189 e = self.last_syntax_error
1185 1190 self.last_syntax_error = None
1186 1191 return e
1187 1192
1188 1193 def stb2text(self, stb):
1189 1194 """Convert a structured traceback (a list) to a string."""
1190 1195 return ''.join(stb)
1191 1196
1192 1197
1193 1198 #----------------------------------------------------------------------------
1194 1199 # module testing (minimal)
1195 1200 if __name__ == "__main__":
1196 1201 def spam(c, (d, e)):
1197 1202 x = c + d
1198 1203 y = c * d
1199 1204 foo(x, y)
1200 1205
1201 1206 def foo(a, b, bar=1):
1202 1207 eggs(a, b + bar)
1203 1208
1204 1209 def eggs(f, g, z=globals()):
1205 1210 h = f + g
1206 1211 i = f - g
1207 1212 return h / i
1208 1213
1209 1214 print ''
1210 1215 print '*** Before ***'
1211 1216 try:
1212 1217 print spam(1, (2, 3))
1213 1218 except:
1214 1219 traceback.print_exc()
1215 1220 print ''
1216 1221
1217 1222 handler = ColorTB()
1218 1223 print '*** ColorTB ***'
1219 1224 try:
1220 1225 print spam(1, (2, 3))
1221 1226 except:
1222 1227 apply(handler, sys.exc_info() )
1223 1228 print ''
1224 1229
1225 1230 handler = VerboseTB()
1226 1231 print '*** VerboseTB ***'
1227 1232 try:
1228 1233 print spam(1, (2, 3))
1229 1234 except:
1230 1235 apply(handler, sys.exc_info() )
1231 1236 print ''
1232 1237
General Comments 0
You need to be logged in to leave comments. Login now