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