Move PyErr_WriteUnraisable call

Author: cdce8p

mypyc/codegen/emitclass.py

@@ -842,15 +842,16 @@ def generate_dealloc_for_class(
     emitter.emit_line("static void")
     emitter.emit_line(f"{dealloc_func_name}({cl.struct_name(emitter.names)} *self)")
     emitter.emit_line("{")
-    emitter.emit_line("int finalize_error = 0;")
-    emitter.emit_line("PyObject *type, *value, *traceback;")
-    emitter.emit_line("PyErr_Fetch(&type, &value, &traceback);")
     if has_tp_finalize:
         emitter.emit_line("if (!PyObject_GC_IsFinalized((PyObject *)self)) {")
+        emitter.emit_line("PyObject *type, *value, *traceback;")
+        emitter.emit_line("PyErr_Fetch(&type, &value, &traceback);")
         emitter.emit_line("PyObject_CallFinalizerFromDealloc((PyObject *)self);")
+        emitter.emit_line("if (PyErr_Occurred() != NULL) {")
+        emitter.emit_line("PyErr_WriteUnraisable((PyObject *)self);")
+        emitter.emit_line("}")
+        emitter.emit_line("PyErr_Restore(type, value, traceback);")
         emitter.emit_line("}")
-    emitter.emit_line("if (PyErr_Occurred() != NULL) finalize_error = 1;")
-    emitter.emit_line("PyErr_Restore(type, value, traceback);")
     emitter.emit_line("PyObject_GC_UnTrack(self);")
     if cl.reuse_freed_instance:
         emit_reuse_dealloc(cl, emitter)
@@ -863,9 +864,6 @@ def generate_dealloc_for_class(
     # # However, the message is slightly different due to the way mypyc compiles classes.
     # # CPython interpreter prints: Exception ignored in: <function F.__del__ at 0x100aed940>
     # # mypyc prints: Exception ignored in: <slot wrapper '__del__' of 'F' objects>
-    emitter.emit_line("if (finalize_error == 1) {")
-    emitter.emit_line("PyErr_WriteUnraisable((PyObject *)self);")
-    emitter.emit_line("}")
     emitter.emit_line("}")