11import logging
2+ from typing import Optional
23from settings import DEBUG
34from core .widgets .base import BaseWidget
45from core .utils .win32 .windows import WinEvent
2122SystemEventListener = None
2223logging .warning ("Failed to load Win32 System Event Listener" )
2324
24-
25+ # Missing from win32con
26+ WS_EX_NOREDIRECTIONBITMAP = 0x20_0000
27+
2528class TaskbarWidget (BaseWidget ):
2629validation_schema = VALIDATION_SCHEMA
2730update_event = pyqtSignal (int ,WinEvent )
@@ -187,10 +190,8 @@ def _update_label(self, hwnd: int, win_info: dict, event: WinEvent) -> None:
187190if self ._animation ['enabled' ]:
188191self ._animate_icon (icon_label ,start_width = 0 ,end_width = icon_label .sizeHint ().width ())
189192
190- def _get_app_icon (self ,hwnd :int ,title :str ,process :dict ,event :WinEvent )-> None :
193+ def _get_app_icon (self ,hwnd :int ,title :str ,process :dict ,event :WinEvent )-> Optional [ QPixmap ] :
191194try :
192- if hwnd != win32gui .GetForegroundWindow ():
193- return
194195pid = process ["pid" ]
195196
196197if event != WinEvent .WinEventOutOfContext :
@@ -201,7 +202,7 @@ def _get_app_icon(self, hwnd: int, title:str, process: dict, event: WinEvent) ->
201202else :
202203icon_img = get_window_icon (hwnd ,self .dpi )
203204if icon_img :
204- icon_img = icon_img .resize ((int (self ._label_icon_size * self .dpi ),int (self ._label_icon_size * self .dpi )),Image .LANCZOS ).convert ("RGBA" )
205+ icon_img = icon_img .resize ((int (self ._label_icon_size * self .dpi ),int (self ._label_icon_size * self .dpi )),Image .Resampling . LANCZOS ).convert ("RGBA" )
205206else :
206207# UWP apps I hate it
207208if process ["name" ]== "ApplicationFrameHost.exe" :
@@ -211,9 +212,9 @@ def _get_app_icon(self, hwnd: int, title:str, process: dict, event: WinEvent) ->
211212return
212213else :
213214self ._update_retry_count = 0
215+ if icon_img :
214216if not DEBUG :
215217self ._icon_cache [(hwnd ,title ,pid )]= icon_img
216- if icon_img :
217218qimage = QImage (icon_img .tobytes (),icon_img .width ,icon_img .height ,QImage .Format .Format_RGBA8888 )
218219pixmap = QPixmap .fromImage (qimage )
219220else :
@@ -223,22 +224,23 @@ def _get_app_icon(self, hwnd: int, title:str, process: dict, event: WinEvent) ->
223224if DEBUG :
224225logging .exception (f"Failed to get icons for window with HWND{ hwnd } { event } )
225226
226- def get_visible_windows (self ,hwnd :int ,win_info :dict ,event :WinEvent )-> None :
227+ def get_visible_windows (self ,_ :int ,win_info :dict ,event :WinEvent )-> list [ tuple [ str , int , Optional [ QPixmap ], WinEvent ]] :
227228process = win_info ['process' ]
228229def is_window_visible_on_taskbar (hwnd ):
229- if win32gui .IsWindowVisible (hwnd )and win32gui . GetWindowText ( hwnd ) :
230+ if win32gui .IsWindowVisible (hwnd ):
230231ex_style = win32gui .GetWindowLong (hwnd ,win32con .GWL_EXSTYLE )
231- if not (ex_style & win32con .WS_EX_TOOLWINDOW ):
232+ if not (ex_style & win32con .WS_EX_TOOLWINDOW or ex_style == WS_EX_NOREDIRECTIONBITMAP ):
232233return True
233234return False
234235
235236visible_windows = []
236237
237- def enum_windows_proc (hwnd ,lParam ):
238+ def enum_windows_proc (hwnd ,_ ):
238239if is_window_visible_on_taskbar (hwnd ):
239240title = win32gui .GetWindowText (hwnd )
240241icon = self ._get_app_icon (hwnd ,title ,process ,event )
241- visible_windows .append ((title ,hwnd ,icon ,process ))
242+ if title and icon :
243+ visible_windows .append ((title ,hwnd ,icon ,process ))
242244return True
243245win32gui .EnumWindows (enum_windows_proc ,None )
244246return visible_windows