ED/EDMarketConnector
1
0
Fork 0
mirror of https://github.com/EDCD/EDMarketConnector.git synced 2025-04-24 12:40:52 +03:00
Code Issues Projects Releases Wiki Activity

monitor: Replace EnumWindows() with pywin32 version

Browse Source
This commit is contained in:
Athanasius 2023-01-12 17:50:52 +00:00
parent 2c469441e0
commit b24e998931
No known key found for this signature in database
GPG Key ID: 772697E181BB2767
1 changed files with 9 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
monitor.py
View File

@ -50,22 +50,12 @@ if sys.platform == 'darwin':
elif sys.platform == 'win32':
import ctypes
from ctypes import WINFUNCTYPE, windll
from ctypes.wintypes import BOOL, HANDLE, HWND, LPARAM
from ctypes.wintypes import BOOL, HANDLE
import win32gui
from watchdog.events import FileCreatedEvent, FileSystemEventHandler
from watchdog.observers import Observer
# <https://learn.microsoft.com/en-us/windows/win32/api/winuser/nf-winuser-enumwindows>
# BOOL EnumWindows(
# [in] WNDENUMPROC lpEnumFunc,
# [in] LPARAM lParam
# );
EnumWindowsProc = WINFUNCTYPE(BOOL, HWND, LPARAM)
prototype = WINFUNCTYPE(BOOL, EnumWindowsProc, LPARAM)
paramflags_enumwindows = (1, "lpEnumFunc"), (1, "lParam")
EnumWindows = prototype(("EnumWindows", windll.user32), paramflags_enumwindows)
# <https://learn.microsoft.com/en-us/windows/win32/api/handleapi/nf-handleapi-closehandle>
# BOOL CloseHandle(
# [in] HANDLE hObject
@ -2047,17 +2037,19 @@ class EDLogs(FileSystemEventHandler): # type: ignore # See below
return None
def callback(hWnd, lParam): # noqa: N803
name = WindowTitle(hWnd)
def callback(hwnd, hwnds):
name = WindowTitle(hwnd)
if name and name.startswith('Elite - Dangerous'):
handle = GetProcessHandleFromHwnd(hWnd)
handle = GetProcessHandleFromHwnd(hwnd)
if handle: # If GetProcessHandleFromHwnd succeeds then the app is already running as this user
CloseHandle(handle)
return False # stop enumeration
hwnds.append(hwnd)
return True
return not EnumWindows(EnumWindowsProc(callback), 0)
# Ref: <http://timgolden.me.uk/python/win32_how_do_i/find-the-window-for-my-subprocess.html>
ed_windows: list[int] = []
win32gui.EnumWindows(callback, ed_windows)
return True if len(ed_windows) > 0 else False
return False