Skip to content

Commit 739f92e

Browse files
youknowoneyouknowone
committed
impl sys.monitoring
1 parent f35d0fd commit 739f92e

8 files changed

Lines changed: 1219 additions & 31 deletions

File tree

‎Lib/test/test_compile.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1012,8 +1012,6 @@ def return_genexp():
10121012
code_lines = self.get_code_lines(genexp_code)
10131013
self.assertEqual(genexp_lines, code_lines)
10141014

1015-
# TODO: RUSTPYTHON; implicit return line number after async for
1016-
@unittest.expectedFailure
10171015
def test_line_number_implicit_return_after_async_for(self):
10181016

10191017
async def test(aseq):

‎crates/codegen/src/compile.rs‎

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2865,6 +2865,9 @@ impl Compiler {
28652865
// Normal path jumps here to skip exception path blocks
28662866
let end_block = self.new_block();
28672867

2868+
// Emit NOP at the try: line so LINE events fire for it
2869+
emit!(self, Instruction::Nop);
2870+
28682871
// Setup a finally block if we have a finally statement.
28692872
// Push fblock with handler info for exception table generation
28702873
// IMPORTANT: handler goes to finally_except_block (exception path), not finally_block
@@ -3018,8 +3021,10 @@ impl Compiler {
30183021
type_,
30193022
name,
30203023
body,
3024+
range: handler_range,
30213025
..
30223026
}) = &handler;
3027+
self.set_source_range(*handler_range);
30233028
let next_handler = self.new_block();
30243029

30253030
// If we gave a typ,
@@ -3313,6 +3318,9 @@ impl Compiler {
33133318
};
33143319
let exit_block = self.new_block();
33153320

3321+
// Emit NOP at the try: line so LINE events fire for it
3322+
emit!(self, Instruction::Nop);
3323+
33163324
// Push fblock with handler info for exception table generation
33173325
if !finalbody.is_empty() {
33183326
emit!(
@@ -3743,6 +3751,9 @@ impl Compiler {
37433751
is_async: bool,
37443752
funcflags: bytecode::MakeFunctionFlags,
37453753
) -> CompileResult<()> {
3754+
// Save source range so MAKE_FUNCTION gets the `def` line, not the body's last line
3755+
let saved_range = self.current_source_range;
3756+
37463757
// Always enter function scope
37473758
self.enter_function(name, parameters)?;
37483759
self.current_code_info()
@@ -3796,6 +3807,9 @@ impl Compiler {
37963807
let code = self.exit_scope();
37973808
self.ctx = prev_ctx;
37983809

3810+
// Restore source range so MAKE_FUNCTION is attributed to the `def` line
3811+
self.set_source_range(saved_range);
3812+
37993813
// Create function object with closure
38003814
self.make_closure(code, funcflags)?;
38013815

@@ -5169,15 +5183,22 @@ impl Compiler {
51695183
if is_async {
51705184
emit!(self, Instruction::EndAsyncFor);
51715185
} else {
5172-
// END_FOR + POP_ITER pattern (CPython 3.14)
5173-
// FOR_ITER jumps to END_FOR, but VM skips it (+1) to reach POP_ITER
5186+
// END_FOR + POP_ITER are on the `for` line, not the body's last line
5187+
let saved_range = self.current_source_range;
5188+
self.set_source_range(iter.range());
51745189
emit!(self, Instruction::EndFor);
51755190
emit!(self, Instruction::PopIter);
5191+
self.set_source_range(saved_range);
51765192
}
51775193
self.compile_statements(orelse)?;
51785194

51795195
self.switch_to_block(after_block);
51805196

5197+
// Restore source range to the `for` line so any implicit return
5198+
// (LOAD_CONST None, RETURN_VALUE) is attributed to the `for` line,
5199+
// not the loop body's last line.
5200+
self.set_source_range(iter.range());
5201+
51815202
self.leave_conditional_block();
51825203
Ok(())
51835204
}

‎crates/codegen/src/ir.rs‎

Lines changed: 42 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -250,15 +250,33 @@ impl CodeInfo {
250250
let mut locations = Vec::new();
251251
let mut linetable_locations: Vec<LineTableLocation> = Vec::new();
252252

253-
// Convert pseudo ops and remove resulting NOPs
253+
// Convert pseudo ops and remove resulting NOPs (keep line-marker NOPs)
254254
convert_pseudo_ops(&mut blocks, varname_cache.len() as u32);
255255
for block in blocks
256256
.iter_mut()
257257
.filter(|b| b.next != BlockIdx::NULL || !b.instructions.is_empty())
258258
{
259-
block
259+
// Collect lines that have non-NOP instructions in this block
260+
let non_nop_lines: std::collections::HashSet<_> = block
260261
.instructions
261-
.retain(|ins| !matches!(ins.instr.real(), Some(Instruction::Nop)));
262+
.iter()
263+
.filter(|ins| !matches!(ins.instr.real(), Some(Instruction::Nop)))
264+
.map(|ins| ins.location.line)
265+
.collect();
266+
let mut kept_nop_lines: std::collections::HashSet<OneIndexed> =
267+
std::collections::HashSet::new();
268+
block.instructions.retain(|ins| {
269+
if matches!(ins.instr.real(), Some(Instruction::Nop)) {
270+
let line = ins.location.line;
271+
// Remove if another instruction covers this line,
272+
// or if we already kept a NOP for this line
273+
if non_nop_lines.contains(&line) || kept_nop_lines.contains(&line) {
274+
return false;
275+
}
276+
kept_nop_lines.insert(line);
277+
}
278+
true
279+
});
262280
}
263281

264282
let mut block_to_offset = vec![Label(0); blocks.len()];
@@ -501,9 +519,14 @@ impl CodeInfo {
501519
let tuple_const = ConstantData::Tuple { elements };
502520
let (const_idx, _) = self.metadata.consts.insert_full(tuple_const);
503521

504-
// Replace preceding LOAD instructions with NOP
522+
// Replace preceding LOAD instructions with NOP, using the
523+
// BUILD_TUPLE location so remove_nops() treats them as
524+
// same-line and removes them (multi-line tuple literals
525+
// would otherwise leave line-introducing NOPs behind).
526+
let folded_loc = block.instructions[i].location;
505527
for j in start_idx..i {
506528
block.instructions[j].instr = Instruction::Nop.into();
529+
block.instructions[j].location = folded_loc;
507530
}
508531

509532
// Replace BUILD_TUPLE with LOAD_CONST
@@ -673,12 +696,23 @@ impl CodeInfo {
673696
}
674697
}
675698

676-
/// Remove NOP instructions from all blocks
699+
/// Remove NOP instructions from all blocks, but keep NOPs that introduce
700+
/// a new source line (they serve as line markers for monitoring LINE events).
677701
fn remove_nops(&mut self) {
678702
for block in &mut self.blocks {
679-
block
680-
.instructions
681-
.retain(|ins| !matches!(ins.instr.real(), Some(Instruction::Nop)));
703+
let mut prev_line = None;
704+
block.instructions.retain(|ins| {
705+
if matches!(ins.instr.real(), Some(Instruction::Nop)) {
706+
let line = ins.location.line;
707+
if prev_line == Some(line) {
708+
// Same line as previous instruction — safe to remove
709+
return false;
710+
}
711+
// This NOP introduces a new line — keep it
712+
}
713+
prev_line = Some(ins.location.line);
714+
true
715+
});
682716
}
683717
}
684718

0 commit comments

Comments
 (0)