Skip to content

gh-150582: Use _winapi for _pyrepl console handles and waits #150584

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
aisk wants to merge 4 commits into
base: main
Choose a base branch
from
+15 −15
Open

gh-150582: Use _winapi for _pyrepl console handles and waits #150584

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
d77c46e
Use _winapi for _pyrepl console handles and waits
aisk May 29, 2026
8d3ef98
Use _winapi calls directly in _pyrepl
aisk May 29, 2026
4dfefb8
Move _winapi import with other imports
aisk May 29, 2026
3ff96ae
Fix mypy for _winapi usage in _pyrepl
aisk May 29, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 15 additions & 15 deletions Lib/_pyrepl/windows_console.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,10 +53,10 @@
from .windows_eventqueue import EventQueue

try:
from ctypes import get_last_error, WinDLL, windll, WinError # type: ignore[attr-defined]
from ctypes import get_last_error, WinDLL, WinError # type: ignore[attr-defined]
except:
# Keep MyPy happy off Windows
from ctypes import CDLL as WinDLL, cdll as windll
from ctypes import CDLL as WinDLL

def get_last_error() -> int:
return 42
Expand All @@ -66,6 +66,12 @@ def __init__(self, err: int | None, descr: str | None = None) -> None:
self.err = err
self.descr = descr

_winapi: types.ModuleType | None
try:
import _winapi
except ImportError:
_winapi = None

# declare nt optional to allow None assignment on other platforms
nt: types.ModuleType | None
try:
Expand Down Expand Up @@ -696,7 +702,10 @@ def wait_for_event(self, timeout: float | None) -> bool:
timeout = INFINITE
else:
timeout = int(timeout)
ret = WaitForSingleObject(InHandle, timeout)
assert _winapi is not None # to make mypy happy
ret = _winapi.WaitForSingleObject( # type: ignore[attr-defined]
InHandle, timeout
)
if ret == WAIT_FAILED:
raise WinError(get_last_error())
elif ret == WAIT_TIMEOUT:
Expand Down Expand Up @@ -796,12 +805,9 @@ class INPUT_RECORD(Structure):
STD_OUTPUT_HANDLE = -11

if sys.platform == "win32":
assert _winapi is not None # to make mypy happy
_KERNEL32 = WinDLL("kernel32", use_last_error=True)

GetStdHandle = windll.kernel32.GetStdHandle
GetStdHandle.argtypes = [DWORD]
GetStdHandle.restype = HANDLE

GetConsoleScreenBufferInfo = _KERNEL32.GetConsoleScreenBufferInfo
GetConsoleScreenBufferInfo.argtypes = [
HANDLE,
Expand Down Expand Up @@ -836,24 +842,18 @@ class INPUT_RECORD(Structure):
FlushConsoleInputBuffer.argtypes = [HANDLE]
FlushConsoleInputBuffer.restype = BOOL

WaitForSingleObject = _KERNEL32.WaitForSingleObject
WaitForSingleObject.argtypes = [HANDLE, DWORD]
WaitForSingleObject.restype = DWORD

OutHandle = GetStdHandle(STD_OUTPUT_HANDLE)
InHandle = GetStdHandle(STD_INPUT_HANDLE)
OutHandle = _winapi.GetStdHandle(STD_OUTPUT_HANDLE)
InHandle = _winapi.GetStdHandle(STD_INPUT_HANDLE)
else:

def _win_only(*args, **kwargs):
raise NotImplementedError("Windows only")

GetStdHandle = _win_only
GetConsoleScreenBufferInfo = _win_only
ScrollConsoleScreenBuffer = _win_only
GetConsoleMode = _win_only
SetConsoleMode = _win_only
ReadConsoleInput = _win_only
FlushConsoleInputBuffer = _win_only
WaitForSingleObject = _win_only
OutHandle = 0
InHandle = 0
Loading