308 lines
11 KiB
Python
308 lines
11 KiB
Python
import pyautogui
|
||
import time
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
import signal
|
||
import psutil
|
||
from PIL import Image, ImageGrab
|
||
import win32gui
|
||
import win32con
|
||
import win32process
|
||
|
||
|
||
def kill_process_by_name(process_name):
|
||
"""终止指定名称的进程"""
|
||
killed = False
|
||
for proc in psutil.process_iter(['pid', 'name']):
|
||
try:
|
||
if process_name.lower() in proc.info['name'].lower():
|
||
print(f"正在终止进程: {proc.info['name']} (PID: {proc.info['pid']})")
|
||
try:
|
||
process = psutil.Process(proc.info['pid'])
|
||
process.terminate()
|
||
killed = True
|
||
except Exception as e:
|
||
print(f"终止进程时出错: {e}")
|
||
except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
|
||
pass
|
||
return killed
|
||
|
||
|
||
def kill_current_python_process():
|
||
"""结束当前Python程序"""
|
||
current_pid = os.getpid()
|
||
parent_pid = psutil.Process(current_pid).ppid()
|
||
print(f"当前Python进程PID: {current_pid}, 父进程PID: {parent_pid}")
|
||
|
||
# 收集需要在退出前结束的进程
|
||
pids_to_kill = []
|
||
|
||
# 查找所有可能与当前进程相关的Python进程
|
||
for proc in psutil.process_iter(['pid', 'name', 'cmdline']):
|
||
try:
|
||
# 如果是Python进程并且不是当前进程
|
||
if ('python' in proc.info['name'].lower() and
|
||
proc.info['pid'] != current_pid and
|
||
proc.info['pid'] != parent_pid):
|
||
# 检查命令行参数，看是否与当前脚本相关
|
||
cmdline = ' '.join(proc.info['cmdline']).lower()
|
||
script_name = os.path.basename(__file__).lower()
|
||
if script_name in cmdline or "win_click.py" in cmdline:
|
||
pids_to_kill.append(proc.info['pid'])
|
||
print(f"将结束Python进程: {proc.info['name']} (PID: {proc.info['pid']})")
|
||
except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
|
||
pass
|
||
|
||
# 注册程序退出时要杀死的进程
|
||
def on_exit():
|
||
for pid in pids_to_kill:
|
||
try:
|
||
os.kill(pid, signal.SIGTERM)
|
||
except:
|
||
pass
|
||
|
||
# 返回退出函数，由调用者决定何时执行
|
||
return on_exit
|
||
|
||
|
||
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(max_retries=3, retry_interval=5):
|
||
"""查找并点击微信登录按钮
|
||
|
||
Args:
|
||
max_retries: 最大重试次数
|
||
retry_interval: 重试间隔(秒)
|
||
"""
|
||
for attempt in range(max_retries):
|
||
# 切换到微信窗口
|
||
if not find_wechat_window():
|
||
print(f"尝试 {attempt + 1}/{max_retries}: 未找到微信窗口，等待 {retry_interval} 秒后重试...")
|
||
time.sleep(retry_interval)
|
||
continue
|
||
|
||
# 等待窗口完全加载
|
||
time.sleep(2)
|
||
|
||
try:
|
||
print(f"尝试 {attempt + 1}/{max_retries}: 查找登录按钮...")
|
||
# 截取屏幕
|
||
screenshot = ImageGrab.grab()
|
||
|
||
# 创建一个绿色区域的掩码
|
||
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("点击操作完成")
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"尝试 {attempt + 1}/{max_retries} 发生错误: {e}")
|
||
if attempt < max_retries - 1:
|
||
print(f"等待 {retry_interval} 秒后重试...")
|
||
time.sleep(retry_interval)
|
||
|
||
print(f"经过 {max_retries} 次尝试后仍未成功点击登录按钮")
|
||
return False
|
||
|
||
|
||
def update_system(wait_time=15):
|
||
"""完整的系统更新流程
|
||
|
||
Args:
|
||
wait_time: 等待微信启动的秒数，默认15秒
|
||
"""
|
||
print("=== 开始系统更新流程 ===")
|
||
|
||
# 步骤1: 结束微信进程
|
||
print("步骤1: 正在结束微信进程...")
|
||
if kill_process_by_name("WeChat"):
|
||
print("微信进程已终止")
|
||
else:
|
||
print("未找到运行中的微信进程")
|
||
|
||
# 步骤2: 注册结束当前Python进程的函数
|
||
exit_handler = kill_current_python_process()
|
||
|
||
# 步骤3: 准备重启系统
|
||
print("步骤3: 准备重启系统...")
|
||
|
||
# 查找bot_start.bat文件
|
||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||
bat_path = os.path.join(current_dir, "bot_start.bat")
|
||
|
||
if not os.path.exists(bat_path):
|
||
# 向上查找一级目录
|
||
parent_dir = os.path.dirname(current_dir)
|
||
bat_path = os.path.join(parent_dir, "bot_start.bat")
|
||
|
||
if os.path.exists(bat_path):
|
||
print(f"找到启动脚本: {bat_path}")
|
||
|
||
# 创建一个临时的启动脚本，用于在当前进程结束后启动系统
|
||
temp_bat = os.path.join(os.environ.get('TEMP', os.getcwd()), "restart_system.bat")
|
||
with open(temp_bat, "w") as f:
|
||
f.write(f"""@echo off
|
||
echo 等待旧进程结束...
|
||
timeout /t 2 /nobreak > nul
|
||
echo 重新启动系统...
|
||
start "" "{bat_path}"
|
||
echo 等待系统启动和微信加载 ({wait_time}秒)...
|
||
timeout /t {wait_time} /nobreak > nul
|
||
echo 尝试登录微信...
|
||
start "" "python" "{os.path.abspath(__file__)}" --login-only
|
||
exit
|
||
""")
|
||
|
||
# 启动临时脚本
|
||
print("启动临时脚本执行重启...")
|
||
# 使用subprocess.Popen而不是run，这样不会等待它完成
|
||
subprocess.Popen(["cmd", "/c", temp_bat],
|
||
shell=True,
|
||
creationflags=subprocess.CREATE_NEW_CONSOLE)
|
||
|
||
# 等待一小段时间确保脚本开始运行
|
||
time.sleep(1)
|
||
|
||
# 结束当前Python相关进程并退出
|
||
print("结束当前进程...")
|
||
exit_handler()
|
||
sys.exit(0)
|
||
else:
|
||
print(f"未找到启动脚本 bot_start.bat，无法自动重启系统")
|
||
return False
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 检查是否只需要执行登录
|
||
if "--login-only" in sys.argv:
|
||
print("仅执行微信登录...")
|
||
find_and_click_wechat_login()
|
||
else:
|
||
# 显示选项菜单
|
||
print("==== 系统工具 ====")
|
||
print("1. 查找并点击微信登录")
|
||
print("2. 更新系统 (结束微信、更新代码、重启系统、自动登录)")
|
||
print("0. 退出")
|
||
|
||
try:
|
||
choice = input("请选择操作 [0-2]: ")
|
||
if choice == "1":
|
||
find_and_click_wechat_login()
|
||
elif choice == "2":
|
||
update_system()
|
||
else:
|
||
print("退出程序")
|
||
except KeyboardInterrupt:
|
||
print("\n程序被中断")
|
||
except Exception as e:
|
||
print(f"发生错误: {e}")
|