Store the window procs inside dictionaries instead of dinamically created global variables for each window

Author: Valer100

hPyT/hPyT.py

@@ -95,6 +95,8 @@ class NCCALCSIZE_PARAMS(ctypes.Structure):
 rnbbcs: List[int] = []
 accent_color_titlebars: List[int] = []
 accent_color_borders: List[int] = []
+old_wndprocs: Dict[int, int] = {}
+new_wndprocs: Dict[int, int] = {}
 
 WINDOWS_VERSION = float(platform.version().split(".")[0])
 
@@ -130,11 +132,10 @@ def handle(hwnd: int, msg: int, wp: int, lp: int) -> int:
 
             # Default processing for other messages
             return call_window_proc(
-                *map(ctypes.c_uint64, (globals()[old], hwnd, msg, wp, lp))
+                *map(ctypes.c_uint64, (old_wndprocs[hwnd], hwnd, msg, wp, lp))
             )
 
         hwnd: int = module_find(window)
-        old, new = f"old_wndproc_{hwnd}", f"new_wndproc_{hwnd}"
 
         prototype = ctypes.WINFUNCTYPE(
             ctypes.c_uint64,
@@ -160,13 +161,13 @@ def handle(hwnd: int, msg: int, wp: int, lp: int) -> int:
         title_bar_height: int = full_height - client_height - border_width
 
         # Override the window procedure if not already done
-        if globals().get(old) is None:
-            globals()[old] = get_window_long(hwnd, GWL_WNDPROC)
+        if old_wndprocs.get(hwnd, 0) == 0:
+            old_wndprocs[hwnd] = get_window_long(hwnd, GWL_WNDPROC)
 
         # Do not remove the top border when on windows 7 or lower
         if WINDOWS_VERSION >= 10.0:  # Windows 10 is version 10.0
-            globals()[new] = prototype(handle)
-            set_window_long(hwnd, GWL_WNDPROC, globals()[new])
+            new_wndprocs[hwnd] = prototype(handle)
+            set_window_long(hwnd, GWL_WNDPROC, new_wndprocs[hwnd])
 
         old_style = get_window_long(hwnd, GWL_STYLE)
         new_style = (old_style & ~WS_CAPTION) | WS_BORDER
@@ -206,10 +207,10 @@ def unhide(cls, window: Any) -> None:
 
         hwnd: int = module_find(window)
 
-        if globals().get(f"old_wndproc_{hwnd}") is not None:
-            set_window_long(hwnd, GWL_WNDPROC, globals()[f"old_wndproc_{hwnd}"])
-            globals()[f"old_wndproc_{hwnd}"] = None
-            globals()[f"new_wndproc_{hwnd}"] = None
+        if old_wndprocs.get(hwnd, 0) != 0:
+            set_window_long(hwnd, GWL_WNDPROC, old_wndprocs[hwnd])
+            old_wndprocs.pop(hwnd)
+            new_wndprocs.pop(hwnd)
 
         # Restore the original height if no_span was used
         height_reduction: int = cls._height_reduction.pop(hwnd, 0)