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