Method _extract_node_info

split_coder.py:350–463

Convert one AST body node into NodeInfo records.

(self, node: ast.AST)

        return self.nodes

    def _extract_node_info(self, node: ast.AST) -> List[NodeInfo]:
        """Convert one AST body node into NodeInfo records."""
        results: List[NodeInfo] = []

        if isinstance(node, (ast.ClassDef,)):
            bases = [self._name_of(b) for b in node.bases]
            info = NodeInfo(
                name=node.name,
                kind="class",
                lineno=node.lineno,
                end_lineno=node.end_lineno or node.lineno,
                bases=bases,
            )
            info.source_hash = self._hash_lines(info.lineno, info.end_lineno)
            results.append(info)

        elif isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)):
            info = NodeInfo(
                name=node.name,
                kind="function",
                lineno=node.lineno,
                end_lineno=node.end_lineno or node.lineno,
            )
            info.source_hash = self._hash_lines(info.lineno, info.end_lineno)
            results.append(info)

        elif isinstance(node, ast.Assign):
            # Capture top-level assignments (constants, config dicts, etc.)
            for target in node.targets:
                name = self._name_of(target)
                if name:
                    end = node.end_lineno or node.lineno
                    kind = "constant" if re.match(r"^[A-Z][A-Z0-9_]{2,}$", name) else "assignment"
                    info = NodeInfo(
                        name=name,
                        kind=kind,
                        lineno=node.lineno,
                        end_lineno=end,
                    )
                    info.source_hash = self._hash_lines(info.lineno, info.end_lineno)
                    results.append(info)

        elif isinstance(node, ast.AnnAssign):
            name = self._name_of(node.target)
            if name:
                end = node.end_lineno or node.lineno
                kind = "constant" if re.match(r"^[A-Z][A-Z0-9_]{2,}$", name) else "assignment"
                info = NodeInfo(
                    name=name,
                    kind=kind,
                    lineno=node.lineno,
                    end_lineno=end,
                )
                info.source_hash = self._hash_lines(info.lineno, info.end_lineno)
                results.append(info)

        elif isinstance(node, (ast.Import, ast.ImportFrom)):
            # Track the first import block as a special node