如何在Windows 10控制台中使用ANSI转义序列的新支持？

Question

最新的Windows 10更新包括对 conhost.exe中的ANSI转义序列的支持.

我已经能够确认在cmd.exe中正确地获取了转义序列,所以我有必要的更新.特别是,我尝试输入prompt $e[?25l,隐藏光标,然后prompt $e[?25h再次显示光标.

但是,如果我启动Python解释器,然后执行以下操作:

>>> import sys
>>> sys.stdout.write("\033[?25l")

好吧,光标没有隐藏.如何以正确的方式设置,以便控制台能够从Python获取转义序列？

Answer 1

问题是Python解释器不能处理ANSI转义序列.ANSI序列在Windows命令提示符下工作,因为cmd它们启用了它们.如果从命令提示符启动Python,您将发现ANSI序列确实有效,包括用于启用和禁用游标的序列.那是因为cmd已经为该控制台窗口启用了它们.

如果你想要点击某些内容来启动带有ANSI转义的Python解释器,你可以创建一个运行命令的快捷方式cmd /c C:\PythonXY\python.

另一个更难的解决方案是使用ctypes通过调用SetConsoleMode带有ENABLE_VIRTUAL_TERMINAL_PROCESSING标志集的Windows API 来为控制台窗口启用ANSI转义序列处理.例如:

import ctypes

kernel32 = ctypes.windll.kernel32
kernel32.SetConsoleMode(kernel32.GetStdHandle(-11), 7)

Answer 2

我在这里建议的一些代码的改编应该可以帮助您入门。在Windows 10上启用ANSI VT模式（虚拟终端处理）。传入stdout或stderr的参数值。12

def _windows_enable_ANSI(std_id):
    """Enable Windows 10 cmd.exe ANSI VT Virtual Terminal Processing."""
    from ctypes import byref, POINTER, windll, WINFUNCTYPE
    from ctypes.wintypes import BOOL, DWORD, HANDLE

    GetStdHandle = WINFUNCTYPE(
        HANDLE,
        DWORD)(('GetStdHandle', windll.kernel32))

    GetFileType = WINFUNCTYPE(
        DWORD,
        HANDLE)(('GetFileType', windll.kernel32))

    GetConsoleMode = WINFUNCTYPE(
        BOOL,
        HANDLE,
        POINTER(DWORD))(('GetConsoleMode', windll.kernel32))

    SetConsoleMode = WINFUNCTYPE(
        BOOL,
        HANDLE,
        DWORD)(('SetConsoleMode', windll.kernel32))

    if std_id == 1:       # stdout
        h = GetStdHandle(-11)
    elif std_id == 2:     # stderr
        h = GetStdHandle(-12)
    else:
        return False

    if h is None or h == HANDLE(-1):
        return False

    FILE_TYPE_CHAR = 0x0002
    if (GetFileType(h) & 3) != FILE_TYPE_CHAR:
        return False

    mode = DWORD()
    if not GetConsoleMode(h, byref(mode)):
        return False

    ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004
    if (mode.value & ENABLE_VIRTUAL_TERMINAL_PROCESSING) == 0:
        SetConsoleMode(h, mode.value | ENABLE_VIRTUAL_TERMINAL_PROCESSING)
    return True