如何在不破壞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