75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
|
import win32gui
|
||
|
|
import win32con
|
||
|
|
import win32api
|
||
|
|
import threading
|
||
|
|
from typing import Callable
|
||
|
|
|
||
|
|
|
||
|
|
def get_active_window() -> str:
|
||
|
|
"""Get the currently active window title on Windows"""
|
||
|
|
try:
|
||
|
|
hwnd = win32gui.GetForegroundWindow()
|
||
|
|
return win32gui.GetWindowText(hwnd) or "Unknown Window"
|
||
|
|
except:
|
||
|
|
return "Error getting window title"
|
||
|
|
|
||
|
|
|
||
|
|
def register_session_listener(callback: Callable[[str], None]) -> None:
|
||
|
|
"""Register a callback for Windows session lock/unlock events"""
|
||
|
|
|
||
|
|
def session_event_handler(hwnd, msg, wparam, lparam):
|
||
|
|
if msg == win32con.WM_WTSSESSION_CHANGE:
|
||
|
|
if wparam == win32con.WTS_SESSION_LOCK:
|
||
|
|
callback("locked")
|
||
|
|
elif wparam == win32con.WTS_SESSION_UNLOCK:
|
||
|
|
callback("unlocked")
|
||
|
|
return win32gui.DefWindowProc(hwnd, msg, wparam, lparam)
|
||
|
|
|
||
|
|
def run_message_loop():
|
||
|
|
hinst = win32api.GetModuleHandle(None)
|
||
|
|
wndclass = win32gui.WNDCLASS()
|
||
|
|
wndclass.hInstance = hinst
|
||
|
|
wndclass.lpszClassName = "SessionWatcher"
|
||
|
|
wndclass.lpfnWndProc = session_event_handler
|
||
|
|
|
||
|
|
try:
|
||
|
|
win32gui.RegisterClass(wndclass)
|
||
|
|
except:
|
||
|
|
pass # Already registered
|
||
|
|
|
||
|
|
hwnd = win32gui.CreateWindow(
|
||
|
|
wndclass.lpszClassName,
|
||
|
|
"Session Watcher",
|
||
|
|
0, 0, 0, 0, 0,
|
||
|
|
0, hinst, None
|
||
|
|
)
|
||
|
|
|
||
|
|
# Register for session notifications
|
||
|
|
try:
|
||
|
|
from win32ts import WTSRegisterSessionNotification
|
||
|
|
WTSRegisterSessionNotification(hwnd, 1) # NOTIFY_FOR_THIS_SESSION
|
||
|
|
except ImportError:
|
||
|
|
# pywin32 might not have win32ts on all systems
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Start message loop
|
||
|
|
win32gui.PumpMessages()
|
||
|
|
|
||
|
|
# Run in a separate thread
|
||
|
|
thread = threading.Thread(target=run_message_loop, daemon=True)
|
||
|
|
thread.start()
|
||
|
|
|
||
|
|
|
||
|
|
def get_idle_time() -> int:
|
||
|
|
"""Get the system idle time in seconds on Windows"""
|
||
|
|
from ctypes import Structure, windll, c_uint, sizeof, byref
|
||
|
|
|
||
|
|
class LASTINPUTINFO(Structure):
|
||
|
|
_fields_ = [('cbSize', c_uint), ('dwTime', c_uint)]
|
||
|
|
|
||
|
|
lastInputInfo = LASTINPUTINFO()
|
||
|
|
lastInputInfo.cbSize = sizeof(lastInputInfo)
|
||
|
|
windll.user32.GetLastInputInfo(byref(lastInputInfo))
|
||
|
|
|
||
|
|
millis = windll.kernel32.GetTickCount() - lastInputInfo.dwTime
|
||
|
|
return millis // 1000 # Convert milliseconds to seconds
|