mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-08-02 01:42:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Implement iter(), len() and is_empty() for all display-literal AST nodes (#12807)

Browse source
This commit is contained in:
Alex Waygood 2024-08-12 11:39:28 +01:00 • committed by GitHub
parent a99a45868c
commit aa0db338d9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
56 changed files with 304 additions and 240 deletions
Download patch file Download diff file Expand all files Collapse all files

28
crates/ruff_python_codegen/src/generator.rs
View file

@ -921,10 +921,10 @@ impl<'a> Generator<'a> {
self.unparse_expr(orelse, precedence::IF_EXP);
});
}
Expr::Dict(ast::ExprDict { items, range: _ }) => {
Expr::Dict(dict) => {
self.p("{");
let mut first = true;
for ast::DictItem { key, value } in items {
for ast::DictItem { key, value } in dict {
self.p_delim(&mut first, ", ");
if let Some(key) = key {
self.unparse_expr(key, precedence::COMMA);
@ -937,15 +937,15 @@ impl<'a> Generator<'a> {
}
self.p("}");
}
Expr::Set(ast::ExprSet { elts, range: _ }) => {
if elts.is_empty() {
Expr::Set(set) => {
if set.is_empty() {
self.p("set()");
} else {
self.p("{");
let mut first = true;
for v in elts {
for item in set {
self.p_delim(&mut first, ", ");
self.unparse_expr(v, precedence::COMMA);
self.unparse_expr(item, precedence::COMMA);
}
self.p("}");
}
@ -1164,26 +1164,26 @@ impl<'a> Generator<'a> {
self.unparse_expr(value, precedence::MAX);
}
Expr::Name(ast::ExprName { id, .. }) => self.p(id.as_str()),
Expr::List(ast::ExprList { elts, .. }) => {
Expr::List(list) => {
self.p("[");
let mut first = true;
for elt in elts {
for item in list {
self.p_delim(&mut first, ", ");
self.unparse_expr(elt, precedence::COMMA);
self.unparse_expr(item, precedence::COMMA);
}
self.p("]");
}
Expr::Tuple(ast::ExprTuple { elts, .. }) => {
if elts.is_empty() {
Expr::Tuple(tuple) => {
if tuple.is_empty() {
self.p("()");
} else {
group_if!(precedence::TUPLE, {
let mut first = true;
for elt in elts {
for item in tuple {
self.p_delim(&mut first, ", ");
self.unparse_expr(elt, precedence::COMMA);
self.unparse_expr(item, precedence::COMMA);
}
self.p_if(elts.len() == 1, ",");
self.p_if(tuple.len() == 1, ",");
});
}
}