mirrors/RustPython
0
0 Fork 0
mirror of https://github.com/RustPython/RustPython.git synced 2026-10-29 03:04:38 +08:00
Code Activity

comprehension starred expression compatibility

Browse Source
This commit is contained in:
Jeong YunWon committed 2020-05-30 01:07:38 +00:00
1 parent 09d849ab0a
commit e4be376336
1 file changed
+15 -3
Download Patch File Download Diff File Expand all files Collapse all files
compiler/src/compile.rs
+15 -3
View File
@@ -2027,21 +2027,33 @@ impl<O: OutputStream> Compiler<O> {
}
}
let mut compile_element = |element| {
self.compile_expression(element).map_err(|e| {
if matches!(e.error, CompileErrorType::InvalidStarExpr) {
self.error(CompileErrorType::SyntaxError(
"iterable unpacking cannot be used in comprehension".to_owned(),
))
} else {
e
}
})
};
match kind {
ast::ComprehensionKind::GeneratorExpression { element } => {
self.compile_expression(element)?;
compile_element(element)?;
self.mark_generator();
self.emit(Instruction::YieldValue);
self.emit(Instruction::Pop);
}
ast::ComprehensionKind::List { element } => {
self.compile_expression(element)?;
compile_element(element)?;
self.emit(Instruction::ListAppend {
i: 1 + generators.len(),
});
}
ast::ComprehensionKind::Set { element } => {
self.compile_expression(element)?;
compile_element(element)?;
self.emit(Instruction::SetAdd {
i: 1 + generators.len(),
});