gh-122300: Preserve AST nodes for format specifiers with single elements (#122308)

This commit is contained in:
Pablo Galindo Salgado 2024-07-26 17:29:41 +01:00 committed by GitHub
parent 7c2921844f
commit db2d8b6db1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 8 additions and 3 deletions

View file

@ -1010,7 +1010,8 @@ _PyPegen_setup_full_format_spec(Parser *p, Token *colon, asdl_expr_seq *spec, in
spec = resized_spec;
}
expr_ty res;
if (asdl_seq_LEN(spec) == 0) {
Py_ssize_t n = asdl_seq_LEN(spec);
if (n == 0 || (n == 1 && asdl_seq_GET(spec, 0)->kind == Constant_kind)) {
res = _PyAST_JoinedStr(spec, lineno, col_offset, end_lineno,
end_col_offset, p->arena);
} else {