python 彩色文本而不修改字符串

Question

我有一个 python 字符串：

my_string = 'thisisjustadummystringfortestingpurposes'

我想用红色打印字符串的某些区域，所以我使用 ANSI 转义码：

• 红色：'\x1b[31m'
• 黑色：'\x1b[0m'

然后我要做的是，创建一个插入了 ANSI 代码的新字符串。

例子：

my_color_seq = seq[:5]       # 0:5 will be black
my_color_seq += '\x1b[31m'   # start red
my_color_seq += seq[5:10]    # 5:10 will be red
my_color_seq += '\x1b[0m'    # start black
my_color_seq += seq[10:]     # 10: will be black

print(my_color_seq)

此代码有效并正确打印区域 5:10 红色。 我什至创建了一个循环，我可以在其中使用索引和长度列表来为我想用红色着色的区域。

现在的问题是我的彩色字符串中包含不可见的字符（ANSI 代码），所以字符的索引不再是它看起来的样子。

想象一下，我现在想通过将其他一些片段涂成蓝色来再次处理彩色字符串。我不再信任索引，因为我的字符串现在比看起来要大。

当然，我可以更新索引以指向新字符串中的正确位置，但我只是想知道是否有更简单的方法来做到这一点。

Answer 1

您可以对字符串着色进行包装，并在保持原始字符串不变的同时打印返回值：

示例中未显示，但您可以对原始字符串进行切片，以选择性地将子字符串传递给包装器，并使用适当的 kwarg 值，并将返回的值重新组合到一个新字符串中，以便进一步处理或打印.

def colorize(token, bold_and_al='', fg_color='', bg_color=''):
    prefix = '\x1b['
    suffix = 'm'
    reset = '\x1b[0m'

    return '\x1b[' + bold_and_al + ";" + fg_color + ";" + bg_color + "m" + token + reset


a = 'thisisjustadummystringfortestingpurposes'
print(colorize(a, bold_and_al='1', fg_color='34', bg_color='42'))

编辑：

在 cmets 中添加 @MadPhysicist 提出的示例，以说明如何使用 colorize 包装器。：

print(''.join([colorize(a[:5], bold_and_al='1', fg_color='34', bg_color='42'), a[5:10], colorize(a[10:], bold_and_al='1', fg_color='32', bg_color='44')]))