mirrors/RustPython-Parser
1
0
Fork 0
mirror of https://github.com/RustPython/Parser.git synced 2025-07-19 19:15:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix async for compilation

Browse source
This commit is contained in:
Noah 2021-04-06 21:00:06 -05:00
parent 593a9d87dc
commit ab1dd4e2a4
1 changed files with 4 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/compile.rs
View file

@ -1391,22 +1391,18 @@ impl Compiler {
// The thing iterated:
self.compile_expression(iter)?;
let check_asynciter_block = if is_async {
let check_asynciter_block = self.new_block();
if is_async {
self.emit(Instruction::GetAIter);
self.switch_to_block(for_block);
self.emit(Instruction::SetupExcept {
handler: check_asynciter_block,
handler: else_block,
});
self.emit(Instruction::GetANext);
self.emit_constant(ConstantData::None);
self.emit(Instruction::YieldFrom);
self.compile_store(target)?;
self.emit(Instruction::PopBlock);
Some(check_asynciter_block)
} else {
// Retrieve Iterator
self.emit(Instruction::GetIter);
@ -1416,7 +1412,6 @@ impl Compiler {
// Start of loop iteration, set targets:
self.compile_store(target)?;
None
};
let was_in_loop = self.ctx.loop_data;
@ -1425,12 +1420,10 @@ impl Compiler {
self.ctx.loop_data = was_in_loop;
self.emit(Instruction::Jump { target: for_block });
if let Some(check_asynciter_block) = check_asynciter_block {
self.switch_to_block(check_asynciter_block);
self.switch_to_block(else_block);
if is_async {
self.emit(Instruction::EndAsyncFor);
}
self.switch_to_block(else_block);
self.emit(Instruction::PopBlock);
self.compile_statements(orelse)?;