如何在不破坏ansi转义码的情况下替换字符串？

Question

而不是印刷 ，下面的脚本打印 .

s = '1\x1b[1;31m2\x1b[0m3'
print(s)
s = s.replace('1', ' ')
print(s)

我知道这是因为.replace操作破坏了 ANSI 转义码。 但这是不幸的。

有什么简单的方法可以让.replace或str.translate或re.sub安全地忽略转义码？

Answer 1

使用正则表达式匹配早期答案中的 ANSI 转义序列，我们可以创建一个辅助函数，该函数仅替换文本中不属于此类序列的部分。

假设这是utils.py ：

import re

# https://stackoverflow.com/a/14693789/18771
ANSICODE = re.compile(r'\x1B[@-_][0-?]*[ -/]*[@-~]')

def replace_ansi(imput_str, search_str, replace_str):
    pos = 0
    result = []
    for m in ANSICODE.finditer(imput_str):
        text = imput_str[pos:m.start()]
        text = text.replace(search_str, replace_str)
        result.append(text)
        result.append(m.group())
        pos = m.end()

    text = imput_str[pos:]
    result.append(text)
    return ''.join(result)

用法

from utils import replace_ansi

s1 = 'bla 1\x1b[1;31mbla 2\x1b[0mbla 3'
s2 = replace_ansi(s1, '1', 'X')
print(s1)
print(s2)

印刷

bla 1[1;31mbla 2[0mbla 3
bla X[1;31mbla 2[0mbla 3