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

Fix visitor to fit in new structure

Browse source
This commit is contained in:
Jeong YunWon 2023-05-07 21:19:29 +09:00
parent 4de9580b92
commit 243ca16b34
1 changed files with 16 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

29
ast/asdl_rs.py
View file

@ -600,15 +600,18 @@ class VisitorTraitDefVisitor(StructVisitor):
self.emit("}", depth)
def simple_sum(self, sum, name, depth):
rustname = get_rust_type(name)
self.emit_visitor(name, rustname, depth)
self.emit_empty_generic_visitor(name, rustname, depth)
enumname = get_rust_type(name)
self.emit_visitor(name, enumname, depth)
self.emit_empty_generic_visitor(name, enumname, depth)
def visit_match_for_type(self, enumname, type_, depth):
self.emit(f"{enumname}::{type_.name} {{", depth)
for field in type_.fields:
self.emit(f"{rust_field(field.name)},", depth + 1)
self.emit(f"}} => self.visit_{type_.name}(", depth)
def visit_match_for_type(self, enumname, rustname, type_, depth):
self.emit(f"{rustname}::{type_.name}", depth)
if type_.fields:
self.emit(f"({enumname}{type_.name} {{", depth)
for field in type_.fields:
self.emit(f"{rust_field(field.name)},", depth + 1)
self.emit("})", depth)
self.emit(f"=> self.visit_{type_.name}(", depth)
self.emit(f"{type_.name}Node {{", depth + 2)
self.emit("location: node.location,", depth + 2)
self.emit("end_location: node.end_location,", depth + 2)
@ -622,7 +625,7 @@ class VisitorTraitDefVisitor(StructVisitor):
def visit_sumtype(self, type_, depth):
rustname = get_rust_type(type_.name) + "Node"
self.emit_visitor(type_.name, rustname, depth)
self.emit_visitor(type_.name , rustname, depth)
self.emit_generic_visitor_signature(type_.name, rustname, depth)
for f in type_.fields:
fieldname = rust_field(f.name)
@ -656,13 +659,13 @@ class VisitorTraitDefVisitor(StructVisitor):
rustname = enumname = get_rust_type(name)
if sum.attributes:
enumname += "Kind"
self.emit_visitor(name, rustname, depth)
self.emit_generic_visitor_signature(name, rustname, depth)
rustname = enumname + "Kind"
self.emit_visitor(name, enumname, depth)
self.emit_generic_visitor_signature(name, enumname, depth)
depth += 1
self.emit("match node.node {", depth)
for t in sum.types:
self.visit_match_for_type(enumname, t, depth + 1)
self.visit_match_for_type(enumname, rustname, t, depth + 1)
self.emit("}", depth)
depth -= 1
self.emit("}", depth)