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

Fix custom

Browse source
This commit is contained in:
Jeong YunWon 2023-05-12 03:56:29 +09:00 • committed by Micha Reiser
parent 904f5c8b37
commit 26aea4030e
No known key found for this signature in database
5 changed files with 551 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

40
ast/asdl_rs.py
View file

@ -344,14 +344,14 @@ class StructVisitor(EmitVisitor):
depth,
)
# if not sum_type_info.has_attributes:
# self.emit('#[cfg(feature = "more-attributes")]', depth)
# self.emit(f"impl Ranged for {payload_name}<TextRange> {{", depth)
# self.emit("#[inline]", depth + 1)
# self.emit("fn range(&self) -> TextRange {", depth + 1)
# self.emit("self.custom", depth + 2)
# self.emit("}", depth + 1)
# self.emit("}", depth)
if not sum_type_info.has_attributes:
self.emit('#[cfg(feature = "more-attributes")]', depth)
self.emit(f"impl<U> Custom<U> for {payload_name}<U> {{", depth)
self.emit("#[inline]", depth + 1)
self.emit("fn custom(self) -> U {", depth + 1)
self.emit("self.custom", depth + 2)
self.emit("}", depth + 1)
self.emit("}", depth)
self.emit("", depth)
def visitConstructor(self, cons, parent, depth):
@ -402,13 +402,13 @@ class StructVisitor(EmitVisitor):
self.emit_custom(product.attributes, depth + 1)
self.emit("}", depth)
# self.emit('#[cfg(feature = "more-attributes")]', depth)
# self.emit(f"impl Ranged for {product_name}<TextRange> {{", depth)
# self.emit("#[inline]", depth + 1)
# self.emit("fn range(&self) -> TextRange {", depth + 1)
# self.emit("self.custom", depth + 2)
# self.emit("}", depth + 1)
# self.emit("}", depth)
self.emit('#[cfg(feature = "more-attributes")]', depth)
self.emit(f"impl<T> Custom<U> for {product_name}<U> {{", depth)
self.emit("#[inline]", depth + 1)
self.emit("fn custom(&self) -> U {", depth + 1)
self.emit("self.custom", depth + 2)
self.emit("}", depth + 1)
self.emit("}", depth)
self.emit("", depth)
@ -421,14 +421,6 @@ class FoldTraitDefVisitor(EmitVisitor):
"fn map_user(&mut self, user: U) -> Result<Self::TargetU, Self::Error>;",
depth + 1,
)
self.emit(
"""
fn map_attributed<T>(&mut self, attributed: Attributed<T, U>) -> Result<Attributed<T, Self::TargetU>, Self::Error> where T: Ranged{
let custom = self.map_user(attributed.custom)?;
Ok(Attributed { custom, node: attributed.node })
}""",
depth + 1,
)
self.emit(
"""
fn fold<X: Foldable<U, Self::TargetU>>(&mut self, node: X) -> Result<X::Mapped, Self::Error> {
@ -968,7 +960,7 @@ def write_ast_def(mod, type_info, f):
def write_fold_def(mod, type_info, f):
f.write("""
use crate::Ranged;
use crate::generic::Custom;
""")
FoldModuleVisitor(f, type_info).visit(mod)