Escape text while keeping ANSI sequences

[pawamoy]

I'm trying to use Rich to render the output of commands, possibly containing ANSI sequence and/or unintentional Rich markup, formatted using intentional Rich markup.

from rich import print as richprint
from rich.markup import escape
from jinja2 import Environment

template_string = (
    "{% if success %}[green]✓[/green]"
    "{% elif nofail %}[yellow]✗[/yellow]"
    "{% else %}[red]✗[/red]{% endif %} "
    "[bold]{{ title or command }}[/bold]"
    "{% if failure %} ({{ code }}){% endif %}"
    "{% if failure and output and not quiet %}\n"
    "{{ ('  > ' + command + '\n') if title and command else '' }}"
    "{{ output|indent(2 * ' ') }}{% endif %}",
)

template = Environment().from_string(template_string)
richprint(
    Text.from_ansi(
        template.render(
            {
                "title": title,
                "command": escape(command),
                "code": code,
                "success": code == 0,
                "failure": code != 0,
                "number": number,
                # output could contain unintentional Rich markup that should be escaped,
                # while keeping ANSI sequences intact 
                "output": output, 
                "nofail": nofail,
                "quiet": quiet,
                "silent": silent,
            },
        ),
    ),
)

The issue is that I can either interpret ANSI sequence, or escape the unintentional Rich markup, but not both.

If I try to interpret ANSI sequences first, Text.from_ansi() returns a Text instance that I can't pass to escape().
If I escape first, ANSI sequences are escaped as well.

Is there a way to get an ANSI-interpreted string rather than a Text instance? 🤔