Function style_factory_helpers

mycli/clistyle.py:170–210

(
    name: str,
    cli_style: dict[str, str],
    warnings: bool = False,
)

def style_factory_helpers(
    name: str,
    cli_style: dict[str, str],
    warnings: bool = False,
) -> PygmentsStyle:
    try:
        style: dict[PygmentsStyle | str, str] = pygments.styles.get_style_by_name(name).styles
    except ClassNotFound:
        style = pygments.styles.get_style_by_name("native").styles

    for token in cli_style:
        if token.startswith("Token."):
            token_type, style_value = parse_pygments_style(token, style, cli_style)
            if is_valid_pygments(style_value):
                style.update({token_type: style_value})
        elif token in PROMPT_STYLE_TO_TOKEN:
            token_type = PROMPT_STYLE_TO_TOKEN[token]
            if is_valid_pygments(cli_style[token]):
                style.update({token_type: cli_style[token]})
        elif token in OVERRIDE_STYLE_TO_TOKEN:
            token_type = OVERRIDE_STYLE_TO_TOKEN[token]
            if is_valid_pygments(cli_style[token]):
                style.update({token_type: cli_style[token]})
        else:
            # TODO: cli helpers will have to switch to ptk.Style
            logger.error("Unhandled style / class name: %s", token)

    if warnings:
        for warning_token in list(style.keys()):
            if 'Warnings' not in str(warning_token):
                continue
            warning_str = str(warning_token)
            output_str = warning_str.replace('Warnings', 'Output')
            output_token = string_to_tokentype(output_str)
            style[output_token] = style[warning_token]

    class OutputStyle(PygmentsStyle):
        default_style = ""
        styles = style

    return OutputStyle