import os import time import shutil import win32ui import win32gui import win32api import win32con import win32process import win32clipboard import pyperclip import ctypes from PIL import Image from wxauto import uiautomation as uia def GetAllWindows(): """ 获取所有窗口的信息，返回一个列表，每个元素包含 (窗口句柄, 类名, 窗口标题) """ windows = [] def enum_windows_proc(hwnd, extra): class_name = win32gui.GetClassName(hwnd) # 获取窗口类名 window_title = win32gui.GetWindowText(hwnd) # 获取窗口标题 windows.append((hwnd, class_name, window_title)) win32gui.EnumWindows(enum_windows_proc, None) return windows def GetCursorWindow(): x, y = win32api.GetCursorPos() hwnd = win32gui.WindowFromPoint((x, y)) window_title = win32gui.GetWindowText(hwnd) class_name = win32gui.GetClassName(hwnd) return hwnd, window_title, class_name def set_cursor_pos(x, y): win32api.SetCursorPos((x, y)) def Click(rect): x = (rect.left + rect.right) // 2 y = (rect.top + rect.bottom) // 2 set_cursor_pos(x, y) win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, x, y, 0, 0) win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, x, y, 0, 0) def GetPathByHwnd(hwnd): """通过窗口句柄获取进程可执行文件路径 - 使用pywin32""" try: thread_id, process_id = win32process.GetWindowThreadProcessId(hwnd) # 获取进程句柄 PROCESS_QUERY_INFORMATION = 0x0400 PROCESS_VM_READ = 0x0010 process_handle = win32api.OpenProcess( PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, False, process_id ) # 获取可执行文件路径 exe_path = win32process.GetModuleFileNameEx(process_handle, 0) # 关闭句柄 win32api.CloseHandle(process_handle) return exe_path except Exception as e: print(f"Error: {e}") return None def GetVersionByPath(file_path): try: info = win32api.GetFileVersionInfo(file_path, '\\') version = "{}.{}.{}.{}".format(win32api.HIWORD(info['FileVersionMS']), win32api.LOWORD(info['FileVersionMS']), win32api.HIWORD(info['FileVersionLS']), win32api.LOWORD(info['FileVersionLS'])) except: version = None return version def GetWindowRect(hwnd): return win32gui.GetWindowRect(hwnd) def capture(hwnd, bbox): # 获取窗口的屏幕坐标 window_rect = win32gui.GetWindowRect(hwnd) win_left, win_top, win_right, win_bottom = window_rect win_width = win_right - win_left win_height = win_bottom - win_top # 获取窗口的设备上下文 hwndDC = win32gui.GetWindowDC(hwnd) mfcDC = win32ui.CreateDCFromHandle(hwndDC) saveDC = mfcDC.CreateCompatibleDC() # 创建位图对象保存整个窗口截图 saveBitMap = win32ui.CreateBitmap() saveBitMap.CreateCompatibleBitmap(mfcDC, win_width, win_height) saveDC.SelectObject(saveBitMap) # 使用PrintWindow捕获整个窗口（包括被遮挡或最小化的窗口） result = ctypes.windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 3) # 转换为PIL图像 bmpinfo = saveBitMap.GetInfo() bmpstr = saveBitMap.GetBitmapBits(True) im = Image.frombuffer( 'RGB', (bmpinfo['bmWidth'], bmpinfo['bmHeight']), bmpstr, 'raw', 'BGRX', 0, 1) # 释放资源 win32gui.DeleteObject(saveBitMap.GetHandle()) saveDC.DeleteDC() mfcDC.DeleteDC() win32gui.ReleaseDC(hwnd, hwndDC) # 计算bbox相对于窗口左上角的坐标 bbox_left, bbox_top, bbox_right, bbox_bottom = bbox # 转换为截图图像中的相对坐标 crop_left = bbox_left - win_left crop_top = bbox_top - win_top crop_right = bbox_right - win_left crop_bottom = bbox_bottom - win_top # 裁剪目标区域 cropped_im = im.crop((crop_left, crop_top, crop_right, crop_bottom)) return cropped_im def GetText(HWND): length = win32gui.SendMessage(HWND, win32con.WM_GETTEXTLENGTH)*2 buffer = win32gui.PyMakeBuffer(length) win32api.SendMessage(HWND, win32con.WM_GETTEXT, length, buffer) address, length_ = win32gui.PyGetBufferAddressAndLen(buffer[:-1]) text = win32gui.PyGetString(address, length_)[:int(length/2)] buffer.release() return text def GetAllWindowExs(HWND): if not HWND: return handles = [] win32gui.EnumChildWindows( HWND, lambda hwnd, param: param.append([hwnd, win32gui.GetClassName(hwnd), GetText(hwnd)]), handles) return handles def FindWindow(classname=None, name=None, timeout=0) -> int: t0 = time.time() while True: HWND = win32gui.FindWindow(classname, name) if HWND: break if time.time() - t0 > timeout: break time.sleep(0.01) return HWND def FindTopLevelControl(classname=None, name=None, timeout=3): hwnd = FindWindow(classname, name, timeout) if hwnd: return uia.ControlFromHandle(hwnd) else: return None def FindWinEx(HWND, classname=None, name=None) -> list: hwnds_classname = [] hwnds_name = [] def find_classname(hwnd, classname): classname_ = win32gui.GetClassName(hwnd) if classname_ == classname: if hwnd not in hwnds_classname: hwnds_classname.append(hwnd) def find_name(hwnd, name): name_ = GetText(hwnd) if name in name_: if hwnd not in hwnds_name: hwnds_name.append(hwnd) if classname: win32gui.EnumChildWindows(HWND, find_classname, classname) if name: win32gui.EnumChildWindows(HWND, find_name, name) if classname and name: hwnds = [hwnd for hwnd in hwnds_classname if hwnd in hwnds_name] else: hwnds = hwnds_classname + hwnds_name return hwnds def ClipboardFormats(unit=0, *units): units = list(units) retry_count = 5 while retry_count > 0: try: win32clipboard.OpenClipboard() try: u = win32clipboard.EnumClipboardFormats(unit) finally: win32clipboard.CloseClipboard() break except Exception as e: retry_count -= 1 units.append(u) if u: units = ClipboardFormats(u, *units) return units def ReadClipboardData(): Dict = {} formats = ClipboardFormats() for i in formats: if i == 0: continue retry_count = 5 while retry_count > 0: try: win32clipboard.OpenClipboard() try: data = win32clipboard.GetClipboardData(i) Dict[str(i)] = data finally: win32clipboard.CloseClipboard() break except Exception as e: retry_count -= 1 return Dict def SetClipboardText(text: str): pyperclip.copy(text) class DROPFILES(ctypes.Structure): _fields_ = [ ("pFiles", ctypes.c_uint), ("x", ctypes.c_long), ("y", ctypes.c_long), ("fNC", ctypes.c_int), ("fWide", ctypes.c_bool), ] pDropFiles = DROPFILES() pDropFiles.pFiles = ctypes.sizeof(DROPFILES) pDropFiles.fWide = True matedata = bytes(pDropFiles) def SetClipboardFiles(paths): for file in paths: if not os.path.exists(file): raise FileNotFoundError(f"file ({file}) not exists!") files = ("\0".join(paths)).replace("/", "\\") data = files.encode("U16")[2:]+b"\0\0" t0 = time.time() while True: if time.time() - t0 > 10: raise TimeoutError(f"设置剪贴板文件超时！ --> {paths}") try: win32clipboard.OpenClipboard() win32clipboard.EmptyClipboard() win32clipboard.SetClipboardData(win32clipboard.CF_HDROP, matedata+data) break except: pass finally: try: win32clipboard.CloseClipboard() except: pass def PasteFile(folder): folder = os.path.realpath(folder) if not os.path.exists(folder): os.makedirs(folder) t0 = time.time() while True: if time.time() - t0 > 10: raise TimeoutError(f"读取剪贴板文件超时！") try: win32clipboard.OpenClipboard() if win32clipboard.IsClipboardFormatAvailable(win32clipboard.CF_HDROP): files = win32clipboard.GetClipboardData(win32clipboard.CF_HDROP) for file in files: filename = os.path.basename(file) dest_file = os.path.join(folder, filename) shutil.copy2(file, dest_file) return True else: print("剪贴板中没有文件") return False except: pass finally: win32clipboard.CloseClipboard() def IsRedPixel(uicontrol): rect = uicontrol.BoundingRectangle hwnd = uicontrol.GetAncestorControl(lambda x,y:x.ClassName=='WeChatMainWndForPC').NativeWindowHandle bbox = (rect.left, rect.top, rect.right, rect.bottom) img = capture(hwnd, bbox) return any(p[0] > p[1] and p[0] > p[2] for p in img.getdata())