codegraph/debug_python_ast2.js

const { getParser, initGrammars, loadAllGrammars } = require('./dist/extraction/grammars');

(async () => {
  await initGrammars();
  await loadAllGrammars();

  const parser = getParser('python');

  const code = `class Child(Parent, Mixin, Base):
    pass`;

  const tree = parser.parse(code);

  function walk(node, depth = 0) {
    const indent = '  '.repeat(depth);
    const preview = node.text.substring(0, 40).replace(/\n/g, '\\n');
    console.log(`${indent}${node.type} "${preview}"`);
    
    for (let i = 0; i < node.namedChildCount; i++) {
      const child = node.namedChild(i);
      if (child) walk(child, depth + 1);
    }
  }

  walk(tree.rootNode);
})();