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

Fine-tune int types

Browse source
This commit is contained in:
Jeong YunWon 2023-05-10 19:26:53 +09:00
parent 455bcc01a0
commit 6fa3d0f90a
15 changed files with 27 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

9
ast/asdl_rs.py
View file

@ -24,6 +24,10 @@ builtin_type_mapping = {
}
assert builtin_type_mapping.keys() == asdl.builtin_types
builtin_int_mapping = {
"simple": "bool",
"is_async": "bool",
}
def rust_type_name(name):
"""Return a string for the C name of the type.
@ -31,7 +35,8 @@ def rust_type_name(name):
This function special cases the default types provided by asdl.
"""
if name in asdl.builtin_types:
return builtin_type_mapping[name]
builtin = builtin_type_mapping[name]
return builtin
elif name.islower():
return "".join(part.capitalize() for part in name.split("_"))
else:
@ -355,6 +360,8 @@ class StructVisitor(EmitVisitor):
typ = f"Option<{typ}>"
if field.seq:
typ = f"Vec<{typ}>"
if typ == "Int":
typ = builtin_int_mapping.get(field.name, typ)
name = rust_field(field.name)
self.emit(f"{vis}{name}: {typ},", depth)

4
ast/src/gen/generic.rs
View file

@ -154,7 +154,7 @@ pub struct StmtAnnAssign<U = ()> {
pub target: Box<Expr<U>>,
pub annotation: Box<Expr<U>>,
pub value: Option<Box<Expr<U>>>,
pub simple: Int,
pub simple: bool,
}
impl<U> From<StmtAnnAssign<U>> for StmtKind<U> {
@ -815,7 +815,7 @@ pub struct Comprehension<U = ()> {
pub target: Expr<U>,
pub iter: Expr<U>,
pub ifs: Vec<Expr<U>>,
pub is_async: Int,
pub is_async: bool,
}
#[derive(Clone, Debug, PartialEq)]

2
ast/src/unparse.rs
View file

@ -429,7 +429,7 @@ impl<'a> Unparser<'a> {
fn unparse_comp<U>(&mut self, generators: &[Comprehension<U>]) -> fmt::Result {
for comp in generators {
self.p(if comp.is_async.to_bool() {
self.p(if comp.is_async {
" async for "
} else {
" for "

4
parser/src/python.lalrpop
View file

@ -132,7 +132,7 @@ ExpressionStatement: ast::Stmt = {
target: Box::new(set_context(target, ast::ExprContext::Store)),
annotation: Box::new(annotation),
value: rhs.map(Box::new),
simple: ast::Int::new_bool(simple),
simple,
}.into(),
)
},
@ -1707,7 +1707,7 @@ SingleForComprehension: ast::Comprehension = {
target: set_context(target, ast::ExprContext::Store),
iter,
ifs,
is_async: ast::Int::new_bool(is_async),
is_async,
}
}
};

6
parser/src/python.rs generated
View file

@ -1,5 +1,5 @@
// auto-generated: "lalrpop 0.20.0"
// sha3: 92f216faaf9f12c4ae180e3d5821a88e6bfa435ffe36293d4485c669f055b18e
// sha3: 7764ff3f82c4bfb364f8e2c55698c0cf314d591d7aeacd8abf4c9dfe354d25e2
use crate::{
ast,
lexer::{LexicalError, LexicalErrorType},
@ -36926,7 +36926,7 @@ fn __action24<
target: Box::new(set_context(target, ast::ExprContext::Store)),
annotation: Box::new(annotation),
value: rhs.map(Box::new),
simple: ast::Int::new_bool(simple),
simple,
}.into(),
)
}
@ -39647,7 +39647,7 @@ fn __action206<
target: set_context(target, ast::ExprContext::Store),
iter,
ifs,
is_async: ast::Int::new_bool(is_async),
is_async,
}
}
}

4
parser/src/snapshots/rustpython_parser__context__tests__ann_assign_name.snap generated
View file

@ -46,9 +46,7 @@ expression: parse_ast
),
},
),
simple: Int(
1,
),
simple: true,
},
),
},

4
parser/src/snapshots/rustpython_parser__context__tests__assign_list_comp.snap generated
View file

@ -101,9 +101,7 @@ expression: parse_ast
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__context__tests__assign_set_comp.snap generated
View file

@ -101,9 +101,7 @@ expression: parse_ast
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__parser__tests__generator_expression_argument.snap generated
View file

@ -201,9 +201,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__parser__tests__parse_dict_comprehension.snap generated
View file

@ -58,9 +58,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

8
parser/src/snapshots/rustpython_parser__parser__tests__parse_double_list_comprehension.snap generated
View file

@ -69,9 +69,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
Comprehension {
target: Attributed {
@ -174,9 +172,7 @@ Attributed {
),
},
],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__parser__tests__parse_generator_comprehension.snap generated
View file

@ -46,9 +46,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__parser__tests__parse_if_else_generator_comprehension.snap generated
View file

@ -78,9 +78,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__parser__tests__parse_list_comprehension.snap generated
View file

@ -46,9 +46,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},

4
parser/src/snapshots/rustpython_parser__parser__tests__parse_named_expression_generator_comprehension.snap generated
View file

@ -87,9 +87,7 @@ Attributed {
),
},
ifs: [],
is_async: Int(
0,
),
is_async: false,
},
],
},