subprocess 如何捕获彩色 ANSI 输出并保留格式

要捕获带颜色的ANSI输出，关键在于欺骗子进程使其认为运行在支持颜色的终端中：优先使用script -qec “cmd”或–color=always参数，其次设环境变量如CLICOLOR=1，unix下可用pty.fork()模拟TTY，windows不支持后者。

subprocess 默认会丢失 ANSI 色彩控制码，因为多数命令（如 ls --color=auto、grep --color=auto）检测到输出不是终端（tty）时，会自动禁用颜色。要捕获带颜色的 ANSI 输出，关键在于**欺骗子进程，让它认为自己正运行在支持颜色的终端中**。

让命令认为 stdout 是 tty（最常用）

很多命令只在检测到 stdout 是终端时才输出 ANSI 转义序列。可通过以下方式模拟：

• 使用 script 命令（linux/macOS）：它创建伪终端（pty），天然支持颜色。
  script -qec "your_command" /dev/NULL —— -q 静默，-e 保持退出码，-c 执行命令，/dev/null 忽略脚本日志头尾。
• python 中调用 script：

  import subprocess
  result = subprocess.run(
      ["script", "-qec", "ls --color=auto /tmp"],
      capture_output=True,
      text=False  # 保持 bytes，避免解码破坏转义码
  )
  print(result.stdout)  # 包含 x1b[34m 等 ANSI 序列

强制命令启用颜色（简单直接）

绕过终端检测，显式要求颜色输出：

• ls --color=always、grep --color=always、docker ps --format "table {{.ID}}t{{.Status}}" --color=always
• 设置环境变量（对部分工具有效）：
  env = {"CLICOLOR": "1", "LS_COLORS": os.getenv("LS_COLORS", "")}，再传给 subprocess.run(..., env=env)

手动模拟 TTY（高级，跨平台兼容性稍弱）

Python 的 pty 模块（Unix only）可创建伪终端：

import pty
import os

pid, fd = pty.fork()
if pid == 0:  # 子进程
    os.execv("/bin/ls", ["ls", "--color=auto", "/tmp"])
else:  # 父进程
    output, _ = os.read(fd, 4096), os.close(fd)
    print(output)  # 含 ANSI

⚠️ 注意：Windows 不支持 pty；且需处理读取阻塞、缓冲、退出状态等细节，适合有控制需求的场景。

捕获后处理与显示

捕获到的字节流含原始 ANSI 序列（如 b'x1b[32mOKx1b[0m'），可：

• 直接写入支持 ANSI 的终端（如 sys.stdout.buffer.write(stdout)）
• 用 ansi2html、rich 或正则清洗/转换（如去除颜色：re.sub(b'x1b[.*?m', b'', stdout)）
• 保存为文件供后续查看（需用支持 ANSI 的查看器，如 less -R）

不复杂但容易忽略：核心是让被调用程序“相信”它在终端里运行，而不是靠 Python 自己加色。优先试 --color=always 或 script，简洁可靠。