diff options
-rw-r--r-- | mesonbuild/ast/printer.py | 3 | ||||
-rw-r--r-- | mesonbuild/ast/visitor.py | 1 |
2 files changed, 3 insertions, 1 deletions
diff --git a/mesonbuild/ast/printer.py b/mesonbuild/ast/printer.py index 2de1d0c..c6fb91a 100644 --- a/mesonbuild/ast/printer.py +++ b/mesonbuild/ast/printer.py @@ -118,6 +118,7 @@ class AstPrinter(AstVisitor): self.newline() def visit_IndexNode(self, node: mparser.IndexNode): + node.iobject.accept(self) self.append('[', node) node.index.accept(self) self.append(']', node) @@ -181,7 +182,7 @@ class AstPrinter(AstVisitor): def visit_ArgumentNode(self, node: mparser.ArgumentNode): break_args = (len(node.arguments) + len(node.kwargs)) > self.arg_newline_cutoff for i in node.arguments + list(node.kwargs.values()): - if not isinstance(i, mparser.ElementaryNode): + if not isinstance(i, (mparser.ElementaryNode, mparser.IndexNode)): break_args = True if break_args: self.newline() diff --git a/mesonbuild/ast/visitor.py b/mesonbuild/ast/visitor.py index fab4ed2..de13dae 100644 --- a/mesonbuild/ast/visitor.py +++ b/mesonbuild/ast/visitor.py @@ -84,6 +84,7 @@ class AstVisitor: def visit_IndexNode(self, node: mparser.IndexNode): self.visit_default_func(node) + node.iobject.accept(self) node.index.accept(self) def visit_MethodNode(self, node: mparser.MethodNode): |