mirrors/RustPython-Parser
1
0
Fork 0
mirror of https://github.com/RustPython/Parser.git synced 2025-08-03 18:29:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #4449 from harupy/fix-dict-spread-in-dict

Browse source 
Fix AST generated from a dict literal containing dict unpacking
This commit is contained in:
Jeong YunWon 2023-01-22 20:44:26 +09:00 • committed by GitHub
commit 1304e4ba2f
8 changed files with 162 additions and 56 deletions
Download patch file Download diff file Expand all files Collapse all files

43
parser/python.lalrpop
View file

@ -1136,44 +1136,11 @@ Atom<Goal>: ast::Expr = {
}.into())
},
<location:@L> "{" <e:DictLiteralValues?> "}" <end_location:@R> => {
let pairs = e.unwrap_or_default();
let (keys, values) = match pairs.iter().position(|(k,_)| k.is_none()) {
Some(unpack_idx) => {
let mut pairs = pairs;
let (keys, mut values): (_, Vec<_>) = pairs.drain(..unpack_idx).map(|(k, v)| (*k.unwrap(), v)).unzip();
fn build_map(items: &mut Vec<(ast::Expr, ast::Expr)>) -> ast::Expr {
let location = items[0].0.location;
let end_location = items[0].0.end_location;
let (keys, values) = items.drain(..).unzip();
ast::Expr {
location,
end_location,
custom: (),
node: ast::ExprKind::Dict { keys, values }
}
}
let mut items = Vec::new();
for (key, value) in pairs.into_iter() {
if let Some(key) = key {
items.push((*key, value));
continue;
}
if !items.is_empty() {
values.push(build_map(&mut items));
}
values.push(value);
}
if !items.is_empty() {
values.push(build_map(&mut items));
}
(keys, values)
},
None => pairs.into_iter().map(|(k, v)| (*k.unwrap(), v)).unzip()
};
let (keys, values) = e
.unwrap_or_default()
.into_iter()
.map(|(k, v)| (k.map(|x| *x), v))
.unzip();
ast::Expr {
location,
end_location: Some(end_location),