168 lines
6.4 KiB
Python
168 lines
6.4 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
自动控制 GUI 界面脚本 - 通过模拟用户操作设置目标压力
|
||
|
||
升级功能:
|
||
1. 加入坐标校准功能，摆脱写死的硬编码坐标
|
||
2. 自动寻找并置顶 GUI 窗口
|
||
3. 加入 PyAutoGUI 故障保护 (防失控)
|
||
|
||
使用方法:
|
||
1. 首次使用建议进行校准: python auto_test.py --calibrate --targets 50 80 100
|
||
2. 后续固定窗口位置后直接运行: python auto_test.py --targets 50 80 100
|
||
python tool/auto_test.py --calibrate --targets 50 80 100 180 170 130 200 210 270 290 280 250 175 165 100 45
|
||
"""
|
||
|
||
import argparse
|
||
import time
|
||
import platform
|
||
import pyautogui
|
||
|
||
try:
|
||
import pygetwindow as gw
|
||
except ImportError:
|
||
gw = None
|
||
|
||
# 配置 PyAutoGUI
|
||
pyautogui.FAILSAFE = True # 将鼠标移动到屏幕四个角落可紧急停止脚本
|
||
pyautogui.PAUSE = 0.3 # 每个动作后默认停顿 0.3 秒，让 UI 有时间反应
|
||
|
||
# 平台相关的全选快捷键：macOS 用 command，Windows/Linux 用 ctrl
|
||
_MODIFIER_KEY = 'command' if platform.system() == 'Darwin' else 'ctrl'
|
||
|
||
|
||
class GUIController:
|
||
def __init__(self):
|
||
# 默认坐标 (如果不使用 calibrate 模式，将使用这些备用坐标)
|
||
# 注意：这些默认值是错误的，请务必使用 --calibrate 参数校准
|
||
self.input_x, self.input_y = 200, 150
|
||
self.btn_x, self.btn_y = 320, 150
|
||
|
||
def activate_window(self, title_keyword="ReinLoop"):
|
||
"""尝试寻找并激活目标窗口（支持部分标题匹配）"""
|
||
if gw is None:
|
||
print("⚠️ 未安装 pygetwindow，请手动确保 GUI 窗口在前台。")
|
||
print(" 安装命令: pip install pygetwindow")
|
||
return
|
||
|
||
print(f"正在寻找包含 '{title_keyword}' 的窗口...")
|
||
try:
|
||
windows = gw.getWindowsWithTitle(title_keyword)
|
||
if windows:
|
||
win = windows[0]
|
||
if win.isMinimized:
|
||
win.restore()
|
||
win.activate()
|
||
print(f"✅ 成功激活窗口: {win.title}")
|
||
time.sleep(1) # 等待窗口彻底弹出
|
||
else:
|
||
print(f"⚠️ 未找到包含 '{title_keyword}' 的窗口。")
|
||
print(f" 当前所有窗口列表:")
|
||
all_wins = gw.getAllWindows()
|
||
for w in all_wins:
|
||
if w.title.strip():
|
||
print(f" - {w.title}")
|
||
print(" 请确保 ReinLoop GUI 已打开，或使用 --calibrate 后手动置顶窗口。")
|
||
except Exception as e:
|
||
print(f"⚠️ 窗口激活失败: {e}，请手动将窗口切换到前台。")
|
||
|
||
def calibrate(self):
|
||
"""交互式坐标校准，动态获取按钮位置"""
|
||
print("\n" + "=" * 40)
|
||
print("🔧 进入坐标校准模式 (请不要切走窗口)")
|
||
print("=" * 40)
|
||
|
||
print("\n👉 请在 5 秒内将鼠标光标移动到【目标压力输入框】中心...")
|
||
for i in range(5, 0, -1):
|
||
print(f"\r倒计时: {i} 秒", end='')
|
||
time.sleep(1)
|
||
self.input_x, self.input_y = pyautogui.position()
|
||
print(f"\n✅ 输入框坐标已记录: ({self.input_x}, {self.input_y})")
|
||
|
||
print("\n👉 请在 5 秒内将鼠标光标移动到【设置目标】按钮中心...")
|
||
for i in range(5, 0, -1):
|
||
print(f"\r倒计时: {i} 秒", end='')
|
||
time.sleep(1)
|
||
self.btn_x, self.btn_y = pyautogui.position()
|
||
print(f"\n✅ 按钮坐标已记录: ({self.btn_x}, {self.btn_y})")
|
||
print("=" * 40 + "\n")
|
||
|
||
def set_target_pressure(self, target):
|
||
"""模拟用户操作设置目标压力"""
|
||
print(f"▶ 正在设置目标压力: {target} kPa")
|
||
try:
|
||
# 点击输入框
|
||
pyautogui.click(x=self.input_x, y=self.input_y)
|
||
|
||
# 全选并删除现有内容（macOS: command+a, Windows/Linux: ctrl+a）
|
||
pyautogui.hotkey(_MODIFIER_KEY, 'a')
|
||
pyautogui.press('backspace')
|
||
|
||
# 输入新的目标压力值
|
||
pyautogui.typewrite(str(target))
|
||
|
||
# 点击"设置目标"按钮
|
||
pyautogui.click(x=self.btn_x, y=self.btn_y)
|
||
|
||
print(f"✅ 成功设置目标压力: {target} kPa")
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ 设置目标压力失败: {e}")
|
||
return False
|
||
|
||
|
||
def auto_control(targets, interval, do_calibrate):
|
||
print("=" * 60)
|
||
print("🤖 GUI 自动控制脚本启动")
|
||
print("提示: 运行过程中将鼠标移动到屏幕四个角落即可紧急停止")
|
||
print("=" * 60)
|
||
|
||
controller = GUIController()
|
||
controller.activate_window()
|
||
|
||
if do_calibrate:
|
||
controller.calibrate()
|
||
else:
|
||
print(
|
||
f"ℹ️ 使用默认坐标 (输入框: {controller.input_x},{controller.input_y} | "
|
||
f"按钮: {controller.btn_x},{controller.btn_y})")
|
||
print("⚠️ 如果点击位置不准确，请使用 --calibrate 参数运行脚本。")
|
||
|
||
print("\n3秒后开始自动控制序列...")
|
||
time.sleep(3)
|
||
|
||
for i, target in enumerate(targets):
|
||
print(f"\n--- 步骤 {i + 1}/{len(targets)} ---")
|
||
|
||
if not controller.set_target_pressure(target):
|
||
print(f"❌ 步骤 {i + 1} 出现异常，提前终止自动控制")
|
||
break
|
||
|
||
if i < len(targets) - 1:
|
||
print(f"等待 {interval} 秒...")
|
||
for j in range(interval, 0, -1):
|
||
print(f"\r剩余时间: {j} 秒 ", end='')
|
||
time.sleep(1)
|
||
print()
|
||
|
||
print("\n🎉 自动控制序列全部完成！")
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description='GUI 自动控制脚本')
|
||
parser.add_argument('--targets', type=float, nargs='+', default=[50, 80, 100, 120],
|
||
help='目标压力值列表，用空格隔开，单位 kPa')
|
||
parser.add_argument('--interval', type=int, default=10,
|
||
help='每个目标压力持续时间，单位秒')
|
||
parser.add_argument('--calibrate', action='store_true',
|
||
help='启动坐标校准模式，动态获取输入框和按钮的屏幕坐标')
|
||
|
||
args = parser.parse_args()
|
||
auto_control(args.targets, args.interval, args.calibrate)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|