Add typing for _wait_for_object.py

trio/_core/_windows_cffi.py

@@ -1,5 +1,11 @@
+from __future__ import annotations
+
 import enum
 import re
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from typing_extensions import NoReturn, TypeAlias
 
 import cffi
 
@@ -215,7 +221,8 @@
 #   being _MSC_VER >= 800)
 LIB = re.sub(r"\bPASCAL\b", "__stdcall", LIB)
 
-ffi = cffi.FFI()
+ffi = cffi.api.FFI()
+CData: TypeAlias = cffi.api.FFI.CData
 ffi.cdef(LIB)
 
 kernel32 = ffi.dlopen("kernel32.dll")
@@ -302,20 +309,24 @@
 ################################################################
 
 
-def _handle(obj):
+def _handle(obj: int | CData) -> CData:
     # For now, represent handles as either cffi HANDLEs or as ints.  If you
     # try to pass in a file descriptor instead, it's not going to work
     # out. (For that msvcrt.get_osfhandle does the trick, but I don't know if
     # we'll actually need that for anything...) For sockets this doesn't
     # matter, Python never allocates an fd. So let's wait until we actually
     # encounter the problem before worrying about it.
-    if type(obj) is int:
+    if isinstance(obj, int):
         return ffi.cast("HANDLE", obj)
-    else:
-        return obj
+    return obj
 
 
-def raise_winerror(winerror=None, *, filename=None, filename2=None):
+def raise_winerror(
+    winerror: int | None = None,
+    *,
+    filename: str | None = None,
+    filename2: str | None = None,
+) -> NoReturn:
     if winerror is None:
         winerror, msg = ffi.getwinerror()
     else:

trio/_wait_for_object.py

@@ -1,11 +1,20 @@
+from __future__ import annotations
+
 import math
 
 import trio
 
-from ._core._windows_cffi import ErrorCodes, _handle, ffi, kernel32, raise_winerror
+from ._core._windows_cffi import (
+    CData,
+    ErrorCodes,
+    _handle,
+    ffi,
+    kernel32,
+    raise_winerror,
+)
 
 
-async def WaitForSingleObject(obj):
+async def WaitForSingleObject(obj: int | CData) -> None:
     """Async and cancellable variant of WaitForSingleObject. Windows only.
 
     Args:
@@ -45,7 +54,7 @@
         kernel32.CloseHandle(cancel_handle)
 
 
-def WaitForMultipleObjects_sync(*handles):
+def WaitForMultipleObjects_sync(*handles: int | CData) -> None:
     """Wait for any of the given Windows handles to be signaled."""
     n = len(handles)
     handle_arr = ffi.new(f"HANDLE[{n}]")