State

@contextmanager
def frame(
    self,
    stmts: Sequence[ASTNodeType] | StmtStream[ASTNodeType],
    region: Optional[Region] = None,
    entr_block: Optional[Block] = None,
    next_block: Optional[Block] = None,
    globals: dict[str, Any] | None = None,
    capture_callback: Optional[CallbackFn] = None,
    finalize_next: bool = True,
) -> Generator[Frame[ASTNodeType], Any, None]:
    """Context manager to push a new frame and pop it after the block."""

    if not isinstance(stmts, StmtStream):
        stmts = StmtStream(stmts)

    region = region or Region()
    region.source = self.source

    entr_block = entr_block or Block()
    region.blocks.append(entr_block)

    if self._current_frame is not None:
        globals = globals or self.current_frame.globals
    else:
        globals = globals or {}

    frame = Frame(
        state=self,
        stream=stmts,
        curr_region=region or Region(entr_block),
        entr_block=entr_block,
        curr_block=entr_block,
        next_block=next_block or Block(),
        globals=globals,
        capture_callback=capture_callback,
    )
    self.push_frame(frame)
    try:
        yield frame
    finally:
        self.pop_frame(finalize_next)

def get_global(self, node: ASTNodeType, *, no_raise: bool = False):
    """Get the global value of a node.
    Args:
        node (ASTNodeType): the node to get the global value of.
        no_raise (bool): if True, do not raise an exception if the value is not found.
    Returns:
        `LoweringABC.Result`: a proxy object to the global value. `.data` is the
            value, and `.expect(type)` will raise an exception if the value is the expected type.
    """
    if no_raise:
        return self.parent.lower_global_no_raise(self, node)
    return self.parent.lower_global(self, node)

def pop_frame(self, finalize_next: bool = True):
    """Pop the current frame and return it.

    Args:
        finalize_next(bool): If True, append the next block of the current frame.

    Returns:
        Frame: The popped frame.
    """
    if self._current_frame is None:
        raise ValueError("No frame to pop")
    frame = self._current_frame

    if finalize_next and frame.next_block.parent is None:
        frame.push(frame.next_block)
    self._current_frame = frame.parent
    return frame