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

时间:2017-12-06 14:58:15

标签: python string colors ansi

我有一个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代码),因此字符的索引不再是它的样子。

想象一下,我现在想要通过用蓝色绘制其他一些片段来再次处理彩色字符串。我不能再信任索引,因为我的字符串现在比它看起来要大。

当然我可以更新索引以指向新字符串中的正确位置,但我只是想知道是否有更简单的方法来执行此操作。

1 个答案:

答案 0 :(得分:3)

您可以在字符串着色周围创建一个包装器,并在保持原始字符串完整的同时打印返回的值:

示例中未显示,但您可以对原始字符串进行切片,以使用适当的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'))

编辑:

在评论中添加@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')]))