python - 如何在 Unix 上运行的 python 脚本输出中显示 ANSI 颜色

Question

我有一个简单的 Python 脚本，它使用 ANSI 转义序列在终端上显示彩色输出。

这很好用，但是当输出在其他地方使用时(例如在 VIM 中)，所有 ANSI 序列都会出现，这使得它真的不可读。

例如，要显示红色，我会这样做:

^[[91m Errors:
-------^[[0m

这在终端中是完全可读的。

我可以在我的 Python 命令行工具中添加一个标志，以避免在我需要处理输出时显示这些字符，但我想知道是否有一种方法可以使彩色输出而不会搞砸输出。

解决方案必须使用 Python Stdlib 或必须避免安装第三方库。

但是，如果该方法在 Windows 上不起作用，我完全没问题:)

Answer 1

您可以将 os.isatty 与 stdout 文件描述符一起使用，它可以通过调用 sys.stdout 上的 fileno 来检索。

编辑: 看起来文件也有一个isatty方法；因此，您可以避免使用 os 模块，而只使用 sys.stdout.isatty()。