The prompt-toolkit utility function format_fragments().

format_fragments.py

"""The prompt-toolkit utility function :func:`format_fragments`."""

from prompt_toolkit.formatted_text import FormattedText


def format_fragments(fragments, *args, **kwargs):
    """Applies :meth:`str.format` to each text fragment in `fragments`.

    Args:
        fragments (FormattedText): The input list of text fragments.
        *args: The positional arguments to supply to :meth:`str.format`.
        **kwargs: The keyword arguments to supply to :meth:`str.format`.

    Returns:
        FormattedText: The resulting list of text fragments.
    """
    out = FormattedText()
    for style, text, *rest in fragments:
        out.append((style, text.format(*args, **kwargs), *rest))
    return out