在Python中捕获ANSI彩色输出需模拟TTY环境或绕过终端检测:一、用script命令创建伪终端;二、设环境变量如LS_COLORS=1并加--color=always参数;三、Linux/macOS用pexpect.spawn分配PTY;四、Windows用winpty封装;五、合并stdout与stderr流。

如果您在 Python 中使用 subprocess 执行带有 ANSI 转义序列的命令(如 ls --color=always、pytest -v 或 git log --oneline --graph),默认情况下这些彩色输出会被丢弃或显示为乱码,因为子进程检测到 stdout 不是终端(TTY)而禁用颜色。以下是实现彩色输出捕获的多种方法:
一、使用 script 命令模拟伪终端
script 命令可创建一个伪终端(PTY),使被调用程序认为它正在向真实终端输出,从而启用 ANSI 颜色支持。
1、确保系统已安装 script 命令(Linux/macOS 默认提供;Windows 需通过 WSL 或第三方工具)。
2、构造 subprocess 调用,将目标命令作为 script 的参数,并指定 /dev/null 作为脚本日志文件以避免写入磁盘。
立即学习“Python免费学习笔记(深入)”;
3、使用 shell=True 并传递完整命令字符串,例如:script -qec "ls --color=always /tmp" /dev/null。
4、捕获 stdout 并解码为字符串,ANSI 序列将完整保留在输出中。
二、强制设置环境变量并禁用 TTY 检测
某些程序(如 ls、grep、git)支持通过环境变量显式启用颜色,无需依赖终端判断,绕过 TTY 检查逻辑。
1、构建包含必要环境变量的字典,例如 env = os.environ.copy()。
2、向该字典添加 LS_COLORS=1、GREP_COLORS=always、GIT_COLOR=always 等键值对,具体依目标命令而定。
3、对支持 --color=always 的命令,在 args 列表中显式加入该参数,例如 ["ls", "--color=always", "/home"]。
4、调用 subprocess.run() 并传入 env 参数和 capture_output=True。
三、使用 pexpect 在 Linux/macOS 上分配伪终端
pexpect 是一个用于控制交互式程序的库,其 spawn 类可分配伪终端,使子进程识别为运行于 TTY 环境。
1、通过 pip install pexpect 安装库(Windows 不原生支持,需使用 pexpect-win 或 WSL)。
2、导入 pexpect 并调用 pexpect.spawn(command, encoding='utf-8', timeout=30)。
3、调用 child.expect(pexpect.EOF) 等待命令结束。
4、读取 child.before 属性,其中包含含 ANSI 序列的原始输出。
四、使用 winpty 在 Windows 上模拟终端
在 Windows 原生环境下,winpty 提供了类似 script 的伪终端封装能力,可让 cmd.exe 或 PowerShell 子进程输出 ANSI 颜色。
1、下载 winpty 并将其 bin 目录加入系统 PATH,或指定完整路径调用。
2、构造 subprocess 调用,将原始命令作为 winpty 的参数,例如 ["winpty", "python", "-c", "print('\x1b[32mOK\x1b[0m')"]。
3、设置 stdout=subprocess.PIPE 并执行 run()。
4、获取 stdout 值,ANSI 字节将被保留,可直接解码或进一步处理。
五、重定向 stderr 并合并流以捕获全部彩色内容
部分程序将彩色输出写入 stderr(如 pytest、some Node.js 工具),仅捕获 stdout 将丢失颜色信息。
1、在 subprocess.run() 中同时设置 stdout=subprocess.PIPE 和 stderr=subprocess.STDOUT。
2、或分别捕获 stdout 和 stderr 后拼接:output = result.stdout + result.stderr。
3、确认目标命令实际输出通道,例如通过手动运行 command 2>&1 | cat -v 观察转义符是否出现在 stderr。
4、对合并后的字节流进行 decode('utf-8', errors='ignore'),保留所有 ANSI 控制字符。










