474 lines
17 KiB
Python
474 lines
17 KiB
Python
"""主自动化逻辑
|
||
|
||
编排整个工作流程：扫描未读聊天 → 点击进入 → 滚动加载历史 → 点击图片/文件 → 关闭 → 循环
|
||
|
||
图片处理流程：点击缩略图 → 打开大图 → 点击"..." → 点击"使用预览打开" → 关闭 Preview.app
|
||
文件处理流程：点击文件气泡触发下载
|
||
"""
|
||
|
||
import logging
|
||
import time
|
||
|
||
from .ax_bridge import AXBridge
|
||
from .config import Config
|
||
from .human_like import HumanBehavior
|
||
from .state_machine import StateMachine, UIState
|
||
from .wechat_ui import WeChatUI
|
||
|
||
logger = logging.getLogger("wechat_clicker.automator")
|
||
|
||
SCROLL_ROUNDS = 5
|
||
SCROLL_LINES_PER_ROUND = -10
|
||
|
||
|
||
class WeChatAutomator:
|
||
"""微信消息自动点击器"""
|
||
|
||
def __init__(self, config: Config, dry_run: bool = False):
|
||
self.config = config
|
||
self.dry_run = dry_run
|
||
self._single_mode = False
|
||
|
||
self.ax = AXBridge()
|
||
self.ui = WeChatUI(self.ax, config.bundle_id)
|
||
self.state = StateMachine(self.ax, self.ui)
|
||
self.human = HumanBehavior(config)
|
||
|
||
self._scan_count = 0
|
||
self._total_chats_processed = 0
|
||
self._total_media_clicked = 0
|
||
|
||
# ----------------------------------------------------------------
|
||
# 启动检查
|
||
# ----------------------------------------------------------------
|
||
|
||
def verify_setup(self) -> bool:
|
||
if not self.ax.check_accessibility():
|
||
logger.error("缺少辅助功能权限，无法继续")
|
||
return False
|
||
|
||
app_ref = self.ui.get_app_ref()
|
||
if app_ref is None:
|
||
logger.error("微信未运行，请先启动微信桌面端")
|
||
return False
|
||
|
||
main_win = self.ui.get_main_window()
|
||
if main_win is None:
|
||
logger.error("未找到微信主窗口，请确保微信已登录并可见")
|
||
return False
|
||
|
||
logger.info("环境检查通过")
|
||
return True
|
||
|
||
# ----------------------------------------------------------------
|
||
# 主循环
|
||
# ----------------------------------------------------------------
|
||
|
||
def run(self):
|
||
logger.info("微信自动点击器启动")
|
||
if not self.verify_setup():
|
||
return
|
||
|
||
while True:
|
||
try:
|
||
self._run_one_cycle()
|
||
except KeyboardInterrupt:
|
||
logger.info("用户中断，正在退出...")
|
||
break
|
||
except Exception as e:
|
||
logger.error(f"主循环异常: {e}", exc_info=True)
|
||
try:
|
||
self.state.recover_to_chat_list()
|
||
except Exception:
|
||
pass
|
||
time.sleep(10)
|
||
|
||
self._print_stats()
|
||
|
||
def run_once(self):
|
||
logger.info("执行单次扫描")
|
||
if not self.verify_setup():
|
||
return
|
||
|
||
self._single_mode = True
|
||
self._run_one_cycle()
|
||
self._print_stats()
|
||
|
||
def _run_one_cycle(self):
|
||
if not self._single_mode:
|
||
if self.human.is_off_hours():
|
||
logger.info("当前为非工作时间，等待中...")
|
||
time.sleep(300)
|
||
return
|
||
|
||
if self.human.should_take_break():
|
||
self.human.long_break()
|
||
return
|
||
|
||
self.ui.ensure_wechat_frontmost()
|
||
time.sleep(0.5)
|
||
|
||
if not self.state.recover_to_chat_list():
|
||
logger.warning("无法恢复到聊天列表，跳过本次循环")
|
||
time.sleep(10)
|
||
return
|
||
|
||
self._scan_count += 1
|
||
|
||
global_unread = self.ui.get_global_unread_count()
|
||
if global_unread == 0:
|
||
logger.debug(f"[扫描#{self._scan_count}] 没有未读消息")
|
||
sleep_time = self.human.scan_interval_with_jitter()
|
||
time.sleep(sleep_time)
|
||
return
|
||
|
||
logger.info(f"[扫描#{self._scan_count}] 全局未读: {global_unread}")
|
||
|
||
unread_chats = self.ui.get_unread_chats()
|
||
if not unread_chats:
|
||
logger.debug("聊天列表中未发现未读项（可能需要滚动）")
|
||
sleep_time = self.human.scan_interval_with_jitter()
|
||
time.sleep(sleep_time)
|
||
return
|
||
|
||
chats_to_process = [
|
||
c for c in unread_chats if self.config.should_process_chat(c.name)
|
||
]
|
||
count = self.human.random_subset_count(
|
||
len(chats_to_process), self.config.max_chats_per_scan
|
||
)
|
||
chats_to_process = chats_to_process[:count]
|
||
|
||
logger.info(
|
||
f"将处理 {len(chats_to_process)} 个聊天 "
|
||
f"(共 {len(unread_chats)} 个未读)"
|
||
)
|
||
|
||
for chat in chats_to_process:
|
||
self.human.delay("before_click_chat")
|
||
self._process_chat(chat)
|
||
self.human.delay("before_close_chat")
|
||
|
||
if not self._single_mode:
|
||
sleep_time = self.human.scan_interval_with_jitter()
|
||
logger.debug(f"等待 {sleep_time:.0f}s 后进行下次扫描")
|
||
time.sleep(sleep_time)
|
||
|
||
# ----------------------------------------------------------------
|
||
# 处理单个聊天
|
||
# ----------------------------------------------------------------
|
||
|
||
def _process_chat(self, chat):
|
||
logger.info(f"处理聊天: {chat.name} (未读: {chat.unread_count})")
|
||
|
||
if self.dry_run:
|
||
logger.info(f" [DRY-RUN] 跳过点击: {chat.name}")
|
||
return
|
||
|
||
chat_pos = self.ax.get_position(chat.element)
|
||
chat_size = self.ax.get_size(chat.element)
|
||
logger.debug(f" 聊天项位置: pos={chat_pos} size={chat_size}")
|
||
|
||
if not self.ax.click_at_element(chat.element):
|
||
logger.warning(f" 点击聊天项失败: {chat.name}")
|
||
return
|
||
|
||
self.human.delay("after_open_chat")
|
||
|
||
# 单窗口模式：会话在主窗口内打开，直接在主窗口中查找消息列表
|
||
main_win = self.ui.get_main_window()
|
||
if main_win is None:
|
||
logger.warning(" 主窗口丢失")
|
||
return
|
||
|
||
msg_list = self.ui.get_message_list(main_win)
|
||
if msg_list is None:
|
||
logger.warning(f" 未找到消息列表，可能聊天未成功打开: {chat.name}")
|
||
return
|
||
|
||
msg_list_pos = self.ax.get_position(msg_list)
|
||
msg_list_size = self.ax.get_size(msg_list)
|
||
logger.debug(f" 消息列表: pos={msg_list_pos} size={msg_list_size}")
|
||
|
||
media_count = self._process_media_with_scroll(main_win, msg_list, chat.name)
|
||
self._total_media_clicked += media_count
|
||
self._total_chats_processed += 1
|
||
logger.info(f" {chat.name}: 本次处理了 {media_count} 个媒体")
|
||
|
||
# ----------------------------------------------------------------
|
||
# 带滚动的媒体处理
|
||
# ----------------------------------------------------------------
|
||
|
||
def _process_media_with_scroll(self, main_win, msg_list, chat_name: str) -> int:
|
||
"""先滚到底部看最新消息，再向上滚动加载历史并处理媒体。"""
|
||
total_clicked = 0
|
||
|
||
# 先滚到消息列表底部（看到最新消息）
|
||
logger.info(f" {chat_name}: 滚动到最新消息...")
|
||
self.ax.scroll_to_bottom(msg_list, rounds=10, lines_per_round=20)
|
||
self.human.random_delay(1.0, 2.0)
|
||
|
||
# 处理当前可见的媒体（最新消息）
|
||
clicked = self._process_visible_media(main_win, chat_name)
|
||
total_clicked += clicked
|
||
|
||
# 向上滚动加载更多历史消息
|
||
for round_idx in range(SCROLL_ROUNDS):
|
||
self.human.delay("between_messages")
|
||
logger.debug(f" {chat_name}: 向上滚动第 {round_idx + 1}/{SCROLL_ROUNDS} 轮")
|
||
self.ax.scroll_at_element(msg_list, lines=SCROLL_LINES_PER_ROUND)
|
||
self.human.random_delay(1.0, 2.5)
|
||
|
||
clicked = self._process_visible_media(main_win, chat_name)
|
||
total_clicked += clicked
|
||
|
||
return total_clicked
|
||
|
||
def _process_visible_media(self, main_win, chat_name: str) -> int:
|
||
"""处理当前可见的所有媒体消息。"""
|
||
media_messages = self.ui.get_media_messages(main_win)
|
||
if not media_messages:
|
||
logger.debug(f" {chat_name}: 当前视图无媒体消息")
|
||
return 0
|
||
|
||
targets = []
|
||
for msg in media_messages:
|
||
if msg.msg_type == "image" and self.config.click_images:
|
||
targets.append(msg)
|
||
elif msg.msg_type == "file" and self.config.click_files:
|
||
targets.append(msg)
|
||
elif msg.msg_type == "video" and self.config.click_videos:
|
||
targets.append(msg)
|
||
|
||
if not targets:
|
||
logger.debug(
|
||
f" {chat_name}: 发现 {len(media_messages)} 个媒体, "
|
||
f"但无符合配置的处理目标"
|
||
)
|
||
return 0
|
||
|
||
logger.info(f" {chat_name}: 当前可见 {len(targets)} 个媒体消息")
|
||
|
||
clicked = 0
|
||
for idx, msg in enumerate(targets):
|
||
self.human.delay("before_click_media")
|
||
|
||
pos = self.ax.get_position(msg.element)
|
||
short_title = msg.title.replace("\n", " ")[:50]
|
||
logger.info(
|
||
f" [{msg.msg_type}] {short_title} "
|
||
f"(pos={pos}, size={msg.size}, #{idx+1}/{len(targets)})"
|
||
)
|
||
|
||
if msg.msg_type == "image":
|
||
success = self._click_image(msg)
|
||
elif msg.msg_type == "file":
|
||
success = self._click_file(msg)
|
||
else:
|
||
success = self._click_generic(msg)
|
||
|
||
if success:
|
||
clicked += 1
|
||
logger.debug(f" 处理成功: {msg.msg_type}")
|
||
else:
|
||
logger.warning(f" 处理失败: {msg.msg_type}")
|
||
|
||
self.human.delay("between_messages")
|
||
|
||
if self.ax.should_backoff():
|
||
logger.warning("连续错误过多，暂停处理")
|
||
self.ax.reset_error_count()
|
||
break
|
||
|
||
return clicked
|
||
|
||
# ----------------------------------------------------------------
|
||
# 图片处理：点击 → "..." → "使用预览打开" → 关闭 Preview
|
||
# ----------------------------------------------------------------
|
||
|
||
def _click_image(self, msg) -> bool:
|
||
"""处理一张图片：点开大图 -> 点... -> 点使用预览打开 -> 关闭 Preview.app。"""
|
||
# Step 1: 点击图片缩略图打开大图预览
|
||
logger.debug(" Step1: 点击图片缩略图")
|
||
if not self.ax.click_at_element(msg.element):
|
||
logger.warning(" 图片缩略图点击失败")
|
||
return False
|
||
|
||
self.human.delay("after_click_media")
|
||
|
||
# Step 2: 检查预览窗口状态
|
||
state = self.state.detect_state()
|
||
logger.debug(f" Step2: 点击后状态={state.value}")
|
||
if state != UIState.MEDIA_PREVIEW:
|
||
logger.debug(" 图片预览窗口未出现，尝试在内嵌预览中继续")
|
||
|
||
# Step 3: 查找 "..." 按钮
|
||
self.human.random_delay(0.5, 1.5)
|
||
logger.debug(" Step3: 搜索'...'按钮")
|
||
more_btn = self._find_more_button_in_preview()
|
||
if more_btn is None:
|
||
logger.warning(" 未找到'...'按钮，Escape 退出")
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.5)
|
||
return False
|
||
|
||
# Step 4: 点击 "..." 按钮
|
||
logger.debug(" Step4: 点击'...'按钮")
|
||
if not self.ax.click_at_element(more_btn):
|
||
if not self.ax.press(more_btn):
|
||
logger.warning(" '...'按钮点击失败")
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.5)
|
||
return False
|
||
|
||
self.human.random_delay(0.5, 1.5)
|
||
|
||
# Step 5: 查找"使用预览打开"菜单项
|
||
logger.debug(" Step5: 搜索'使用预览打开'菜单项")
|
||
preview_item = self.ui.find_menu_item('使用"预览"打开')
|
||
if preview_item is None:
|
||
preview_item = self.ui.find_menu_item("预览")
|
||
if preview_item is None:
|
||
preview_item = self.ui.find_menu_item("Preview")
|
||
|
||
if preview_item is None:
|
||
logger.warning(" 未找到'使用预览打开'菜单项，Escape 退出")
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.3)
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.5)
|
||
return False
|
||
|
||
# Step 6: 点击"使用预览打开"
|
||
logger.debug(" Step6: 点击'使用预览打开'")
|
||
if not self.ax.click_at_element(preview_item):
|
||
self.ax.press(preview_item)
|
||
|
||
self.human.random_delay(2.0, 4.0)
|
||
|
||
# Step 7: 关闭 Preview.app
|
||
if self.state.is_preview_app_running():
|
||
logger.debug(" Step7: 关闭 Preview.app")
|
||
self.state.close_preview_app()
|
||
else:
|
||
logger.debug(" Step7: Preview.app 未运行，跳过关闭")
|
||
|
||
# Step 8: 确保微信回到前台，关闭大图预览
|
||
logger.debug(" Step8: 恢复微信前台，Escape 关闭预览")
|
||
self.ui.ensure_wechat_frontmost()
|
||
time.sleep(0.3)
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.5)
|
||
|
||
state = self.state.detect_state()
|
||
if state == UIState.MEDIA_PREVIEW:
|
||
logger.debug(" 仍在预览状态，再次 Escape")
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.3)
|
||
|
||
return True
|
||
|
||
def _find_more_button_in_preview(self):
|
||
"""在图片预览区域查找"..."按钮（排除侧边栏区域）。"""
|
||
# 先在非主窗口（独立预览窗口）中找
|
||
windows = self.ui.get_all_windows()
|
||
for win in windows:
|
||
title = self.ax.get_title(win)
|
||
if title == "微信":
|
||
continue
|
||
btn = self.ui.find_preview_more_button(win)
|
||
if btn is not None:
|
||
pos = self.ax.get_position(btn)
|
||
logger.debug(f" 在独立窗口 '{title}' 中找到'...'按钮, pos={pos}")
|
||
return btn
|
||
|
||
# 内嵌预览：只在主窗口的右侧区域找（排除左侧侧边栏，x > 200）
|
||
main_win = self.ui.get_main_window()
|
||
if main_win:
|
||
btn = self.ui.find_preview_more_button(main_win, min_x=200)
|
||
if btn is not None:
|
||
pos = self.ax.get_position(btn)
|
||
logger.debug(f" 在主窗口右侧区域找到'...'按钮, pos={pos}")
|
||
return btn
|
||
|
||
logger.debug(" 未在任何窗口中找到'...'按钮")
|
||
return None
|
||
|
||
# ----------------------------------------------------------------
|
||
# 文件处理：直接点击触发下载
|
||
# ----------------------------------------------------------------
|
||
|
||
def _click_file(self, msg) -> bool:
|
||
"""点击文件触发下载。"""
|
||
if not self.ax.click_at_element(msg.element):
|
||
logger.warning(" 文件点击失败")
|
||
return False
|
||
|
||
self.human.delay("after_click_media")
|
||
|
||
# 文件点击后可能会打开文件预览，关闭它
|
||
self.human.random_delay(1.0, 2.0)
|
||
state = self.state.detect_state()
|
||
if state == UIState.MEDIA_PREVIEW:
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.5)
|
||
|
||
return True
|
||
|
||
# ----------------------------------------------------------------
|
||
# 通用处理（视频等）
|
||
# ----------------------------------------------------------------
|
||
|
||
def _click_generic(self, msg) -> bool:
|
||
if not self.ax.click_at_element(msg.element):
|
||
return False
|
||
self.human.delay("after_click_media")
|
||
self.human.micro_jitter()
|
||
self.ax.send_escape_key()
|
||
time.sleep(0.5)
|
||
return True
|
||
|
||
# ----------------------------------------------------------------
|
||
# 调试工具
|
||
# ----------------------------------------------------------------
|
||
|
||
def dump_ui_tree(self):
|
||
if not self.verify_setup():
|
||
return
|
||
|
||
self.ui.ensure_wechat_frontmost()
|
||
time.sleep(0.5)
|
||
|
||
main_win = self.ui.get_main_window()
|
||
if main_win:
|
||
print("=== 主窗口 (微信) ===")
|
||
print(self.ax.dump_element(main_win, max_depth=5))
|
||
|
||
conv_windows = self.ui.get_conversation_windows()
|
||
for win in conv_windows:
|
||
title = self.ax.get_title(win)
|
||
print(f"\n=== 会话窗口 ({title}) ===")
|
||
print(self.ax.dump_element(win, max_depth=5))
|
||
|
||
print("\n=== 聊天列表解析 ===")
|
||
items = self.ui.get_chat_items()
|
||
for item in items:
|
||
pos = self.ax.get_position(item.element)
|
||
size = self.ax.get_size(item.element)
|
||
status = f"[未读:{item.unread_count}]" if item.unread_count > 0 else ""
|
||
print(
|
||
f" {item.name} {status} | {item.preview} | {item.timestamp} "
|
||
f"| pos={pos} size={size}"
|
||
)
|
||
|
||
# ----------------------------------------------------------------
|
||
# 统计
|
||
# ----------------------------------------------------------------
|
||
|
||
def _print_stats(self):
|
||
logger.info(
|
||
f"运行统计: 扫描={self._scan_count}, "
|
||
f"处理聊天={self._total_chats_processed}, "
|
||
f"点击媒体={self._total_media_clicked}"
|
||
)
|