import ctypes, time
from ctypes import wintypes
import win32gui
from PIL import ImageGrab

user32 = ctypes.windll.user32
hdesk = user32.OpenInputDesktop(0, False, 0x01FF)
if hdesk: user32.SetThreadDesktop(hdesk)

for pid, hwnd in [(30076, 0xe207f6), (29956, 0x60b66)]:
    if user32.IsWindow(hwnd):
        user32.SetForegroundWindow(hwnd)
        time.sleep(0.3)
        rect = win32gui.GetWindowRect(hwnd)
        if rect[2] > rect[0] and rect[3] > rect[1]:
            img = ImageGrab.grab(bbox=rect)
            img.save(rf"C:\Projects\JX\logs\game_inspect_{pid}.png")
            print(f"Saved game_inspect_{pid}.png")

with open(r'c:\Projects\JX\logs\inspect_other_done.txt', 'w', encoding='utf-8') as f:
    f.write("Done\n")
