Skip to content

Commit a5cf286

Browse files
committed
Enhance GrammarVisitor to differentiate visit methods for single and multi-character node classes
1 parent c26bed1 commit a5cf286

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

Tools/peg_generator/pegen/grammar.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,10 @@ class GrammarVisitor:
2020
def visit(self, node: Any, *args: Any, **kwargs: Any) -> Any:
2121
"""Visit a node."""
2222
method = "visit_" + node.__class__.__name__
23+
if len(node.__class__.__name__) == 1:
24+
method = "visit_single_" + node.__class__.__name__
25+
elif len(node.__class__.__name__) > 1:
26+
method = "visit_multi_" + node.__class__.__name__
2327
visitor = getattr(self, method, self.generic_visit)
2428
return visitor(node, *args, **kwargs)
2529

0 commit comments

Comments
 (0)