asyncio meh

Author: youknowone

crates/codegen/src/compile.rs

@@ -1288,6 +1288,19 @@ impl Compiler {
             }
         }
 
+        // Clear the exception handlers of all FinallyTry fblocks that we're going to unwind.
+        // This prevents exceptions raised during inline finally compilation from being
+        // caught by other FinallyTry handlers in the stack, which would cause those
+        // finally blocks to re-execute.
+        {
+            let code = self.current_code_info();
+            for info in &unwind_infos {
+                if let UnwindInfo::FinallyTry { fblock_idx, .. } = info {
+                    code.fblock[*fblock_idx].fb_handler = None;
+                }
+            }
+        }
+
         // Process each fblock
         for info in unwind_infos {
             match info {
@@ -1301,40 +1314,19 @@ impl Compiler {
                     let saved_fblock = code.fblock.remove(fblock_idx);
 
                     // Push PopValue fblock if preserving tos
-                    // IMPORTANT: When preserving TOS (return value), we need to update the
-                    // exception handler's stack_depth to account for the return value on stack.
-                    // Otherwise, if an exception occurs during the finally body, the stack
-                    // will be unwound to the wrong depth and the return value will be lost.
+                    // IMPORTANT: Do NOT set an exception handler here!
+                    // When the finally body is compiled inline (for return/break/continue),
+                    // if an exception occurs during execution, it should propagate naturally
+                    // up the call stack, NOT go to the try-finally's exception handler.
+                    // Setting a handler here would cause the finally body to run twice.
                     if preserve_tos {
-                        // Get the handler info from the saved fblock (or current handler)
-                        // and create a new handler with stack_depth + 1
-                        let (handler, stack_depth, preserve_lasti) =
-                            if let Some(handler) = saved_fblock.fb_handler {
-                                (
-                                    Some(handler),
-                                    saved_fblock.fb_stack_depth + 1, // +1 for return value
-                                    saved_fblock.fb_preserve_lasti,
-                                )
-                            } else {
-                                // No handler in saved_fblock, check current handler
-                                if let Some(current_handler) = self.current_except_handler() {
-                                    (
-                                        Some(current_handler.handler_block),
-                                        current_handler.stack_depth + 1, // +1 for return value
-                                        current_handler.preserve_lasti,
-                                    )
-                                } else {
-                                    (None, 1, false) // No handler, but still track the return value
-                                }
-                            };
-
                         self.push_fblock_with_handler(
                             FBlockType::PopValue,
                             saved_fblock.fb_block,
                             saved_fblock.fb_block,
-                            handler,
-                            stack_depth,
-                            preserve_lasti,
+                            None, // No handler - let exceptions propagate naturally
+                            1,    // Stack depth for the return value
+                            false,
                         )?;
                     }
 
@@ -1344,7 +1336,7 @@ impl Compiler {
                         self.pop_fblock(FBlockType::PopValue);
                     }
 
-                    // Restore the fblock
+                    // Restore the fblock (with its handler already cleared)
                     let code = self.current_code_info();
                     code.fblock.insert(fblock_idx, saved_fblock);
                 }

crates/stdlib/src/lib.rs

@@ -11,6 +11,7 @@ extern crate alloc;
 #[macro_use]
 pub(crate) mod macros;
 
+mod _asyncio;
 pub mod array;
 mod binascii;
 mod bisect;