command

@contextlib.contextmanager
def knob_values(
    values: dict[nuke.Knob, typing.Any],
) -> typing.Generator[None, None, None]:
    """Context manager to set knob values temporarily.

    Args:
        values: Dictionary of knob and value pairs to set.

    Returns:
        Generator to yield the context manager.

    """
    original_values = {knob: knob.value() for knob in values}
    try:
        for knob, value in values.items():
            knob.setValue(value)
        yield
    finally:
        for knob, value in original_values.items():
            knob.setValue(value)

@contextlib.contextmanager
def undo_chunk(name: str = ""):
    """Context manager to wrap multiple actions into a single undo chunk.

    The name of the undo chunk can either be given when entering the block,
    or updated later using the `nuke.Undo.name(name)` method.

    Args:
        name (str): Name of the undo chunk.

    Examples:
        As a context manager:
        >>> with undo_chunk("Load Image"):
        >>>    # do multiple actions here
        >>>    ...

        As a decorator:
        >>> @undo_chunk("Load Image")
        >>> def load_image():
        >>>    # do multiple actions here
        >>>    ...

    """
    nuke.Undo.begin()
    if name:
        nuke.Undo.name(name)

    try:
        yield
    finally:
        nuke.Undo.end()

@contextlib.contextmanager
def viewer_update_and_undo_stop():
    """Lock viewer from updating and stop recording undo steps"""
    try:
        # stop active viewer to update any change
        viewer = nuke.activeViewer()
        if viewer:
            viewer.stop()
        else:
            log.warning("No available active Viewer")
        nuke.Undo.disable()
        yield
    finally:
        nuke.Undo.enable()