mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-02-17 16:24:40 +04:00
48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
from ctypes import POINTER
|
|
from ctypes import windll
|
|
from ctypes import WinError
|
|
from ctypes import WINFUNCTYPE
|
|
from ctypes.wintypes import BOOL
|
|
from ctypes.wintypes import DWORD
|
|
from ctypes.wintypes import HANDLE
|
|
|
|
STD_OUTPUT_HANDLE = -11
|
|
ENABLE_VIRTUAL_TERMINAL_PROCESSING = 4
|
|
|
|
|
|
def bool_errcheck(result, func, args):
|
|
if not result:
|
|
raise WinError()
|
|
return args
|
|
|
|
|
|
GetStdHandle = WINFUNCTYPE(HANDLE, DWORD)(
|
|
("GetStdHandle", windll.kernel32), ((1, "nStdHandle"),),
|
|
)
|
|
|
|
GetConsoleMode = WINFUNCTYPE(BOOL, HANDLE, POINTER(DWORD))(
|
|
("GetConsoleMode", windll.kernel32),
|
|
((1, "hConsoleHandle"), (2, "lpMode")),
|
|
)
|
|
GetConsoleMode.errcheck = bool_errcheck
|
|
|
|
SetConsoleMode = WINFUNCTYPE(BOOL, HANDLE, DWORD)(
|
|
("SetConsoleMode", windll.kernel32),
|
|
((1, "hConsoleHandle"), (1, "dwMode")),
|
|
)
|
|
SetConsoleMode.errcheck = bool_errcheck
|
|
|
|
|
|
def enable_virtual_terminal_processing():
|
|
"""As of Windows 10, the Windows console supports (some) ANSI escape
|
|
sequences, but it needs to be enabled using `SetConsoleMode` first.
|
|
|
|
More info on the escape sequences supported:
|
|
https://msdn.microsoft.com/en-us/library/windows/desktop/mt638032(v=vs.85).aspx
|
|
"""
|
|
stdout = GetStdHandle(STD_OUTPUT_HANDLE)
|
|
flags = GetConsoleMode(stdout)
|
|
SetConsoleMode(stdout, flags | ENABLE_VIRTUAL_TERMINAL_PROCESSING)
|