TextIO是Python中用于处理文本输入输出流的基础接口，在终端场景下实现彩色文本输出需要结合终端对ANSI转义序列的支持特性，同时遵循TextIO的编码和写入规范才能保证输出效果稳定。
核心原理：ANSI转义序列
大多数现代终端都支持ANSI转义序列来控制文本样式，彩色文本的输出本质是在文本内容前后添加对应的转义控制符。常见的颜色控制序列格式为 33[显示方式;前景色;背景色m，其中 33是转义起始符，对应八进制的ESC字符。
常用的前景色编码如下：
- 30：黑色
- 31：红色
- 32：绿色
- 33：黄色
- 34：蓝色
- 35：紫色
- 36：青色
- 37：白色
显示方式常用编码：0为重置所有样式，1为加粗，4为下划线。
基础实现方式
通过TextIO的write方法直接写入包含转义序列的字符串即可实现彩色输出，需要注意字符串的编码要符合终端要求，通常UTF-8编码不会出现兼容问题。
import sys
# 获取标准输出的TextIO对象
text_io = sys.stdout
# 输出红色加粗文本
text_io.write(" 33[1;31m这是红色加粗文本 33[0mn")
# 输出绿色普通文本
text_io.write(" 33[0;32m这是绿色普通文本 33[0mn")
# 刷新缓冲区确保内容立即输出
text_io.flush()
封装为可复用的工具函数
为了避免重复拼接转义序列，可以封装通用的彩色输出函数，同时处理样式重置逻辑，减少使用时的出错概率。
import sys
def colored_output(text_io, text, color_code, bold=False):
"""
通过TextIO对象输出彩色文本
:param text_io: TextIO类型的输出流对象
:param text: 要输出的文本内容
:param color_code: 前景色编码，如31代表红色
:param bold: 是否加粗，默认为False
"""
# 拼接显示方式部分
display_mode = "1" if bold else "0"
# 拼接完整的转义序列和内容
content = f" 33[{display_mode};{color_code}m{text} 33[0m"
text_io.write(content)
text_io.flush()
# 使用示例
if __name__ == "__main__":
stdout_io = sys.stdout
colored_output(stdout_io, "蓝色提示信息n", 34)
colored_output(stdout_io, "黄色加粗警告n", 33, bold=True)
兼容性处理
部分老旧终端或者特定环境可能不支持ANSI转义序列，直接输出会导致乱码。可以通过检测终端能力来决定是否添加颜色控制符，或者使用第三方库来做兼容处理。
简单的兼容性检测逻辑如下：
import sys
import os
def is_terminal_support_color():
"""检测当前终端是否支持ANSI颜色"""
# 检查是否是交互式终端
if not sys.stdout.isatty():
return False
# 检查环境变量，Windows下的TERM或者ANSICON变量
if os.name == "nt":
return "ANSICON" in os.environ or "WT_SESSION" in os.environ
# 类Unix系统检查TERM变量
term = os.environ.get("TERM", "")
return term != "dumb" and "color" in term
def safe_colored_output(text_io, text, color_code, bold=False):
"""带兼容性检查的彩色输出"""
if is_terminal_support_color():
display_mode = "1" if bold else "0"
content = f" 33[{display_mode};{color_code}m{text} 33[0m"
else:
content = text
text_io.write(content)
text_io.flush()
# 使用示例
if __name__ == "__main__":
safe_colored_output(sys.stdout, "兼容模式的彩色文本n", 35)
注意事项
在使用TextIO实现彩色文本输出时，需要注意以下几点：
- 转义序列中的
33也可以用x1b表示，两者效果一致 - 输出完成后一定要调用flush方法，避免内容停留在缓冲区不显示
- 不要在非终端场景（如写入文件）时使用颜色转义序列，会导致文件内容包含无效控制符
- 如果需要更复杂的颜色控制，可以考虑使用colorama等第三方库，其内部已经做好了全平台兼容处理
如果需要在非标准输出的TextIO对象（如自定义的StringIO）中测试彩色输出逻辑，只需要保证写入的字符串包含正确的转义序列即可，只是最终显示效果取决于输出目标是否支持ANSI序列。
from io import StringIO # 自定义TextIO对象测试 test_io = StringIO() colored_output(test_io, "测试文本", 36) # 获取写入的内容，包含转义序列 print(repr(test_io.getvalue()))