Function _extract_body_code

marimo/_ast/scanner.py:397–421

Extract just the body code from a cell source string. Strips decorator lines, def/class/with header, and trailing return. Used when a cell fails ast.parse() to get the body for UnparsableCell.

(cell_source: str, kind: str)

def _extract_body_code(cell_source: str, kind: str) -> str:
    """Extract just the body code from a cell source string.

    Strips decorator lines, def/class/with header, and trailing return.
    Used when a cell fails ast.parse() to get the body for UnparsableCell.
    """
    from marimo._ast.parse import fixed_dedent

    if kind == "unparsable":
        # Already app._unparsable_cell(...) — return as-is
        return cell_source

    lines = cell_source.splitlines(keepends=True)
    body_start = _find_body_start(lines, kind)

    if body_start >= len(lines):
        return ""

    body_lines = list(lines[body_start:])

    # Strip trailing return at body indentation level
    _strip_trailing_return(body_lines)

    body = "".join(body_lines)
    return fixed_dedent(body).strip()


def _find_body_start(lines: list[str], kind: str) -> int: