您可以使用ctype调用GetConsoleMode
和SetConsoleMode
。
ctypes定义:
import msvcrt
import atexit
import ctypes
from ctypes import wintypes
kernel32 = ctypes.WinDLL('kernel32', use_last_error=True)
# input flags
ENABLE_PROCESSED_INPUT = 0x0001
ENABLE_LINE_INPUT = 0x0002
ENABLE_ECHO_INPUT = 0x0004
ENABLE_WINDOW_INPUT = 0x0008
ENABLE_MOUSE_INPUT = 0x0010
ENABLE_INSERT_MODE = 0x0020
ENABLE_QUICK_EDIT_MODE = 0x0040
ENABLE_EXTENDED_FLAGS = 0x0080
# output flags
ENABLE_PROCESSED_OUTPUT = 0x0001
ENABLE_WRAP_AT_EOL_OUTPUT = 0x0002
ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004 # VT100 (Win 10)
def check_zero(result, func, args):
if not result:
err = ctypes.get_last_error()
if err:
raise ctypes.WinError(err)
return args
if not hasattr(wintypes, 'LPDWORD'): # PY2
wintypes.LPDWORD = ctypes.POINTER(wintypes.DWORD)
kernel32.GetConsoleMode.errcheck= check_zero
kernel32.GetConsoleMode.argtypes = (
wintypes.HANDLE, # _In_ hConsoleHandle
wintypes.LPDWORD,) # _Out_ lpMode
kernel32.SetConsoleMode.errcheck= check_zero
kernel32.SetConsoleMode.argtypes = (
wintypes.HANDLE, # _In_ hConsoleHandle
wintypes.DWORD,) # _Out_ lpMode
下面将底层WinAPI函数包装为get_console_mode
和set_console_mode
。我将包装器限制为仅在控制台的活动输入缓冲区或活动输出缓冲区上操作,即\\。\CONIN$
和\\。\CONOUT$
。我认为这比担心文件描述符和句柄更简单。值得注意的是,sys.stdin
和sys.stdout
可能会被重定向到其他地方,就像C运行时的标准I/OFILE
流、文件描述符和Windows标准句柄一样可以从GetStdHandle
获取。在这些情况下,只要进程附加到控制台,您仍然可以打开CONIN$
和CONOUT$
。
def get_console_mode(output=False):
'''Get the mode of the active console input or output
buffer. Note that if the process isn't attached to a
console, this function raises an EBADF IOError.
'''
device = r'\\.\CONOUT$' if output else r'\\.\CONIN$'
with open(device, 'r+') as con:
mode = wintypes.DWORD()
hCon = msvcrt.get_osfhandle(con.fileno())
kernel32.GetConsoleMode(hCon, ctypes.byref(mode))
return mode.value
def set_console_mode(mode, output=False):
'''Set the mode of the active console input or output
buffer. Note that if the process isn't attached to a
console, this function raises an EBADF IOError.
'''
device = r'\\.\CONOUT$' if output else r'\\.\CONIN$'
with open(device, 'r+') as con:
hCon = msvcrt.get_osfhandle(con.fileno())
kernel32.SetConsoleMode(hCon, mode)
update\u console\u mode
结合了后一种功能,允许您传入要设置的标志和要修改的标志的
掩码。这包括要清除的标志。它还允许通过注册atexit函数恢复以前的模式。
def update_console_mode(flags, mask, output=False, restore=False):
'''Update a masked subset of the current mode of the active
console input or output buffer. Note that if the process
isn't attached to a console, this function raises an
EBADF IOError.
'''
current_mode = get_console_mode(output)
if current_mode & mask != flags & mask:
mode = current_mode & ~mask | flags & mask
set_console_mode(mode, output)
else:
restore = False
if restore:
atexit.register(set_console_mode, current_mode, output)
例子:
if __name__ == '__main__':
import os
import sys
import time
if sys.stderr is None:
os.close(2)
sys.stderr = open('stderr.txt', 'w', buffering=1)
print("%#06x, %#06x" % (get_console_mode(),
get_console_mode(output=True)))
flags = mask = ENABLE_EXTENDED_FLAGS | ENABLE_QUICK_EDIT_MODE
update_console_mode(flags, mask, restore=True)
print("%#06x, %#06x" % (get_console_mode(),
get_console_mode(output=True)))
time.sleep(10) # check console properties
对于试图仅为当前控制台禁用QuickEdit和Insert模式且无法找到简单解决方案的任何人:
import ctypes
kernel32 = ctypes.windll.kernel32
kernel32.SetConsoleMode(kernel32.GetStdHandle(-10), 128)
这可能有助于任何试图在Windows中仅启用和禁用快速编辑模式而不禁用其他功能的人。
def quickedit(enabled=1): # This is a patch to the system that sometimes hangs
import ctypes
'''
Enable or disable quick edit mode to prevent system hangs, sometimes when using remote desktop
Param (Enabled)
enabled = 1(default), enable quick edit mode in python console
enabled = 0, disable quick edit mode in python console
'''
# -10 is input handle => STD_INPUT_HANDLE (DWORD) -10 | https://docs.microsoft.com/en-us/windows/console/getstdhandle
# default = (0x4|0x80|0x20|0x2|0x10|0x1|0x40|0x200)
# 0x40 is quick edit, #0x20 is insert mode
# 0x8 is disabled by default
# https://docs.microsoft.com/en-us/windows/console/setconsolemode
kernel32 = ctypes.windll.kernel32
if enabled:
kernel32.SetConsoleMode(kernel32.GetStdHandle(-10), (0x4|0x80|0x20|0x2|0x10|0x1|0x40|0x100))
print("Console Quick Edit Enabled")
else:
kernel32.SetConsoleMode(kernel32.GetStdHandle(-10), (0x4|0x80|0x20|0x2|0x10|0x1|0x00|0x100))
print("Console Quick Edit Disabled")
quickedit(0) # Disable quick edit in terminal
只需禁用0x40标志,这是快速编辑