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

Cmpop::as_str (#72)

Browse source 
* clean up pyo3 generation

* Cmpop::as_str
This commit is contained in:
Jeong, YunWon 2023-05-29 01:53:54 +09:00 • committed by GitHub
parent 4de0cb1827
commit 531aeb3511
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 48 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

23
ast/asdl_rs.py
View file

@ -1078,11 +1078,6 @@ class ToPyo3AstVisitor(EmitVisitor):
else:
assert False, self.namespace
@property
def location(self):
# lineno, col_offset
pass
def visitModule(self, mod):
for dfn in mod.dfns:
self.visit(dfn)
@ -1133,12 +1128,13 @@ class ToPyo3AstVisitor(EmitVisitor):
def emit_to_pyo3_with_fields(self, cons, type, name):
type_info = self.type_info[type.name]
self.emit(
f"""
impl ToPyAst for ast::{name}{self.generics} {{
#[inline]
fn to_py_ast<'py>(&self, py: Python<'py>) -> PyResult<&'py PyAny> {{
let cache = Self::py_type_cache().get().unwrap();
fn to_py_ast<'py>(&self, py: Python<'py>) -> PyResult<&'py PyAny> {{
let cache = Self::py_type_cache().get().unwrap();
""",
0,
)
@ -1171,12 +1167,7 @@ class ToPyo3AstVisitor(EmitVisitor):
3,
)
continue
if field.name in (
"lineno",
"col_offset",
"end_lineno",
"end_col_offset",
):
if field.name == "lineno":
self.emit(
f"{rust_field(field.name)}.to_u32().to_object(py),",
3,
@ -1192,11 +1183,11 @@ class ToPyo3AstVisitor(EmitVisitor):
)
else:
self.emit(
"let instance = Py::<PyAny>::as_ref(&cache.0, py).call0()?;",
"let Self { range: _range } = self;",
1,
)
self.emit(
"let Self { range: _range } = self;",
"""let instance = Py::<PyAny>::as_ref(&cache.0, py).call0()?;""",
1,
)
if type.value.attributes and self.namespace == "located":
@ -1210,7 +1201,7 @@ class ToPyo3AstVisitor(EmitVisitor):
instance.setattr(cache.end_col_offset.as_ref(py), end.column.get())?;
}
""",
1,
0,
)
self.emit(
"""