fix return value of wait() and use it in _read_input()

Author: chris-eibl

Lib/_pyrepl/windows_console.py

@@ -416,12 +416,8 @@ def _getscrollbacksize(self) -> int:
         return info.srWindow.Bottom  # type: ignore[no-any-return]
 
     def _read_input(self, block: bool = True) -> INPUT_RECORD | None:
-        if not block:
-            ret = WaitForSingleObject(InHandle, 0)
-            if ret == WAIT_FAILED:
-                raise WinError(get_last_error())
-            elif ret == WAIT_TIMEOUT:
-                return None
+        if not block and not self.wait(timeout=0):
+            return None
 
         rec = INPUT_RECORD()
         read = DWORD()
@@ -536,6 +532,9 @@ def wait(self, timeout: float | None) -> bool:
         ret = WaitForSingleObject(InHandle, timeout)
         if ret == WAIT_FAILED:
             raise WinError(get_last_error())
+        elif ret == WAIT_TIMEOUT:
+            return False
+        return True
 
     def repaint(self) -> None:
         raise NotImplementedError("No repaint support")