自动点击事件

2025-03-31 11:42:55 +08:00
parent aa75d55d85
commit 4639e6d685
1 changed files with 164 additions and 0 deletions
--- a/win_click.py
+++ b/win_click.py
@@ -0,0 +1,164 @@
+import pyautogui
+import time
+import os
+import numpy as np
+from PIL import Image, ImageGrab
+import win32gui
+import win32con
+import win32process
+import psutil
+
+def find_wechat_window():
+    """查找微信窗口并将其激活"""
+    # 方法1: 通过窗口标题查找
+    wechat_window = win32gui.FindWindow(None, "微信")
+    if wechat_window == 0:
+        # 备用方法: 尝试查找部分标题匹配的窗口
+        def callback(hwnd, windows):
+            if win32gui.IsWindowVisible(hwnd) and "微信" in win32gui.GetWindowText(hwnd):
+                windows.append(hwnd)
+            return True
+        
+        windows = []
+        win32gui.EnumWindows(callback, windows)
+        
+        if windows:
+            wechat_window = windows[0]
+    
+    # 方法2: 如果窗口标题查找失败，通过进程名查找
+    if wechat_window == 0:
+        print("通过窗口标题未找到微信，尝试通过进程查找...")
+        for proc in psutil.process_iter(['pid', 'name']):
+            if proc.info['name'] and 'WeChat' in proc.info['name']:
+                pid = proc.info['pid']
+                # 查找与此PID关联的窗口
+                def enum_windows_callback(hwnd, target_pid):
+                    _, found_pid = win32process.GetWindowThreadProcessId(hwnd)
+                    if found_pid == target_pid and win32gui.IsWindowVisible(hwnd):
+                        target_pid.append(hwnd)
+                    return True
+                
+                target_hwnds = []
+                win32gui.EnumWindows(lambda hwnd, param: enum_windows_callback(hwnd, target_hwnds), pid)
+                
+                if target_hwnds:
+                    wechat_window = target_hwnds[0]
+                    break
+    
+    if wechat_window != 0:
+        print(f"找到微信窗口: {win32gui.GetWindowText(wechat_window)}")
+        # 还原最小化的窗口
+        if win32gui.IsIconic(wechat_window):
+            win32gui.ShowWindow(wechat_window, win32con.SW_RESTORE)
+        # 激活并前置窗口
+        win32gui.SetForegroundWindow(wechat_window)
+        return True
+    else:
+        print("未找到微信窗口")
+        return False
+
+def is_green_pixel(r, g, b):
+    """判断像素是否为绿色"""
+    # 微信的绿色按钮大约是 RGB(7, 193, 96)
+    return g > 150 and g > r*1.5 and g > b*1.5
+
+def find_and_click_wechat_login():
+    """查找并点击微信登录按钮"""
+    # 切换到微信窗口
+    if not find_wechat_window():
+        print("无法找到微信窗口，正在尝试启动微信...")
+        try:
+            # 尝试启动微信（常见安装路径）
+            wechat_paths = [
+                r"C:\Program Files (x86)\Tencent\WeChat\WeChat.exe",
+                r"C:\Program Files\Tencent\WeChat\WeChat.exe",
+                r"D:\Program Files (x86)\Tencent\WeChat\WeChat.exe",
+                r"D:\Program Files\Tencent\WeChat\WeChat.exe"
+            ]
+            
+            for path in wechat_paths:
+                if os.path.exists(path):
+                    os.startfile(path)
+                    print(f"正在启动微信: {path}")
+                    time.sleep(5)  # 等待微信启动
+                    if find_wechat_window():
+                        break
+            else:
+                print("未找到微信安装路径，请手动启动微信")
+                return
+        except Exception as e:
+            print(f"启动微信时出错: {e}")
+            return
+    
+    # 等待窗口完全加载
+    time.sleep(2)
+    
+    try:
+        print("尝试查找登录按钮...")
+        # 截取屏幕
+        screenshot = ImageGrab.grab()
+        screenshot_np = np.array(screenshot)
+        
+        # 创建一个绿色区域的掩码
+        green_areas = []
+        width, height = screenshot.size
+        
+        # 分析图像，识别绿色区域
+        for y in range(0, height, 5):  # 每5个像素采样一次以提高效率
+            for x in range(0, width, 5):
+                r, g, b = screenshot.getpixel((x, y))
+                if is_green_pixel(r, g, b):
+                    # 发现绿色像素，向四周扩散检查是否为按钮
+                    left, top, right, bottom = x, y, x, y
+                    # 向右扩散
+                    for nx in range(x, min(x+300, width)):
+                        r, g, b = screenshot.getpixel((nx, y))
+                        if not is_green_pixel(r, g, b):
+                            break
+                        right = nx
+                    # 向下扩散
+                    for ny in range(y, min(y+60, height)):
+                        r, g, b = screenshot.getpixel((x, ny))
+                        if not is_green_pixel(r, g, b):
+                            break
+                        bottom = ny
+                    
+                    width_area = right - left
+                    height_area = bottom - top
+                    
+                    # 如果区域符合按钮尺寸
+                    if 100 < width_area < 300 and 30 < height_area < 60:
+                        green_areas.append((left, top, right, bottom))
+        
+        login_button_found = False
+        for left, top, right, bottom in green_areas:
+            # 点击按钮中心
+            center_x = (left + right) // 2
+            center_y = (top + bottom) // 2
+            pyautogui.click(center_x, center_y)
+            print(f"已点击位置: ({center_x}, {center_y})")
+            login_button_found = True
+            break
+        
+        # 方法2：如果图像识别失败，尝试使用固定位置
+        if not login_button_found:
+            print("未通过图像识别找到登录按钮，尝试使用备用方法...")
+            
+            # 获取屏幕分辨率
+            screen_width, screen_height = pyautogui.size()
+            
+            # 估计登录按钮位置 (通常在窗口中下部偏右)
+            button_x = screen_width // 2  # 水平中心
+            button_y = (screen_height // 2) + 100  # 垂直中心偏下
+            
+            # 移动到估计位置并点击
+            pyautogui.click(button_x, button_y)
+            print(f"已点击估计位置: ({button_x}, {button_y})")
+        
+        print("点击操作完成")
+        
+    except Exception as e:
+        print(f"发生错误: {e}")
+
+if __name__ == "__main__":
+    find_and_click_wechat_login()