Function unwrap_cell_body

marimo/_ast/parse.py:87–106

Extract the body of a wrapped `def _():` cell after ruff formatting. Used by RuffFormatter to unwrap cells that were temporarily wrapped in a dummy function so ruff applies function-scope blank-line rules (E301) instead of file-scope rules (E302).

(formatted: str)

def unwrap_cell_body(formatted: str) -> str:
    """Extract the body of a wrapped `def _():` cell after ruff formatting.

    Used by RuffFormatter to unwrap cells that were temporarily wrapped in a
    dummy function so ruff applies function-scope blank-line rules (E301)
    instead of file-scope rules (E302).
    """
    tree = ast_parse(formatted)
    fn = tree.body[0]
    if not isinstance(fn, ast.FunctionDef):
        raise ValueError(
            f"Expected a FunctionDef node, got {type(fn).__name__}"
        )
    if fn.end_lineno is None:
        raise ValueError("FunctionDef node has no end_lineno")
    extractor = Extractor(formatted)
    raw = extractor.extract_from_offsets(
        fn.body[0].lineno - 1, 0, fn.end_lineno - 1, fn.end_col_offset
    )
    return fixed_dedent(raw).strip()


def extract_lineno(node: Node) -> int: