Provide a local definition of user32.SetTimer, user32.KillTimer and ole32.CoGetApartmentType on Windows

By overriding the type definition of SetTimer, we prevent None being passed as the callback parameter. This causes problems when integrating Bleak in larger Windows applications that call SetTimer in this way.
By defining the function type definition locally, we don't override the global definition so existing code that uses None as a callback keeps working.

To prevent other problems, we also make the definitions of KillTimer and CoGetApartmentType local.

Author: bramd

AUTHORS.rst

@@ -23,6 +23,7 @@ Contributors
 * Robbe Gaeremynck <[email protected]>
 * David Johansen <[email protected]>
 * JP Hutchins <[email protected]>
+* Bram Duvigneau <[email protected]>
 
 Sponsors
 --------

CHANGELOG.rst

@@ -10,6 +10,10 @@ and this project adheres to `Semantic Versioning <https://semver.org/spec/v2.0.0
 `Unreleased`_
 =============
 
+Changed
+-------
+* In bleak.backends.winrt.util the SetTimer, KillTimer and CoGetApartmentType functions define their own prototype and don't change ctypes' global state anymore
+
 `0.22.2`_ (2024-06-01)
 ======================
 

bleak/backends/winrt/util.py

@@ -36,24 +36,43 @@ def _check_hresult(result, func, args):
 )
 
 # https://learn.microsoft.com/en-us/windows/win32/api/winuser/nf-winuser-settimer
-_SetTimer = ctypes.windll.user32.SetTimer
-_SetTimer.restype = _UINT_PTR
-_SetTimer.argtypes = [wintypes.HWND, _UINT_PTR, wintypes.UINT, _TIMERPROC]
+_SET_TIMER_PROTOTYPE = ctypes.WINFUNCTYPE(
+    _UINT_PTR, wintypes.HWND, _UINT_PTR, wintypes.UINT, _TIMERPROC
+)
+_SET_TIMER_PARAM_FLAGS = (
+    (1, "hwnd", None),
+    (1, "nidevent"),
+    (1, "uelapse"),
+    (1, "lptimerfunc", None),
+)
+_SetTimer = _SET_TIMER_PROTOTYPE(
+    ("SetTimer", ctypes.windll.user32), _SET_TIMER_PARAM_FLAGS
+)
 _SetTimer.errcheck = _check_result
 
 # https://learn.microsoft.com/en-us/windows/win32/api/winuser/nf-winuser-killtimer
-_KillTimer = ctypes.windll.user32.KillTimer
-_KillTimer.restype = wintypes.BOOL
-_KillTimer.argtypes = [wintypes.HWND, wintypes.UINT]
-
+_KILL_TIMER_PROTOTYPE = ctypes.WINFUNCTYPE(wintypes.BOOL, wintypes.HWND, _UINT_PTR)
+_KILL_TIMER_PARAM_FLAGS = (
+    (1, "hwnd", None),
+    (1, "uidevent"),
+)
+_KillTimer = _KILL_TIMER_PROTOTYPE(
+    ("KillTimer", ctypes.windll.user32), _KILL_TIMER_PARAM_FLAGS
+)
 
 # https://learn.microsoft.com/en-us/windows/win32/api/combaseapi/nf-combaseapi-cogetapartmenttype
-_CoGetApartmentType = ctypes.windll.ole32.CoGetApartmentType
-_CoGetApartmentType.restype = ctypes.c_int
-_CoGetApartmentType.argtypes = [
+_CO_GET_APARTMENT_TYPE_PROTOTYPE = ctypes.WINFUNCTYPE(
+    ctypes.c_int,
     ctypes.POINTER(ctypes.c_int),
     ctypes.POINTER(ctypes.c_int),
-]
+)
+_CO_GET_APARTMENT_TYPE_PARAM_FLAGS = (
+    (1, "papttype", None),
+    (1, "paptqualifier", None),
+)
+_CoGetApartmentType = _CO_GET_APARTMENT_TYPE_PROTOTYPE(
+    ("CoGetApartmentType", ctypes.windll.ole32), _CO_GET_APARTMENT_TYPE_PARAM_FLAGS
+)
 _CoGetApartmentType.errcheck = _check_hresult
 
 _CO_E_NOTINITIALIZED = -2147221008