Question

为了检测控制台是否正确sys.stderr或sys.stdout，我正在进行以下测试：

if hasattr(sys.stderr, "isatty") and sys.stderr.isatty():
   if platform.system()=='Windows':
       # win code (ANSI not supported but there are alternatives)
   else:
       # use ANSI escapes
else:
   # no colors, usually this is when you redirect the output to a file

现在，通过IDE（如PyCharm）运行此Python代码时，问题变得更加复杂。最近PyCharm添加了对ANSI的支持，但第一次测试失败：它具有isatty属性，但设置为False。

我想修改逻辑，以便正确检测输出是否支持ANSI着色。一个要求是在任何情况下我都不应该在输出重定向到文件时输出一些内容（对于控制台，它是可以接受的）。

更新

在https://gist.github.com/1316877

添加了更复杂的ANSI测试脚本

Answer 1

Django用户可以使用django.core.management.color.supports_color功能。

if supports_color():
    ...

他们使用的代码是：

def supports_color():
    """
    Returns True if the running system's terminal supports color, and False
    otherwise.
    """
    plat = sys.platform
    supported_platform = plat != 'Pocket PC' and (plat != 'win32' or
                                                  'ANSICON' in os.environ)
    # isatty is not always implemented, #6223.
    is_a_tty = hasattr(sys.stdout, 'isatty') and sys.stdout.isatty()
    if not supported_platform or not is_a_tty:
        return False
    return True

请参阅https://github.com/django/django/blob/master/django/core/management/color.py

Answer 2

我可以告诉你其他人是如何解决这个问题的，但它并不漂亮。如果你以ncurses为例（它需要能够在各种不同的终端上运行），你会看到他们使用terminal capabilities database来存储各种终端及其功能。关键是，即使他们从未能够自动“检测”这些事情。

我不知道是否存在跨平台的termcap，但它可能值得您花时间去寻找它。即使它在那里，它可能没有列出你的终端，你可能不得不手动添加它。

如何检测控制台是否支持Python中的ANSI转义码？

更新

2 个答案: