Skip to content

Commit

Permalink
[3.12] pythongh-113358: Fix rendering tracebacks with exceptions with…
Browse files Browse the repository at this point in the history
… a broken __getattr__ (pythonGH-113359)

(cherry picked from commit 04fabe2)

Co-authored-by: Jérome Perrin <perrinjerome@gmail.com>
Co-authored-by: Irit Katriel <1055913+iritkatriel@users.noreply.github.com>
  • Loading branch information
perrinjerome and iritkatriel committed Jan 20, 2024
1 parent ae2a25b commit 3588252
Show file tree
Hide file tree
Showing 4 changed files with 52 additions and 2 deletions.
15 changes: 15 additions & 0 deletions Lib/test/test_traceback.py
Original file line number Diff line number Diff line change
Expand Up @@ -1654,6 +1654,21 @@ def __repr__(self):
err_msg = "b'please do not show me as numbers'"
self.assertEqual(self.get_report(e), vanilla + err_msg + '\n')

# an exception with a broken __getattr__ raising a non expected error
class BrokenException(Exception):
broken = False
def __getattr__(self, name):
if self.broken:
raise ValueError(f'no {name}')
raise AttributeError(name)

e = BrokenException(123)
vanilla = self.get_report(e)
e.broken = True
self.assertEqual(
self.get_report(e),
vanilla + "Ignored error getting __notes__: ValueError('no __notes__')\n")

def test_exception_with_multiple_notes(self):
for e in [ValueError(42), SyntaxError('bad syntax')]:
with self.subTest(e=e):
Expand Down
6 changes: 5 additions & 1 deletion Lib/traceback.py
Original file line number Diff line number Diff line change
Expand Up @@ -738,7 +738,11 @@ def __init__(self, exc_type, exc_value, exc_traceback, *, limit=None,
# Capture now to permit freeing resources: only complication is in the
# unofficial API _format_final_exc_line
self._str = _safe_string(exc_value, 'exception')
self.__notes__ = getattr(exc_value, '__notes__', None)
try:
self.__notes__ = getattr(exc_value, '__notes__', None)
except Exception as e:
self.__notes__ = [
f'Ignored error getting __notes__: {_safe_string(e, '__notes__', repr)}']

if exc_type and issubclass(exc_type, SyntaxError):
# Handle SyntaxError's specially
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Fix rendering tracebacks for exceptions with a broken ``__getattr__``.
32 changes: 31 additions & 1 deletion Python/pythonrun.c
Original file line number Diff line number Diff line change
Expand Up @@ -1164,6 +1164,36 @@ print_exception_notes(struct exception_print_context *ctx, PyObject *notes)
return -1;
}

static int
get_exception_notes(struct exception_print_context *ctx, PyObject *value, PyObject **notes) {
PyObject *note = NULL;

if (_PyObject_LookupAttr(value, &_Py_ID(__notes__), notes) < 0) {
PyObject *type, *errvalue, *tback;
PyErr_Fetch(&type, &errvalue, &tback);
note = PyUnicode_FromFormat("Ignored error getting __notes__: %R", errvalue);
Py_XDECREF(type);
Py_XDECREF(errvalue);
Py_XDECREF(tback);
if (!note) {
goto error;
}
*notes = PyList_New(1);
if (!*notes) {
goto error;
}
if (PyList_SetItem(*notes, 0, note) < 0) {
Py_DECREF(*notes);
goto error;
}
}

return 0;
error:
Py_XDECREF(note);
return -1;
}

static int
print_exception(struct exception_print_context *ctx, PyObject *value)
{
Expand All @@ -1183,7 +1213,7 @@ print_exception(struct exception_print_context *ctx, PyObject *value)

/* grab the type and notes now because value can change below */
PyObject *type = (PyObject *) Py_TYPE(value);
if (_PyObject_LookupAttr(value, &_Py_ID(__notes__), &notes) < 0) {
if (get_exception_notes(ctx, value, &notes) < 0) {
goto error;
}

Expand Down

0 comments on commit 3588252

Please sign in to comment.