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