10000 gh-102192: Replace PyErr_Fetch/Restore etc by more efficient alternatives by iritkatriel · Pull Request #102743 · python/cpython · GitHub
[go: up one dir, main page]

Skip to content

gh-102192: Replace PyErr_Fetch/Restore etc by more efficient alternatives #102743

New issue 8000

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Mar 16, 2023
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
gh-102192: Replace PyErr_Fetch/Restore etc by more efficient alternat…
…ives
  • Loading branch information
iritkatriel committed Mar 15, 2023
commit d220a9ffe69ca8789e756da2e79caac1e5b79208
42 changes: 20 additions & 22 deletions Python/pythonrun.c
Original file line number Diff line number Diff line change
Expand Up @@ -698,30 +698,31 @@ _Py_HandleSystemExit(int *exitcode_p)
return 0;
}

PyObject *exception, *value, *tb;
PyErr_Fetch(&exception, &value, &tb);

fflush(stdout);

int exitcode = 0;
if (value == NULL || value == Py_None) {

PyObject *exc = PyErr_GetRaisedException();
assert(exc != Py_None);
if (exc == NULL) {
goto done;
}

if (PyExceptionInstance_Check(value)) {
/* The error code should be in the `code' attribute. */
PyObject *code = PyObject_GetAttr(value, &_Py_ID(code));
if (code) {
Py_SETREF(value, code);
if (value == Py_None)
goto done;
assert(PyExceptionInstance_Check(exc));
/* The error code should be in the `code' attribute. */
PyObject *code = PyObject_GetAttr(exc, &_Py_ID(code));
if (code) {
Py_SETREF(exc, code);
if (exc == Py_None) {
goto done;
}
/* If we failed to dig out the 'code' attribute,
just let the else clause below print the error. */
}
/* If we failed to dig out the 'code' attribute,
* just let the else clause below print the error.
*/

if (PyLong_Check(value)) {
exitcode = (int)PyLong_AsLong(value);
if (PyLong_Check(exc)) {
exitcode = (int)PyLong_AsLong(exc);
}
else {
PyThreadState *tstate = _PyThreadState_GET();
Expand All @@ -732,20 +733,17 @@ _Py_HandleSystemExit(int *exitcode_p)
*/
PyErr_Clear();
if (sys_stderr != NULL && sys_stderr != Py_None) {
PyFile_WriteObject(value, sys_stderr, Py_PRINT_RAW);
PyFile_WriteObject(exc, sys_stderr, Py_PRINT_RAW);
} else {
PyObject_Print(value, stderr, Py_PRINT_RAW);
PyObject_Print(exc, stderr, Py_PRINT_RAW);
fflush(stderr);
}
PySys_WriteStderr("\n");
exitcode = 1;
}

done:
/* Cleanup the exception */
Py_CLEAR(exception);
Py_CLEAR(value);
Py_CLEAR(tb);
done:
Py_CLEAR(exc);
*exitcode_p = exitcode;
return 1;
}
Expand Down
0