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