3830 lines
150 KiB
Python
3830 lines
150 KiB
Python
"""
|
||
微信控制技能 - Agent端实现
|
||
|
||
功能模块：
|
||
1. 消息管理 - 发送/获取消息、批量发送
|
||
2. 好友管理 - 添加/通过好友、设置备注、删除好友
|
||
3. 群聊管理 - 创建群/邀请入群/群发消息/设置群欢迎语
|
||
4. 标签管理 - 添加/删除/查询标签
|
||
5. 朋友圈管理 - 发布/点赞/评论
|
||
|
||
@author 卡若
|
||
@version 3.0.0
|
||
"""
|
||
|
||
import time
|
||
import logging
|
||
import re
|
||
import shlex
|
||
import sys
|
||
import os
|
||
from typing import Dict, Any, List, Optional
|
||
_agent_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
if _agent_dir not in sys.path:
|
||
sys.path.insert(0, _agent_dir)
|
||
from skills.base import BaseSkill
|
||
from error_handler import retry, with_error_handling, ErrorHandler
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 截屏 + AI 视觉（每次操作都先看屏再决定）
|
||
class _VisionFallback(Exception):
|
||
"""视觉流程失败，回退到规则流程"""
|
||
pass
|
||
|
||
|
||
def _vision_step(self, goal: str, step_hint: str = "") -> dict:
|
||
"""截屏 → AI 看屏 → 返回下一步动作"""
|
||
try:
|
||
from vision_helper import ask_vision
|
||
img = self.screenshot()
|
||
return ask_vision(img, goal, step_hint)
|
||
except Exception as e:
|
||
logger.warning(f"视觉调用失败: {e}")
|
||
return {"action": "error", "error": str(e)}
|
||
|
||
|
||
class WechatSkill(BaseSkill):
|
||
"""
|
||
微信控制技能 — 96个action全覆盖
|
||
|
||
控制通道（按优先级）：
|
||
1. Frida Hook — 直接调用微信内部方法/读SQLite, 50-200ms
|
||
2. u2 UI自动化 — uiautomator2模拟点击/输入, 2-15s（本文件实现）
|
||
3. ADB Shell — adb命令+坐标点击, 1-10s（wechat_adb_engine.py）
|
||
|
||
功能模块（29个模块 / 96个action）:
|
||
消息: send_message, get_messages, forward_message, recall_message, send_card, batch_send_message
|
||
好友: add_friend, accept_friend, set_remark, delete_friend, get_contacts, search_contact,
|
||
get_friend_info, batch_add_friend
|
||
群聊: create_group, invite_to_group, remove_from_group, set_group_notice, set_group_name,
|
||
send_group_message, set_group_welcome, get_groups, get_group_members, quit_group
|
||
标签: add_tag, remove_tag, create_tag, delete_tag, get_tags, get_users_by_tag
|
||
朋友圈: post_moments, like_moments, comment_moments, get_moments, delete_moments,
|
||
set_moments_cover, set_moments_privacy, forward_moments_link
|
||
个人: get_profile, set_nickname, set_signature, set_avatar, set_gender, set_region
|
||
安全: check_account_status, unblock_account, safety_center, change_password,
|
||
unblock_self, unblock_appeal, check_restrictions, appeal_restriction, unblock_with_sms
|
||
支付: send_red_packet, transfer, show_payment_code, receive_payment, view_wallet,
|
||
view_transactions, receive_red_packet
|
||
聊天: set_chat_top, set_mute_chat, clear_chat_history
|
||
收藏: add_to_favorites, get_favorites
|
||
小程序: open_mini_program | 公众号: follow_official_account
|
||
视频号: open_video_channel, get_video_list, like_video, comment_video,
|
||
follow_video_creator, share_video
|
||
扫码: scan_qr_code, scan_add_friend, show_my_qr, extract_qr_from_image
|
||
通话: voice_call, video_call
|
||
群发: mass_send | 搜索: wechat_search | 发现: top_stories
|
||
运动: get_steps, like_steps
|
||
位置: send_location, share_real_time_location
|
||
表情: send_emoji, get_sticker_list
|
||
语音: send_voice_message | 文件: send_file_from_chat, download_file
|
||
设置: toggle_do_not_disturb, clear_cache, check_for_update, logout, switch_account
|
||
截图: screenshot
|
||
"""
|
||
|
||
PACKAGE = "com.tencent.mm"
|
||
NAME = "微信"
|
||
SEND_MESSAGE_TIMEOUT = 45
|
||
MAX_TEST_PAYMENT_AMOUNT = 1.0
|
||
|
||
def _payment_guard(self, amount: str = "", confirm: bool = False, **kwargs) -> Dict[str, Any]:
|
||
"""资金类动作保护：默认只导航到页面；真确认必须显式授权且金额在测试上限内。"""
|
||
try:
|
||
amt = float(str(amount or "0").strip() or "0")
|
||
except Exception:
|
||
return {"pass": False, "error": "amount 必须是数字"}
|
||
if amt < 0:
|
||
return {"pass": False, "error": "amount 不能为负数"}
|
||
if not confirm:
|
||
return {
|
||
"pass": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"amount": amt,
|
||
"note": "已按内部测试保护策略停止在最终确认前",
|
||
}
|
||
if amt > self.MAX_TEST_PAYMENT_AMOUNT:
|
||
return {
|
||
"pass": False,
|
||
"error": f"测试金额超过上限 {self.MAX_TEST_PAYMENT_AMOUNT}",
|
||
"amount": amt,
|
||
}
|
||
if not kwargs.get("test_whitelist_ok", False):
|
||
return {
|
||
"pass": False,
|
||
"error": "资金动作需要 test_whitelist_ok=true",
|
||
"amount": amt,
|
||
}
|
||
return {"pass": True, "dry_run": False, "confirm_required": False, "amount": amt}
|
||
|
||
def send_message_with_vision(self, to_id: str, content: str, msg_type: str = "text") -> Dict[str, Any]:
|
||
"""
|
||
发送消息（以截屏+AI为主，每次操作都先看屏再决定）
|
||
"""
|
||
try:
|
||
start = time.time()
|
||
self.launch()
|
||
self.sleep(0.8)
|
||
goal = f"给 {to_id} 发消息：{content}"
|
||
max_steps = 15
|
||
for step in range(max_steps):
|
||
if time.time() - start > self.SEND_MESSAGE_TIMEOUT:
|
||
return {"success": False, "error": "timeout"}
|
||
hint = f"步骤 {step+1}/{max_steps}。"
|
||
act = _vision_step(self, goal, hint)
|
||
a = act.get("action", "")
|
||
if a == "done":
|
||
logger.info(f"微信消息发送成功(视觉): {to_id}")
|
||
return {"success": True, "message_id": f"wx_{int(time.time() * 1000)}"}
|
||
if a == "error":
|
||
logger.warning(f"视觉返回错误: {act.get('error')}，回退到规则流程")
|
||
raise _VisionFallback()
|
||
if a == "click":
|
||
x, y = int(act.get("x", 540)), int(act.get("y", 1200))
|
||
self.click(x, y)
|
||
self.sleep(0.3)
|
||
elif a == "input":
|
||
t = act.get("text", "")
|
||
if t:
|
||
self.input_text(t, clear=(step < 2))
|
||
self.sleep(0.2)
|
||
elif a == "back":
|
||
self.back()
|
||
self.sleep(0.3)
|
||
raise _VisionFallback()
|
||
except _VisionFallback:
|
||
raise
|
||
except Exception as e:
|
||
logger.error(f"视觉发送失败: {e}")
|
||
raise _VisionFallback()
|
||
|
||
@with_error_handling("发送微信消息")
|
||
@retry(max_retries=1, delay=1)
|
||
def send_message(self, to_id: str, content: str, msg_type: str = "text") -> Dict[str, Any]:
|
||
"""
|
||
发送消息（坐标导航 + accessibility 双通道）。
|
||
微信的自定义 View 不暴露给 accessibility tree，核心流程用坐标点击。
|
||
"""
|
||
self.natural_behavior_before("send_message")
|
||
start = time.time()
|
||
info = self.d.info
|
||
W, H = info['displayWidth'], info['displayHeight']
|
||
|
||
try:
|
||
# ── 1. 确保在微信主界面 ──
|
||
curr = self.d.app_current()
|
||
if curr.get("package") != self.PACKAGE:
|
||
self.launch()
|
||
self.sleep(2)
|
||
elif "ChattingUI" in (curr.get("activity") or ""):
|
||
self.press("back")
|
||
self.sleep(0.5)
|
||
|
||
for _ in range(3):
|
||
curr = self.d.app_current()
|
||
act = curr.get("activity", "")
|
||
if act.endswith(".LauncherUI"):
|
||
break
|
||
self.press("back")
|
||
self.sleep(0.5)
|
||
else:
|
||
self.launch()
|
||
self.sleep(2)
|
||
|
||
# ── 2. 打开搜索（坐标：标题栏放大镜，在 + 号左侧） ──
|
||
search_x = int(W * 0.82)
|
||
search_y = int(H * 0.06)
|
||
self.click(search_x, search_y)
|
||
self.sleep(1.5)
|
||
|
||
curr = self.d.app_current()
|
||
if "FTSMainUI" not in (curr.get("activity") or ""):
|
||
self.click(search_x - 60, search_y)
|
||
self.sleep(1)
|
||
|
||
# ── 3. 输入联系人 ──
|
||
self.click(int(W * 0.5), int(H * 0.035))
|
||
self.sleep(0.3)
|
||
try:
|
||
self.d.set_input_ime(True)
|
||
except Exception as e:
|
||
logger.warning(f"ADBKeyboard 启用失败，使用 root input text 兜底: {e}")
|
||
self.sleep(0.2)
|
||
self.input_text(to_id, clear=False)
|
||
self.sleep(2.5)
|
||
|
||
# ── 4. 点击搜索结果（微信自定义控件不在 a11y tree，用坐标） ──
|
||
result_x = int(W * 0.3)
|
||
entered_chat = False
|
||
for result_y in [int(H * 0.20), int(H * 0.22), int(H * 0.18), int(H * 0.25)]:
|
||
self.click(result_x, result_y)
|
||
self.sleep(2)
|
||
curr = self.d.app_current()
|
||
if "ChattingUI" in (curr.get("activity") or ""):
|
||
entered_chat = True
|
||
break
|
||
if "FTSMainUI" not in (curr.get("activity") or ""):
|
||
self.press("back")
|
||
self.sleep(0.5)
|
||
|
||
if not entered_chat:
|
||
if self.click_text(to_id, timeout=2) or self.click_contains(to_id, timeout=2):
|
||
self.sleep(2)
|
||
curr = self.d.app_current()
|
||
entered_chat = "ChattingUI" in (curr.get("activity") or "")
|
||
|
||
if not entered_chat:
|
||
return {"success": False, "error": f"未找到联系人: {to_id}"}
|
||
|
||
if time.time() - start > self.SEND_MESSAGE_TIMEOUT:
|
||
return {"success": False, "error": "timeout"}
|
||
|
||
# ── 5. 聚焦输入框 ──
|
||
input_y = int(H * 0.955)
|
||
self.click(int(W * 0.3), input_y)
|
||
self.sleep(0.5)
|
||
|
||
# ── 6. 输入消息 ──
|
||
try:
|
||
self.d.set_input_ime(True)
|
||
except Exception as e:
|
||
logger.warning(f"ADBKeyboard 启用失败，使用 root input text 兜底: {e}")
|
||
self.sleep(0.2)
|
||
self.input_text(content, clear=False)
|
||
self.sleep(0.8)
|
||
|
||
# ── 7. 点击发送按钮（先尝试 xpath/text，再坐标兜底） ──
|
||
sent = False
|
||
try:
|
||
if self.d.xpath('//*[@text="发送"]').exists:
|
||
self.d.xpath('//*[@text="发送"]').click()
|
||
sent = True
|
||
elif self.d(text="发送").exists(timeout=1):
|
||
self.d(text="发送").click()
|
||
sent = True
|
||
except Exception:
|
||
pass
|
||
|
||
if not sent:
|
||
send_x = int(W * 0.91)
|
||
self.click(send_x, input_y)
|
||
self.sleep(0.3)
|
||
|
||
self.sleep(0.5)
|
||
logger.info(f"微信消息发送成功: {to_id}")
|
||
self.say(f"已发消息给 {to_id}", data={"to_id": to_id, "content_preview": content[:30]})
|
||
return {"success": True, "message_id": f"wx_{int(time.time() * 1000)}"}
|
||
except Exception as e:
|
||
logger.error(f"微信发送失败: {e}")
|
||
return {"success": False, "error": str(e)}
|
||
|
||
|
||
def get_messages(self, limit: int = 20, conversation_id: str = None, count: int = None, **kwargs) -> Dict[str, Any]:
|
||
"""
|
||
获取消息列表
|
||
|
||
Args:
|
||
limit: 消息数量限制
|
||
count: limit 的别名（AI Brain 兼容）
|
||
conversation_id: 会话ID（联系人名称），如果指定则获取该会话的消息
|
||
|
||
Returns:
|
||
消息列表
|
||
"""
|
||
if count is not None and limit == 20:
|
||
limit = count
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
messages = []
|
||
|
||
# 如果指定了会话，先进入
|
||
if conversation_id:
|
||
# 使用搜索进入会话
|
||
if self.click_text("搜索") or self.click_desc("搜索"):
|
||
self.sleep(0.5)
|
||
self.input_text(conversation_id)
|
||
self.sleep(1.5)
|
||
|
||
# 点击搜索结果
|
||
if not self.click_text(conversation_id):
|
||
if not self.click_contains(conversation_id):
|
||
return {"success": False, "error": f"未找到会话: {conversation_id}", "messages": []}
|
||
|
||
self.sleep(1)
|
||
|
||
# 获取UI树分析消息
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
# 解析XML提取消息
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
|
||
# 查找消息元素（通常在聊天界面）
|
||
# 微信消息通常有特定的resource-id或class
|
||
for elem in root.iter():
|
||
# 查找消息文本
|
||
text = elem.get("text", "")
|
||
resource_id = elem.get("resource-id", "")
|
||
bounds = elem.get("bounds", "")
|
||
|
||
if text and len(text) > 0:
|
||
# 判断是否是消息内容（排除按钮、输入框等）
|
||
if "发送" not in text and "输入" not in text:
|
||
# 提取坐标判断位置（消息通常在屏幕中间）
|
||
if bounds:
|
||
try:
|
||
coords = bounds.replace("[", "").replace("]", "").split(",")
|
||
if len(coords) >= 4:
|
||
y = int(coords[1])
|
||
# 消息通常在屏幕上半部分
|
||
info = self.d.info
|
||
if y < info['displayHeight'] * 0.8:
|
||
messages.append({
|
||
"text": text,
|
||
"bounds": bounds,
|
||
"timestamp": int(time.time() * 1000)
|
||
})
|
||
except:
|
||
pass
|
||
|
||
# 限制数量
|
||
messages = messages[:limit]
|
||
|
||
except Exception as e:
|
||
logger.warning(f"解析UI树失败: {e}")
|
||
|
||
return {
|
||
"success": True,
|
||
"messages": messages,
|
||
"count": len(messages)
|
||
}
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取消息失败: {e}")
|
||
return {"success": False, "error": str(e), "messages": []}
|
||
|
||
def get_contacts(self, limit: int = 100) -> Dict[str, Any]:
|
||
"""
|
||
获取联系人列表
|
||
|
||
Args:
|
||
limit: 联系人数量限制
|
||
|
||
Returns:
|
||
联系人列表
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 点击通讯录
|
||
if not self.click_text("通讯录"):
|
||
return {"success": False, "error": "无法进入通讯录", "contacts": []}
|
||
|
||
self.sleep(1)
|
||
|
||
contacts = []
|
||
|
||
# 获取UI树
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
# 解析联系人
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
|
||
# 查找联系人元素
|
||
for elem in root.iter():
|
||
text = elem.get("text", "")
|
||
resource_id = elem.get("resource-id", "")
|
||
|
||
# 微信联系人通常有特定的resource-id或class
|
||
if text and len(text) > 0:
|
||
# 排除系统元素
|
||
if "通讯录" not in text and "搜索" not in text:
|
||
# 检查是否是联系人（通常有头像或特定布局）
|
||
parent = elem
|
||
for _ in range(3): # 向上查找3层
|
||
if parent is not None:
|
||
parent_id = parent.get("resource-id", "")
|
||
if "contact" in parent_id.lower() or "item" in parent_id.lower():
|
||
contacts.append({
|
||
"name": text,
|
||
"resource_id": resource_id
|
||
})
|
||
break
|
||
parent = parent.getparent() if hasattr(parent, 'getparent') else None
|
||
|
||
# 去重
|
||
seen = set()
|
||
unique_contacts = []
|
||
for contact in contacts:
|
||
name = contact["name"]
|
||
if name not in seen:
|
||
seen.add(name)
|
||
unique_contacts.append(contact)
|
||
|
||
contacts = unique_contacts[:limit]
|
||
|
||
except Exception as e:
|
||
logger.warning(f"解析联系人失败: {e}")
|
||
# 降级方案：使用搜索功能获取常用联系人
|
||
# 可以通过搜索历史记录获取
|
||
|
||
return {
|
||
"success": True,
|
||
"contacts": contacts,
|
||
"count": len(contacts)
|
||
}
|
||
|
||
except Exception as e:
|
||
logger.error(f"获取联系人失败: {e}")
|
||
return {"success": False, "error": str(e), "contacts": []}
|
||
|
||
def add_friend(self, user_id: str, message: str = "") -> Dict[str, Any]:
|
||
"""添加好友"""
|
||
self.natural_behavior_before("add_friend")
|
||
try:
|
||
self.launch()
|
||
self.sleep(2)
|
||
|
||
# 点击+号
|
||
self.click_desc("+") or self.click_text("+")
|
||
self.sleep(0.5)
|
||
|
||
# 添加朋友
|
||
self.click_text("添加朋友")
|
||
self.sleep(0.5)
|
||
|
||
# 输入微信号
|
||
self.input_text(user_id)
|
||
self.sleep(1)
|
||
|
||
# 搜索
|
||
self.click_text("搜索") or self.click_contains("搜索")
|
||
self.sleep(2)
|
||
|
||
# 添加
|
||
if self.click_text("添加到通讯录"):
|
||
if message:
|
||
self.sleep(0.5)
|
||
self.input_text(message)
|
||
self.click_text("发送")
|
||
return {"success": True}
|
||
|
||
return {"success": False, "error": "未找到用户或已是好友"}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def accept_friend(self, user_id: str = None, **kwargs) -> Dict[str, Any]:
|
||
"""通过好友请求 [控制形式: u2 UI自动化, 耗时~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.sleep(2)
|
||
self.click_text("通讯录")
|
||
self.sleep(1)
|
||
self.click_text("新的朋友")
|
||
self.sleep(1)
|
||
|
||
if user_id:
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
found = False
|
||
for elem in root.iter():
|
||
if elem.get("text", "") == user_id:
|
||
bounds = elem.get("bounds", "")
|
||
if bounds:
|
||
coords = bounds.replace("[", "").replace("]", ",").split(",")
|
||
y_center = (int(coords[1]) + int(coords[4])) // 2
|
||
accept_x = int(self.d.info["displayWidth"] * 0.85)
|
||
self.click(accept_x, y_center)
|
||
self.sleep(1)
|
||
found = True
|
||
break
|
||
if not found:
|
||
if self.click_text("接受"):
|
||
return {"success": True}
|
||
return {"success": False, "error": f"未找到用户 {user_id} 的好友请求"}
|
||
return {"success": True, "user_id": user_id}
|
||
|
||
if self.click_text("接受"):
|
||
return {"success": True}
|
||
return {"success": False, "error": "无待处理的好友请求"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ==========================================================================
|
||
# 三、群聊管理
|
||
# ==========================================================================
|
||
|
||
def create_group(self, group_name: str, member_ids: List[str]) -> Dict[str, Any]:
|
||
"""
|
||
创建群聊
|
||
|
||
Args:
|
||
group_name: 群名称
|
||
member_ids: 成员名称列表（至少2人）
|
||
|
||
Returns:
|
||
执行结果
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 点击右上角+号
|
||
self.click_desc("+") or self.click_text("+")
|
||
self.sleep(0.5)
|
||
|
||
# 发起群聊
|
||
self.click_text("发起群聊")
|
||
self.sleep(1)
|
||
|
||
# 选择群成员
|
||
for member_id in member_ids:
|
||
# 搜索成员
|
||
if self.click_text("搜索") or self.click_desc("搜索"):
|
||
self.sleep(0.3)
|
||
self.input_text(member_id)
|
||
self.sleep(1)
|
||
|
||
# 选中成员
|
||
self.click_text(member_id) or self.click_contains(member_id)
|
||
self.sleep(0.5)
|
||
|
||
# 清空搜索框
|
||
self.back()
|
||
self.sleep(0.3)
|
||
|
||
# 确定创建
|
||
self.click_text("完成") or self.click_text("确定")
|
||
self.sleep(2)
|
||
|
||
# 修改群名
|
||
if group_name:
|
||
# 点击群名区域
|
||
self.click_text("群聊") or self.click_contains("的群聊")
|
||
self.sleep(0.5)
|
||
self.input_text(group_name, clear=True)
|
||
self.click_text("完成") or self.click_text("确定")
|
||
|
||
logger.info(f"创建群聊成功: {group_name}")
|
||
return {
|
||
"success": True,
|
||
"group_name": group_name,
|
||
"member_count": len(member_ids) + 1
|
||
}
|
||
|
||
except Exception as e:
|
||
logger.error(f"创建群聊失败: {e}")
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def invite_to_group(self, group_id: str, member_ids: List[str]) -> Dict[str, Any]:
|
||
"""
|
||
邀请入群
|
||
|
||
Args:
|
||
group_id: 群名称
|
||
member_ids: 要邀请的成员名称列表
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 搜索进入群聊
|
||
if self.click_text("搜索") or self.click_desc("搜索"):
|
||
self.sleep(0.5)
|
||
self.input_text(group_id)
|
||
self.sleep(1.5)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
# 点击右上角群设置
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3) or \
|
||
self.click_text("...")
|
||
self.sleep(1)
|
||
|
||
# 点击+号添加成员
|
||
self.click_desc("+") or self.click_text("+")
|
||
self.sleep(1)
|
||
|
||
# 选择成员
|
||
invited_count = 0
|
||
for member_id in member_ids:
|
||
if self.click_text("搜索") or self.click_desc("搜索"):
|
||
self.sleep(0.3)
|
||
self.input_text(member_id)
|
||
self.sleep(1)
|
||
|
||
if self.click_text(member_id) or self.click_contains(member_id):
|
||
invited_count += 1
|
||
|
||
self.back()
|
||
self.sleep(0.3)
|
||
|
||
# 确定
|
||
self.click_text("完成") or self.click_text("确定")
|
||
|
||
return {
|
||
"success": True,
|
||
"invited_count": invited_count
|
||
}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def remove_from_group(self, group_id: str, member_ids: List[str]) -> Dict[str, Any]:
|
||
"""移出群聊"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入群聊
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
# 进入群设置
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 点击-号移除成员
|
||
self.click_desc("-")
|
||
self.sleep(1)
|
||
|
||
removed_count = 0
|
||
for member_id in member_ids:
|
||
if self.click_text(member_id):
|
||
removed_count += 1
|
||
|
||
self.click_text("完成") or self.click_text("删除")
|
||
|
||
return {"success": True, "removed_count": removed_count}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def send_group_message(
|
||
self,
|
||
group_id: str,
|
||
content: str,
|
||
msg_type: str = "text",
|
||
at_all: bool = False,
|
||
at_list: List[str] = None
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
发送群消息
|
||
|
||
Args:
|
||
group_id: 群名称
|
||
content: 消息内容
|
||
msg_type: 消息类型
|
||
at_all: 是否@所有人
|
||
at_list: @的成员列表
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 搜索进入群聊
|
||
self.search(group_id)
|
||
self.sleep(1)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
# 处理@
|
||
if at_all:
|
||
content = "@所有人 " + content
|
||
elif at_list:
|
||
at_str = " ".join([f"@{name}" for name in at_list])
|
||
content = at_str + " " + content
|
||
|
||
# 输入消息
|
||
self.input_text(content, clear=False)
|
||
self.sleep(0.3)
|
||
|
||
# 发送
|
||
self.click_text("发送")
|
||
|
||
return {
|
||
"success": True,
|
||
"message_id": f"wx_group_{int(time.time() * 1000)}"
|
||
}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_group_notice(self, group_id: str, notice: str) -> Dict[str, Any]:
|
||
"""设置群公告"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入群聊
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
# 进入群设置
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 群公告
|
||
self.click_text("群公告")
|
||
self.sleep(1)
|
||
|
||
# 输入公告
|
||
self.input_text(notice, clear=True)
|
||
|
||
# 发布
|
||
self.click_text("完成") or self.click_text("发布")
|
||
|
||
return {"success": True}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_group_name(self, group_id: str, group_name: str) -> Dict[str, Any]:
|
||
"""设置群名"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入群聊
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
# 进入群设置
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 点击群名
|
||
self.click_text("群聊名称") or self.click_contains("群名")
|
||
self.sleep(0.5)
|
||
|
||
# 输入新群名
|
||
self.input_text(group_name, clear=True)
|
||
|
||
# 确定
|
||
self.click_text("完成") or self.click_text("确定")
|
||
|
||
return {"success": True}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_group_welcome(self, group_id: str, welcome_text: str, welcome_image: str = None) -> Dict[str, Any]:
|
||
"""设置群欢迎语（需要群主/管理员权限）"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入群聊设置
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 群管理
|
||
if not self.click_text("群管理"):
|
||
return {"success": False, "error": "未找到群管理选项，可能不是群主/管理员"}
|
||
self.sleep(1)
|
||
|
||
# 入群欢迎语
|
||
self.click_text("入群欢迎语") or self.click_contains("欢迎语")
|
||
self.sleep(1)
|
||
|
||
# 输入欢迎语
|
||
self.input_text(welcome_text, clear=True)
|
||
|
||
# 保存
|
||
self.click_text("完成") or self.click_text("保存")
|
||
|
||
return {"success": True}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_groups(self, limit: int = 100) -> Dict[str, Any]:
|
||
"""获取群聊列表"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 通讯录 -> 群聊
|
||
self.click_text("通讯录")
|
||
self.sleep(1)
|
||
|
||
self.click_text("群聊")
|
||
self.sleep(1)
|
||
|
||
# 获取UI树解析群列表
|
||
groups = []
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
# 解析群名（简化实现）
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
for elem in root.iter():
|
||
text = elem.get("text", "")
|
||
if text and "群聊" not in text and "通讯录" not in text:
|
||
# 可能是群名
|
||
groups.append({"name": text})
|
||
except:
|
||
pass
|
||
|
||
return {
|
||
"success": True,
|
||
"groups": groups[:limit],
|
||
"count": len(groups[:limit])
|
||
}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "groups": []}
|
||
|
||
def get_group_members(self, group_id: str) -> Dict[str, Any]:
|
||
"""获取群成员列表"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入群聊
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
|
||
# 进入群设置
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 点击查看全部成员
|
||
self.click_contains("查看全部") or self.click_text("全部群成员")
|
||
self.sleep(1)
|
||
|
||
# 获取成员列表
|
||
members = []
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
for elem in root.iter():
|
||
text = elem.get("text", "")
|
||
if text and len(text) > 0:
|
||
if "群成员" not in text and "搜索" not in text:
|
||
members.append({"name": text})
|
||
except:
|
||
pass
|
||
|
||
return {
|
||
"success": True,
|
||
"members": members,
|
||
"count": len(members)
|
||
}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "members": []}
|
||
|
||
# ==========================================================================
|
||
# 四、标签管理
|
||
# ==========================================================================
|
||
|
||
def add_tag(self, user_id: str, tags: List[str]) -> Dict[str, Any]:
|
||
"""
|
||
给好友添加标签
|
||
|
||
Args:
|
||
user_id: 好友名称
|
||
tags: 标签列表
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 搜索好友
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
|
||
# 进入聊天界面后点击头像进入详情
|
||
# 或者直接在搜索结果点击头像
|
||
|
||
# 点击更多设置
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3) or \
|
||
self.click_text("...")
|
||
self.sleep(1)
|
||
|
||
# 设置标签
|
||
self.click_text("设置标签") or self.click_contains("标签")
|
||
self.sleep(1)
|
||
|
||
# 选择/创建标签
|
||
for tag in tags:
|
||
if self.exists(tag):
|
||
self.click_text(tag)
|
||
else:
|
||
# 创建新标签
|
||
self.click_text("添加标签") or self.click_text("新建标签")
|
||
self.sleep(0.5)
|
||
self.input_text(tag)
|
||
self.click_text("完成") or self.click_text("确定")
|
||
self.sleep(0.3)
|
||
|
||
# 保存
|
||
self.click_text("完成") or self.click_text("保存")
|
||
|
||
return {"success": True, "tags_added": tags}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def remove_tag(self, user_id: str, tags: List[str]) -> Dict[str, Any]:
|
||
"""移除好友标签"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入好友详情
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 设置标签
|
||
self.click_text("设置标签") or self.click_contains("标签")
|
||
self.sleep(1)
|
||
|
||
# 取消选择标签
|
||
for tag in tags:
|
||
if self.exists(tag):
|
||
self.click_text(tag) # 再次点击取消选择
|
||
self.sleep(0.3)
|
||
|
||
# 保存
|
||
self.click_text("完成") or self.click_text("保存")
|
||
|
||
return {"success": True, "tags_removed": tags}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def create_tag(self, tag_name: str) -> Dict[str, Any]:
|
||
"""创建标签"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 通讯录 -> 标签
|
||
self.click_text("通讯录")
|
||
self.sleep(1)
|
||
|
||
self.click_text("标签")
|
||
self.sleep(1)
|
||
|
||
# 新建标签
|
||
self.click_text("新建") or self.click_text("添加标签")
|
||
self.sleep(0.5)
|
||
|
||
self.input_text(tag_name)
|
||
self.click_text("完成") or self.click_text("保存")
|
||
|
||
return {"success": True, "tag_name": tag_name}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def delete_tag(self, tag_name: str) -> Dict[str, Any]:
|
||
"""删除标签"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 通讯录 -> 标签
|
||
self.click_text("通讯录")
|
||
self.sleep(1)
|
||
|
||
self.click_text("标签")
|
||
self.sleep(1)
|
||
|
||
# 找到并长按标签
|
||
if self.exists(tag_name):
|
||
# 长按删除
|
||
element = self.d(text=tag_name)
|
||
if element.exists:
|
||
element.long_click()
|
||
self.sleep(0.5)
|
||
self.click_text("删除")
|
||
self.click_text("确定")
|
||
return {"success": True}
|
||
|
||
return {"success": False, "error": f"未找到标签: {tag_name}"}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_tags(self) -> Dict[str, Any]:
|
||
"""获取标签列表"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 通讯录 -> 标签
|
||
self.click_text("通讯录")
|
||
self.sleep(1)
|
||
|
||
self.click_text("标签")
|
||
self.sleep(1)
|
||
|
||
# 解析标签列表
|
||
tags = []
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
for elem in root.iter():
|
||
text = elem.get("text", "")
|
||
if text and "标签" not in text and "新建" not in text:
|
||
# 提取标签名（格式可能是"标签名(数量)"）
|
||
match = re.match(r'^(.+?)(?:\(\d+\))?$', text)
|
||
if match:
|
||
tags.append(match.group(1))
|
||
except:
|
||
pass
|
||
|
||
return {"success": True, "tags": tags}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "tags": []}
|
||
|
||
def get_users_by_tag(self, tag_name: str, limit: int = 100) -> Dict[str, Any]:
|
||
"""根据标签获取好友列表"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 通讯录 -> 标签
|
||
self.click_text("通讯录")
|
||
self.sleep(1)
|
||
|
||
self.click_text("标签")
|
||
self.sleep(1)
|
||
|
||
# 点击标签
|
||
self.click_text(tag_name) or self.click_contains(tag_name)
|
||
self.sleep(1)
|
||
|
||
# 解析好友列表
|
||
users = []
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
for elem in root.iter():
|
||
text = elem.get("text", "")
|
||
if text and len(text) > 0:
|
||
if tag_name not in text and "人" not in text:
|
||
users.append({"name": text})
|
||
except:
|
||
pass
|
||
|
||
return {
|
||
"success": True,
|
||
"users": users[:limit],
|
||
"count": len(users[:limit])
|
||
}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "users": []}
|
||
|
||
# ==========================================================================
|
||
# 五、朋友圈管理
|
||
# ==========================================================================
|
||
|
||
def post_moments(
|
||
self,
|
||
content: str,
|
||
images: List[str] = None,
|
||
video_url: str = None,
|
||
location: str = None,
|
||
visible_list: List[str] = None,
|
||
invisible_list: List[str] = None
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
发布朋友圈
|
||
|
||
Args:
|
||
content: 文字内容
|
||
images: 图片URL列表（需要先下载到本地）
|
||
video_url: 视频URL
|
||
location: 位置
|
||
visible_list: 可见名单
|
||
invisible_list: 不可见名单
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 发现 -> 朋友圈
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("朋友圈")
|
||
self.sleep(2)
|
||
|
||
# 点击相机图标发布
|
||
camera = self.d.xpath('//*[@content-desc="拍照分享"]')
|
||
if camera.exists:
|
||
if images or video_url:
|
||
# 短按：选择图片/视频
|
||
camera.click()
|
||
else:
|
||
# 长按：纯文字
|
||
camera.long_click()
|
||
self.sleep(1)
|
||
|
||
if images:
|
||
# 从相册选择
|
||
self.click_text("从相册选择")
|
||
self.sleep(1)
|
||
# TODO: 选择图片（需要图片已在相册中）
|
||
# 这里简化处理，假设选择第一张
|
||
self.click_text("完成") or self.click_text("发送")
|
||
|
||
self.sleep(1)
|
||
|
||
# 输入文字内容
|
||
if self.click_contains("这一刻的想法") or self.click_contains("说点什么"):
|
||
self.input_text(content)
|
||
else:
|
||
self.input_text(content)
|
||
|
||
# 设置位置
|
||
if location:
|
||
self.click_text("所在位置")
|
||
self.sleep(1)
|
||
self.input_text(location)
|
||
self.sleep(1)
|
||
self.click_text(location) or self.click_contains(location)
|
||
|
||
# 设置可见范围
|
||
if visible_list or invisible_list:
|
||
self.click_text("谁可以看")
|
||
self.sleep(0.5)
|
||
if visible_list:
|
||
self.click_text("部分可见")
|
||
# TODO: 选择可见的人
|
||
elif invisible_list:
|
||
self.click_text("不给谁看")
|
||
# TODO: 选择不可见的人
|
||
self.click_text("完成")
|
||
|
||
# 发表
|
||
self.click_text("发表")
|
||
self.sleep(2)
|
||
|
||
return {
|
||
"success": True,
|
||
"post_id": f"moments_{int(time.time() * 1000)}"
|
||
}
|
||
|
||
except Exception as e:
|
||
logger.error(f"发布朋友圈失败: {e}")
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def like_moments(self, user_id: str, post_index: int = 0) -> Dict[str, Any]:
|
||
"""
|
||
点赞朋友圈
|
||
|
||
Args:
|
||
user_id: 好友名称
|
||
post_index: 第几条朋友圈（0表示最新一条）
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入好友朋友圈
|
||
self._enter_user_moments(user_id)
|
||
|
||
# 滚动到指定位置
|
||
for _ in range(post_index):
|
||
self.swipe("up", scale=0.3)
|
||
self.sleep(0.5)
|
||
|
||
# 点击评论按钮
|
||
self.click_desc("评论") or self.d.xpath('//*[@content-desc="评论"]').click()
|
||
self.sleep(0.5)
|
||
|
||
# 点击赞
|
||
self.click_text("赞")
|
||
|
||
return {"success": True}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def comment_moments(
|
||
self,
|
||
user_id: str,
|
||
comment: str,
|
||
post_index: int = 0,
|
||
reply_to: str = None
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
评论朋友圈
|
||
|
||
Args:
|
||
user_id: 好友名称
|
||
comment: 评论内容
|
||
post_index: 第几条朋友圈
|
||
reply_to: 回复某人
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入好友朋友圈
|
||
self._enter_user_moments(user_id)
|
||
|
||
# 滚动到指定位置
|
||
for _ in range(post_index):
|
||
self.swipe("up", scale=0.3)
|
||
self.sleep(0.5)
|
||
|
||
# 点击评论按钮
|
||
self.click_desc("评论") or self.d.xpath('//*[@content-desc="评论"]').click()
|
||
self.sleep(0.5)
|
||
|
||
# 点击评论
|
||
self.click_text("评论")
|
||
self.sleep(0.3)
|
||
|
||
# 如果要回复某人
|
||
if reply_to:
|
||
# 找到该人的评论并点击
|
||
self.click_text(reply_to)
|
||
self.sleep(0.3)
|
||
|
||
# 输入评论
|
||
self.input_text(comment)
|
||
|
||
# 发送
|
||
self.click_text("发送")
|
||
|
||
return {"success": True}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_moments(self, user_id: str = None, limit: int = 10) -> Dict[str, Any]:
|
||
"""
|
||
获取朋友圈列表
|
||
|
||
Args:
|
||
user_id: 好友名称，不指定则获取自己的朋友圈
|
||
limit: 获取数量
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
if user_id:
|
||
self._enter_user_moments(user_id)
|
||
else:
|
||
# 进入自己的朋友圈
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("朋友圈")
|
||
self.sleep(2)
|
||
# 点击自己的头像
|
||
# TODO: 进入个人相册
|
||
|
||
# 获取朋友圈内容
|
||
moments = []
|
||
ui_tree = self.d.dump_hierarchy()
|
||
|
||
# 简化解析
|
||
import xml.etree.ElementTree as ET
|
||
try:
|
||
root = ET.fromstring(ui_tree)
|
||
for elem in root.iter():
|
||
text = elem.get("text", "")
|
||
if text and len(text) > 10: # 可能是朋友圈内容
|
||
moments.append({"content": text[:100]})
|
||
except:
|
||
pass
|
||
|
||
return {
|
||
"success": True,
|
||
"moments": moments[:limit],
|
||
"count": len(moments[:limit])
|
||
}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "moments": []}
|
||
|
||
def _enter_user_moments(self, user_id: str):
|
||
"""进入好友的朋友圈（内部方法）"""
|
||
# 搜索好友
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
|
||
# 进入好友详情
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 点击头像进入详情页
|
||
# 然后点击朋友圈
|
||
self.click_text("朋友圈") or self.click_contains("个人相册")
|
||
self.sleep(2)
|
||
|
||
# ==========================================================================
|
||
# 六、批量操作
|
||
# ==========================================================================
|
||
|
||
def batch_send_message(
|
||
self,
|
||
to_ids: List[str],
|
||
content: str,
|
||
msg_type: str = "text",
|
||
interval: float = 2.0
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
批量发送消息
|
||
|
||
Args:
|
||
to_ids: 接收者ID列表
|
||
content: 消息内容
|
||
msg_type: 消息类型
|
||
interval: 发送间隔(秒)
|
||
|
||
Returns:
|
||
批量发送结果
|
||
"""
|
||
results = []
|
||
success_count = 0
|
||
failed_count = 0
|
||
|
||
for to_id in to_ids:
|
||
try:
|
||
result = self.send_message(to_id, content, msg_type)
|
||
|
||
if result.get("success"):
|
||
success_count += 1
|
||
else:
|
||
failed_count += 1
|
||
|
||
results.append({
|
||
"to_id": to_id,
|
||
"success": result.get("success", False),
|
||
"error": result.get("error"),
|
||
"message_id": result.get("message_id")
|
||
})
|
||
|
||
# 发送间隔
|
||
time.sleep(interval)
|
||
|
||
except Exception as e:
|
||
failed_count += 1
|
||
results.append({
|
||
"to_id": to_id,
|
||
"success": False,
|
||
"error": str(e)
|
||
})
|
||
|
||
return {
|
||
"success": True,
|
||
"success_count": success_count,
|
||
"failed_count": failed_count,
|
||
"results": results
|
||
}
|
||
|
||
def batch_add_friend(
|
||
self,
|
||
user_ids: List[str],
|
||
message: str = "",
|
||
interval: float = 5.0
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
批量添加好友
|
||
|
||
Args:
|
||
user_ids: 用户ID列表
|
||
message: 验证消息
|
||
interval: 添加间隔(秒)
|
||
"""
|
||
results = []
|
||
success_count = 0
|
||
failed_count = 0
|
||
|
||
for user_id in user_ids:
|
||
try:
|
||
result = self.add_friend(user_id, message)
|
||
|
||
if result.get("success"):
|
||
success_count += 1
|
||
else:
|
||
failed_count += 1
|
||
|
||
results.append({
|
||
"user_id": user_id,
|
||
"success": result.get("success", False),
|
||
"error": result.get("error")
|
||
})
|
||
|
||
time.sleep(interval)
|
||
|
||
except Exception as e:
|
||
failed_count += 1
|
||
results.append({
|
||
"user_id": user_id,
|
||
"success": False,
|
||
"error": str(e)
|
||
})
|
||
|
||
return {
|
||
"success": True,
|
||
"success_count": success_count,
|
||
"failed_count": failed_count,
|
||
"results": results
|
||
}
|
||
|
||
# ==========================================================================
|
||
# 七、好友管理扩展
|
||
# ==========================================================================
|
||
|
||
def set_remark(self, user_id: str, remark: str) -> Dict[str, Any]:
|
||
"""
|
||
设置好友备注
|
||
|
||
Args:
|
||
user_id: 好友名称
|
||
remark: 新备注
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 进入好友详情
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
|
||
# 设置备注和标签
|
||
self.click_text("设置备注和标签") or self.click_contains("备注")
|
||
self.sleep(0.5)
|
||
|
||
# 输入备注
|
||
self.input_text(remark, clear=True)
|
||
|
||
# 保存
|
||
self.click_text("完成") or self.click_text("保存")
|
||
|
||
return {"success": True, "remark": remark}
|
||
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def delete_friend(self, user_id: str, **kwargs) -> Dict[str, Any]:
|
||
"""删除好友 [控制形式: u2 UI自动化, 耗时~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
self.click(int(W * 0.2), int(self.d.info["displayHeight"] * 0.12))
|
||
self.sleep(1.5)
|
||
self.click_text("...") or self.click_desc("更多")
|
||
self.sleep(0.5)
|
||
self.click_text("删除")
|
||
self.sleep(0.5)
|
||
self.click_text("删除") or self.click_text("确定")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ==========================================================================
|
||
# 八、辅助方法
|
||
# ==========================================================================
|
||
|
||
def send_message_with_search(self, to_id: str, content: str) -> Dict[str, Any]:
|
||
"""
|
||
通过搜索发送消息（更可靠）
|
||
|
||
Args:
|
||
to_id: 联系人名称/备注
|
||
content: 消息内容
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
|
||
# 使用搜索功能
|
||
search_result = self.search(to_id)
|
||
if not search_result.get("success"):
|
||
return {"success": False, "error": "搜索联系人失败"}
|
||
|
||
self.sleep(1)
|
||
|
||
# 点击第一个搜索结果
|
||
if not self.click_text(to_id):
|
||
if not self.click_contains(to_id):
|
||
return {"success": False, "error": f"未找到联系人: {to_id}"}
|
||
|
||
self.sleep(1)
|
||
|
||
# 确保在聊天界面
|
||
if not self.exists("发送") and not self.exists("按住 说话"):
|
||
return {"success": False, "error": "未能进入聊天界面"}
|
||
|
||
# 输入并发送消息
|
||
self.input_text(content, clear=False)
|
||
self.sleep(0.3)
|
||
self.click_text("发送")
|
||
|
||
return {
|
||
"success": True,
|
||
"message_id": f"wx_{int(time.time() * 1000)}"
|
||
}
|
||
|
||
except Exception as e:
|
||
logger.error(f"微信发送失败: {e}")
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def execute_compound_task(self, task_description: str) -> Dict[str, Any]:
|
||
"""
|
||
执行复合任务（如"打开微信给张三发消息：下午开会"）
|
||
|
||
Args:
|
||
task_description: 任务描述
|
||
|
||
Returns:
|
||
执行结果
|
||
"""
|
||
try:
|
||
# 解析任务
|
||
# 提取联系人
|
||
contact_match = re.search(r'给(.+?)发', task_description)
|
||
if not contact_match:
|
||
return {"success": False, "error": "无法解析联系人"}
|
||
|
||
contact = contact_match.group(1).strip()
|
||
|
||
# 提取消息内容
|
||
content_match = re.search(r'发消息[：:](.+)', task_description)
|
||
if not content_match:
|
||
content_match = re.search(r'发(.+)', task_description)
|
||
|
||
if not content_match:
|
||
return {"success": False, "error": "无法解析消息内容"}
|
||
|
||
content = content_match.group(1).strip()
|
||
|
||
# 执行发送
|
||
return self.send_message_with_search(contact, content)
|
||
|
||
except Exception as e:
|
||
logger.error(f"执行复合任务失败: {e}")
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_current_screen_info(self) -> Dict[str, Any]:
|
||
"""获取当前屏幕信息（调试用）"""
|
||
try:
|
||
ui_tree = self.d.dump_hierarchy()
|
||
info = self.d.info
|
||
return {
|
||
"success": True,
|
||
"display_width": info.get("displayWidth"),
|
||
"display_height": info.get("displayHeight"),
|
||
"current_package": info.get("currentPackageName"),
|
||
"ui_tree_length": len(ui_tree)
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 九、消息高级操作（新增）
|
||
# 控制形式: u2 UI自动化 | Hook可加速(forward/recall走Frida RPC更快)
|
||
# ======================================================================
|
||
|
||
def forward_message(self, to_id: str = "", content: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""转发消息 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.sleep(1)
|
||
self.long_click(int(self.d.info["displayWidth"] * 0.5),
|
||
int(self.d.info["displayHeight"] * 0.5), duration=1.0)
|
||
self.sleep(0.5)
|
||
if not (self.click_text("转发") or self.click_text("逐条转发")):
|
||
return {"success": False, "error": "未找到转发选项"}
|
||
self.sleep(1)
|
||
if to_id:
|
||
self.click_text("搜索") or self.click_desc("搜索")
|
||
self.sleep(0.3)
|
||
self.input_text(to_id)
|
||
self.sleep(1)
|
||
self.click_text(to_id) or self.click_contains(to_id)
|
||
self.sleep(0.5)
|
||
self.click_text("发送") or self.click_text("确定")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def forward_multiple(self, msg_svr_ids: List[str] = None, to_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""合并/逐条转发多条消息 [u2, 默认基于当前聊天上下文进入转发流程]"""
|
||
result = self.forward_message(to_id=to_id, **kwargs)
|
||
if result.get("success"):
|
||
result["msg_svr_ids"] = msg_svr_ids or []
|
||
result.setdefault("note", "已进入转发流程；多选依赖当前聊天上下文")
|
||
return result
|
||
|
||
def recall_message(self, **kwargs) -> Dict[str, Any]:
|
||
"""撤回最近一条消息 [u2, ~3s]"""
|
||
try:
|
||
self.launch()
|
||
self.sleep(0.5)
|
||
H = self.d.info["displayHeight"]
|
||
W = self.d.info["displayWidth"]
|
||
self.long_click(int(W * 0.7), int(H * 0.85), duration=1.0)
|
||
self.sleep(0.5)
|
||
if self.click_text("撤回"):
|
||
self.sleep(0.5)
|
||
self.click_text("确定")
|
||
return {"success": True}
|
||
return {"success": False, "error": "未找到撤回选项"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def send_card(self, to_id: str = "", card_wxid: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""发送名片 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(to_id)
|
||
self.click_text(to_id) or self.click_contains(to_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("名片") or self.click_contains("名片")
|
||
self.sleep(1)
|
||
if card_wxid:
|
||
self.click_text("搜索") or self.click_desc("搜索")
|
||
self.sleep(0.3)
|
||
self.input_text(card_wxid)
|
||
self.sleep(1)
|
||
self.click_text(card_wxid) or self.click_contains(card_wxid)
|
||
self.sleep(0.5)
|
||
self.click_text("发送") or self.click_text("确定")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十、好友管理扩展（新增）
|
||
# 控制形式: u2 UI自动化 | Hook可读(search_contact/get_friend_info走SQLite)
|
||
# ======================================================================
|
||
|
||
def search_contact(self, keyword: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""搜索联系人 [Hook优先(SQLite), u2降级, ~2s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(keyword)
|
||
self.sleep(1.5)
|
||
results = []
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
skip = {"搜索", "取消", "清除", keyword}
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and len(t) > 0 and t not in skip:
|
||
results.append({"name": t})
|
||
return {"success": True, "contacts": results[:20]}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "contacts": []}
|
||
|
||
def get_friend_info(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""获取好友详细资料 [Hook优先(SQLite rcontact), u2降级, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
H = self.d.info["displayHeight"]
|
||
self.click(int(W * 0.2), int(H * 0.12))
|
||
self.sleep(1.5)
|
||
info = {}
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
texts = [e.get("text", "") for e in root.iter() if e.get("text", "").strip()]
|
||
info["raw_texts"] = texts[:20]
|
||
for t in texts:
|
||
if "微信号" in t:
|
||
info["wxid"] = t.replace("微信号：", "").replace("微信号:", "").strip()
|
||
if "地区" in t or "Region" in t:
|
||
info["region"] = t
|
||
info["nickname"] = user_id
|
||
return {"success": True, **info}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十一、群聊扩展（新增）
|
||
# ======================================================================
|
||
|
||
def quit_group(self, group_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""退出群聊 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
for _ in range(5):
|
||
self.swipe("up", scale=0.5)
|
||
self.sleep(0.5)
|
||
if self.click_text("删除并退出") or self.click_text("退出群聊"):
|
||
self.sleep(0.5)
|
||
self.click_text("确定") or self.click_text("退出")
|
||
return {"success": True}
|
||
return {"success": False, "error": "未找到退出群聊按钮"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十二、朋友圈增强（新增）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def delete_moments(self, post_index: int = 0, **kwargs) -> Dict[str, Any]:
|
||
"""删除朋友圈 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("朋友圈")
|
||
self.sleep(2)
|
||
self.click(int(self.d.info["displayWidth"] * 0.15),
|
||
int(self.d.info["displayHeight"] * 0.08))
|
||
self.sleep(2)
|
||
for _ in range(post_index):
|
||
self.swipe("up", scale=0.3)
|
||
self.sleep(0.5)
|
||
self.long_click(int(self.d.info["displayWidth"] * 0.5),
|
||
int(self.d.info["displayHeight"] * 0.4), duration=1.0)
|
||
self.sleep(0.5)
|
||
if self.click_text("删除"):
|
||
self.sleep(0.3)
|
||
self.click_text("确定") or self.click_text("删除")
|
||
return {"success": True}
|
||
return {"success": False, "error": "未找到删除选项"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_moments_cover(self, **kwargs) -> Dict[str, Any]:
|
||
"""设置朋友圈封面 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("朋友圈")
|
||
self.sleep(2)
|
||
self.click(int(self.d.info["displayWidth"] * 0.5),
|
||
int(self.d.info["displayHeight"] * 0.15))
|
||
self.sleep(1)
|
||
self.click_text("换封面") or self.click_text("更换封面") or self.click_contains("封面")
|
||
self.sleep(1)
|
||
self.click_text("从手机相册选择") or self.click_text("相册")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开相册选择页面"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_moments_privacy(self, days: int = 180, **kwargs) -> Dict[str, Any]:
|
||
"""设置朋友圈可见天数 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("朋友权限") or self.click_text("隐私")
|
||
self.sleep(1)
|
||
self.click_text("朋友圈") or self.click_contains("允许朋友查看")
|
||
self.sleep(1)
|
||
day_map = {3: "最近三天", 30: "最近一个月", 180: "最近半年", 0: "全部"}
|
||
target = day_map.get(days, "最近半年")
|
||
if self.click_text(target) or self.click_contains(target):
|
||
return {"success": True, "days": days}
|
||
return {"success": False, "error": f"未找到选项: {target}"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def forward_moments_link(self, url: str = "", title: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""分享链接到朋友圈 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("朋友圈")
|
||
self.sleep(2)
|
||
camera = self.d.xpath('//*[@content-desc="拍照分享"]')
|
||
if camera.exists:
|
||
camera.click()
|
||
self.sleep(1)
|
||
if title:
|
||
self.input_text(f"{title}\n{url}" if url else title)
|
||
elif url:
|
||
self.input_text(url)
|
||
self.sleep(0.5)
|
||
self.click_text("发表")
|
||
self.sleep(2)
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十三、个人设置（新增6个）
|
||
# 控制形式: u2 UI自动化 | Hook只能读(get_profile走SQLite)
|
||
# ======================================================================
|
||
|
||
def _goto_personal_info(self):
|
||
"""进入个人信息页（内部方法）"""
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
self.click(int(W * 0.3), int(self.d.info["displayHeight"] * 0.12))
|
||
self.sleep(1.5)
|
||
|
||
def get_profile(self, **kwargs) -> Dict[str, Any]:
|
||
"""获取当前微信账号资料 [Hook优先(SQLite userinfo, ~50ms), u2降级~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._goto_personal_info()
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
texts = [e.get("text", "").strip() for e in root.iter() if e.get("text", "").strip()]
|
||
profile = {"raw_texts": texts[:30]}
|
||
for t in texts:
|
||
if "微信号" in t:
|
||
profile["wxid"] = t.split("：")[-1].split(":")[-1].strip()
|
||
if "地区" in t:
|
||
profile["region"] = t
|
||
if texts:
|
||
profile["nickname"] = texts[0]
|
||
return {"success": True, **profile}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_nickname(self, nickname: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""修改微信昵称 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._goto_personal_info()
|
||
self.click_text("名字") or self.click_contains("昵称")
|
||
self.sleep(0.5)
|
||
self.input_text(nickname, clear=True)
|
||
self.sleep(0.3)
|
||
self.click_text("保存") or self.click_text("完成")
|
||
return {"success": True, "nickname": nickname}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_signature(self, signature: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""修改个性签名 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._goto_personal_info()
|
||
if not (self.click_text("个性签名") or self.click_contains("签名")):
|
||
self.click_text("更多信息")
|
||
self.sleep(0.5)
|
||
self.click_text("个性签名")
|
||
self.sleep(0.5)
|
||
self.input_text(signature, clear=True)
|
||
self.sleep(0.3)
|
||
self.click_text("保存") or self.click_text("完成")
|
||
return {"success": True, "signature": signature}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_status(self, status: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""设置微信状态 [u2, 默认打开状态入口]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("+状态") or self.click_text("状态") or self.click_contains("状态")
|
||
self.sleep(1)
|
||
if status and bool(kwargs.get("confirm")):
|
||
self.input_text(status)
|
||
self.sleep(0.3)
|
||
self.click_text("就这样") or self.click_text("完成")
|
||
return {
|
||
"success": True,
|
||
"status": status,
|
||
"dry_run": not bool(kwargs.get("confirm")),
|
||
"confirm_required": bool(status) and not bool(kwargs.get("confirm")),
|
||
"note": "已打开微信状态入口",
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_avatar(self, image_path: str = None, **kwargs) -> Dict[str, Any]:
|
||
"""修改头像 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._goto_personal_info()
|
||
self.click_text("头像")
|
||
self.sleep(1)
|
||
self.click_text("从手机相册选择") or self.click_text("相册")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开相册选择页面"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_gender(self, gender: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""设置性别 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._goto_personal_info()
|
||
if not (self.click_text("性别") or self.click_contains("性别")):
|
||
self.click_text("更多信息")
|
||
self.sleep(0.5)
|
||
self.click_text("性别")
|
||
self.sleep(0.5)
|
||
target = "男" if gender.lower() in ("male", "男", "m") else "女"
|
||
self.click_text(target)
|
||
self.sleep(0.3)
|
||
self.click_text("保存") or self.click_text("完成")
|
||
return {"success": True, "gender": target}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_region(self, region: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""设置地区 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._goto_personal_info()
|
||
if not (self.click_text("地区") or self.click_contains("地区")):
|
||
self.click_text("更多信息")
|
||
self.sleep(0.5)
|
||
self.click_text("地区")
|
||
self.sleep(0.5)
|
||
parts = region.split(" ") if region else ["中国", "福建", "厦门"]
|
||
for part in parts:
|
||
self.click_text(part) or self.click_contains(part)
|
||
self.sleep(0.5)
|
||
self.click_text("完成") or self.click_text("确定")
|
||
return {"success": True, "region": region}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十四、账号安全（新增9个）
|
||
# 控制形式: u2 UI自动化 | Hook仅check_account_status可读
|
||
# ======================================================================
|
||
|
||
def check_account_status(self, **kwargs) -> Dict[str, Any]:
|
||
"""检查账号状态 [Hook优先(SQLite, ~50ms), u2降级~4s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
curr = self.d.app_current()
|
||
act = curr.get("activity", "")
|
||
logged_in = "LauncherUI" in act or "ChattingUI" in act
|
||
return {
|
||
"success": True,
|
||
"logged_in": logged_in,
|
||
"activity": act,
|
||
"status": "normal" if logged_in else "unknown"
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def unblock_account(self, helper_wxid: str = None, **kwargs) -> Dict[str, Any]:
|
||
"""微信解封 [u2, ~10s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
if self.click_text("微信安全中心") or self.click_contains("安全中心"):
|
||
self.sleep(1)
|
||
self.click_text("解封账号") or self.click_contains("解封")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已进入解封页面"}
|
||
return {"success": False, "error": "未找到安全中心"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def safety_center(self, **kwargs) -> Dict[str, Any]:
|
||
"""打开微信安全中心 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
self.click_text("微信安全中心") or self.click_contains("安全中心")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开安全中心"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def change_password(self, old_pwd: str = "", new_pwd: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""修改微信密码 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
self.click_text("微信密码") or self.click_contains("密码")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已进入密码修改页面"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def bind_phone(self, phone: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""绑定手机号 [u2, 默认打开手机号设置入口]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
self.click_text("手机号") or self.click_contains("手机")
|
||
self.sleep(1)
|
||
return {"success": True, "phone": phone, "note": "已打开手机号设置入口"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def unbind_phone(self, **kwargs) -> Dict[str, Any]:
|
||
"""解绑银行卡式手机号入口 [u2, 默认只导航]"""
|
||
return self.bind_phone(**kwargs)
|
||
|
||
def get_login_devices(self, **kwargs) -> Dict[str, Any]:
|
||
"""查看登录设备 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
self.click_text("登录设备管理") or self.click_contains("登录设备")
|
||
self.sleep(1)
|
||
items = []
|
||
try:
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(self.d.dump_hierarchy())
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and t not in ("登录设备管理", "当前设备"):
|
||
items.append(t[:80])
|
||
except Exception:
|
||
pass
|
||
return {"success": True, "devices": items[:30], "count": len(items[:30])}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "devices": []}
|
||
|
||
def remove_login_device(self, device_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""移除登录设备 [u2, 默认停在列表/确认前]"""
|
||
res = self.get_login_devices(**kwargs)
|
||
if not res.get("success") or not bool(kwargs.get("confirm")):
|
||
res.update({"dry_run": True, "confirm_required": True, "target_device_id": device_id})
|
||
return res
|
||
try:
|
||
if device_id:
|
||
self.click_text(device_id) or self.click_contains(device_id)
|
||
self.sleep(0.5)
|
||
self.click_text("删除") or self.click_text("移除")
|
||
self.sleep(0.5)
|
||
self.click_text("确定") or self.click_text("删除")
|
||
return {"success": True, "device_id": device_id}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def enable_fingerprint(self, enable: bool = True, **kwargs) -> Dict[str, Any]:
|
||
"""开启/关闭指纹支付入口 [u2, 默认只导航]"""
|
||
return self._open_payment_settings("指纹", enable=enable)
|
||
|
||
def set_account_protection(self, enable: bool = True, **kwargs) -> Dict[str, Any]:
|
||
"""账号保护入口 [u2, 默认只导航]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
self.click_text("账号保护") or self.click_contains("保护")
|
||
self.sleep(1)
|
||
return {"success": True, "enable": enable, "note": "已打开账号保护入口"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def _open_payment_settings(self, keyword: str = "", **kwargs) -> Dict[str, Any]:
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("服务") or self.click_text("支付")
|
||
self.sleep(1)
|
||
self.click_text("钱包")
|
||
self.sleep(1)
|
||
self.click_text("支付设置") or self.click_contains("支付设置")
|
||
self.sleep(1)
|
||
if keyword:
|
||
self.click_text(keyword) or self.click_contains(keyword)
|
||
self.sleep(0.5)
|
||
return {"success": True, "keyword": keyword, "note": "已打开支付设置入口"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def unblock_self(self, **kwargs) -> Dict[str, Any]:
|
||
"""自助解封 [u2, ~8s]"""
|
||
return self.unblock_account(**kwargs)
|
||
|
||
def unblock_appeal(self, **kwargs) -> Dict[str, Any]:
|
||
"""申诉解封 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("账号与安全") or self.click_text("帐号与安全")
|
||
self.sleep(1)
|
||
self.click_text("微信安全中心") or self.click_contains("安全中心")
|
||
self.sleep(1)
|
||
self.click_text("申诉") or self.click_contains("申诉")
|
||
return {"success": True, "note": "已进入申诉页面"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def check_restrictions(self, **kwargs) -> Dict[str, Any]:
|
||
"""检查当前功能限制 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
curr = self.d.app_current()
|
||
return {
|
||
"success": True,
|
||
"activity": curr.get("activity", ""),
|
||
"restricted": False,
|
||
"note": "功能限制需在安全中心查看"
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def appeal_restriction(self, **kwargs) -> Dict[str, Any]:
|
||
"""申诉功能限制 [u2, ~8s]"""
|
||
return self.unblock_appeal(**kwargs)
|
||
|
||
def check_login_state(self, **kwargs) -> Dict[str, Any]:
|
||
"""检测微信是否已登录 [u2]"""
|
||
try:
|
||
from .wechat_auto_login import WeChatAutoLogin
|
||
return WeChatAutoLogin(self).check_login_state()
|
||
except Exception as exc:
|
||
return {"logged_in": False, "state": "error", "error": str(exc)}
|
||
|
||
def login_by_password(
|
||
self,
|
||
account: str = "",
|
||
phone: str = "",
|
||
password: str = "",
|
||
**kwargs,
|
||
) -> Dict[str, Any]:
|
||
"""微信号/手机号 + 密码登录 [u2]"""
|
||
try:
|
||
from .wechat_auto_login import WeChatAutoLogin
|
||
acct = (account or phone or kwargs.get("username") or "").strip()
|
||
pwd = password or kwargs.get("pwd") or ""
|
||
return WeChatAutoLogin(self).login_by_password(acct, pwd)
|
||
except Exception as exc:
|
||
return {"success": False, "error": str(exc)}
|
||
|
||
def ensure_logged_in(
|
||
self,
|
||
account: str = "",
|
||
phone: str = "",
|
||
password: str = "",
|
||
**kwargs,
|
||
) -> Dict[str, Any]:
|
||
"""未登录则按配置/参数自动登录 [u2]"""
|
||
acct = (account or phone or kwargs.get("username") or "").strip()
|
||
pwd = password or kwargs.get("pwd") or ""
|
||
if not acct or not pwd:
|
||
try:
|
||
import os
|
||
import yaml
|
||
cfg_path = os.path.join(
|
||
os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))),
|
||
"config", "wechat_login.yaml",
|
||
)
|
||
if os.path.isfile(cfg_path):
|
||
with open(cfg_path, "r", encoding="utf-8") as f:
|
||
cfg = yaml.safe_load(f) or {}
|
||
dev_id = getattr(getattr(self, "d", None), "serial", None) or ""
|
||
dev_cfg = (cfg.get("devices") or {}).get(dev_id) or {}
|
||
default = cfg.get("default") or {}
|
||
acct = acct or dev_cfg.get("account") or default.get("account") or ""
|
||
pwd = pwd or dev_cfg.get("password") or default.get("password") or ""
|
||
except Exception as exc:
|
||
logger.warning(f"[ensure_logged_in] 读配置失败: {exc}")
|
||
try:
|
||
from .wechat_auto_login import WeChatAutoLogin
|
||
return WeChatAutoLogin(self).ensure_logged_in(acct, pwd)
|
||
except Exception as exc:
|
||
return {"success": False, "error": str(exc)}
|
||
|
||
def unblock_with_sms(self, phone: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""短信验证解封 [u2, ~8s]"""
|
||
return self.unblock_account(**kwargs)
|
||
|
||
def unblock_via_customer_service(
|
||
self,
|
||
reason: str = "",
|
||
phone: str = "",
|
||
wxid: str = "",
|
||
max_rounds: int = 20,
|
||
chat_interval_sec: int = 8,
|
||
use_web_search: bool = True,
|
||
**kwargs,
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
联系客服解封（自动化全链路 + AI 持续对话）[u2 + AI Brain]
|
||
|
||
链路：设置 → 账号与安全 → 微信安全中心 → 联系客服 → 客服会话/小程序
|
||
→ 抓客服消息 → AI 生成回复 → 发送 → 循环直到解封/拒绝/超时
|
||
|
||
参数:
|
||
reason: 账号被封原因（用户自述，会带入 AI 上下文）
|
||
phone / wxid: 申诉时可能用到的账号信息
|
||
max_rounds: 最大对话轮次（默认 20）
|
||
chat_interval_sec: 每轮等待客服回复时间（默认 8s）
|
||
use_web_search: 是否在开始前联网搜「类似申诉经验」作为参考
|
||
|
||
返回: {
|
||
"success": bool,
|
||
"status": "success|reject|timeout|fallback|error",
|
||
"message": "客服最终消息",
|
||
"session": { 完整对话上下文 }
|
||
}
|
||
"""
|
||
import asyncio
|
||
try:
|
||
from .unblock_customer_service import (
|
||
UnblockCustomerServiceFlow,
|
||
UnblockSession,
|
||
)
|
||
except Exception as exc:
|
||
return {"success": False, "error": f"加载解封模块失败: {exc}"}
|
||
|
||
# 从 agent 反向取 AIBrain（可选，缺省时使用兜底话术）
|
||
ai_brain = None
|
||
agent = getattr(self, "agent", None) or getattr(self, "_agent", None)
|
||
if agent is not None:
|
||
ai_brain = getattr(agent, "ai_brain", None)
|
||
|
||
session = UnblockSession(
|
||
reason=reason,
|
||
phone=phone,
|
||
wxid=wxid,
|
||
max_rounds=max(1, min(int(max_rounds or 20), 50)),
|
||
chat_interval_sec=max(2, min(int(chat_interval_sec or 8), 60)),
|
||
use_web_search=bool(use_web_search),
|
||
)
|
||
flow = UnblockCustomerServiceFlow(self, ai_brain)
|
||
|
||
try:
|
||
loop = asyncio.get_event_loop()
|
||
if loop.is_running():
|
||
# 在已有事件循环中（Agent 内部）— 用 run_coroutine_threadsafe 或同步降级
|
||
import concurrent.futures
|
||
fut = asyncio.run_coroutine_threadsafe(flow.run(session), loop)
|
||
return fut.result(timeout=max_rounds * (chat_interval_sec + 4) + 60)
|
||
except RuntimeError:
|
||
pass
|
||
|
||
return asyncio.run(flow.run(session))
|
||
|
||
# ======================================================================
|
||
# 十五、支付（新增7个）
|
||
# 控制形式: u2 UI自动化（支付类操作不走Hook，安全考虑）
|
||
# ======================================================================
|
||
|
||
def send_red_packet(self, to_id: str = "", amount: str = "", message: str = "恭喜发财", **kwargs) -> Dict[str, Any]:
|
||
"""发红包 [u2, ~10s]"""
|
||
try:
|
||
guard = self._payment_guard(amount, bool(kwargs.get("confirm")), **kwargs)
|
||
if not guard.get("pass"):
|
||
return {"success": False, **guard}
|
||
if guard.get("dry_run"):
|
||
return {
|
||
"success": True,
|
||
"action": "send_red_packet",
|
||
"to_id": to_id,
|
||
"message": message,
|
||
**guard,
|
||
}
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(to_id)
|
||
self.click_text(to_id) or self.click_contains(to_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("红包")
|
||
self.sleep(1)
|
||
if amount:
|
||
self.input_text(amount)
|
||
if not guard.get("dry_run") and (self.click_text("塞钱进红包") or self.click_contains("塞钱")):
|
||
return {"success": True, "note": "已进入红包支付确认", **guard}
|
||
return {"success": True, "note": "已打开红包页面", **guard}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def transfer(self, to_id: str = "", amount: str = "", message: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""转账 [u2, ~10s]"""
|
||
try:
|
||
guard = self._payment_guard(amount, bool(kwargs.get("confirm")), **kwargs)
|
||
if not guard.get("pass"):
|
||
return {"success": False, **guard}
|
||
if guard.get("dry_run"):
|
||
return {
|
||
"success": True,
|
||
"action": "transfer",
|
||
"to_id": to_id,
|
||
"message": message,
|
||
**guard,
|
||
}
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(to_id)
|
||
self.click_text(to_id) or self.click_contains(to_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("转账")
|
||
self.sleep(1)
|
||
if amount:
|
||
self.input_text(amount)
|
||
if not guard.get("dry_run") and (self.click_text("转账") or self.click_text("确认转账")):
|
||
return {"success": True, "note": "已进入转账支付确认", **guard}
|
||
return {"success": True, "note": "已打开转账页面", **guard}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def send_transfer(self, to_id: str = "", amount: str = "", description: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""SDK 标准动作名：转账 [u2, 默认停在确认前]"""
|
||
return self.transfer(to_id=to_id, amount=amount, message=description, **kwargs)
|
||
|
||
def show_payment_code(self, **kwargs) -> Dict[str, Any]:
|
||
"""显示付款码 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("服务") or self.click_text("支付")
|
||
self.sleep(1)
|
||
self.click_text("收付款") or self.click_contains("付款码")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已显示付款码"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def receive_payment(self, amount: str = "", desc: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""收款 [u2, ~5s]"""
|
||
try:
|
||
guard = self._payment_guard(amount or "0", bool(kwargs.get("confirm")), **kwargs)
|
||
if not guard.get("pass"):
|
||
return {"success": False, **guard}
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("服务") or self.click_text("支付")
|
||
self.sleep(1)
|
||
self.click_text("收付款") or self.click_contains("付款码")
|
||
self.sleep(1)
|
||
self.click_text("二维码收款") or self.click_contains("收款")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已显示收款码", **guard}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def view_wallet(self, **kwargs) -> Dict[str, Any]:
|
||
"""查看钱包 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("服务") or self.click_text("支付")
|
||
self.sleep(1)
|
||
self.click_text("钱包")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开钱包"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def view_transactions(self, limit: int = 20, **kwargs) -> Dict[str, Any]:
|
||
"""查看账单 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("服务") or self.click_text("支付")
|
||
self.sleep(1)
|
||
self.click_text("钱包")
|
||
self.sleep(1)
|
||
self.click_text("账单") or self.click_contains("账单")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开账单页面"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def receive_red_packet(self, from_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""领取红包 [u2, ~5s]"""
|
||
try:
|
||
if not bool(kwargs.get("confirm")):
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "领取红包默认不自动点击“开”，需 confirm=true",
|
||
}
|
||
self.launch()
|
||
self.sleep(1)
|
||
if self.click_text("微信红包") or self.click_contains("红包"):
|
||
self.sleep(0.5)
|
||
self.click_text("开") or self.click_contains("开")
|
||
self.sleep(1)
|
||
return {"success": True}
|
||
return {"success": False, "error": "未找到红包"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def receive_transfer(self, msg_svr_id: str = "", from_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""领取转账 [u2, 默认停在确认前]"""
|
||
if not bool(kwargs.get("confirm")):
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "领取转账默认不自动确认收款，需 confirm=true",
|
||
}
|
||
try:
|
||
self.launch()
|
||
self.sleep(1)
|
||
if self.click_text("转账") or self.click_contains("转账"):
|
||
self.sleep(0.5)
|
||
self.click_text("确认收款") or self.click_contains("收款")
|
||
self.sleep(1)
|
||
return {"success": True}
|
||
return {"success": False, "error": "未找到转账消息"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十六、聊天设置（新增3个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def set_chat_top(self, user_id: str = "", enable: bool = True, **kwargs) -> Dict[str, Any]:
|
||
"""置顶/取消置顶聊天 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
self.click_text("置顶聊天") or self.click_contains("置顶")
|
||
self.sleep(0.5)
|
||
return {"success": True, "top": enable}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_mute_chat(self, user_id: str = "", enable: bool = True, **kwargs) -> Dict[str, Any]:
|
||
"""消息免打扰 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
self.click_text("消息免打扰") or self.click_contains("免打扰")
|
||
self.sleep(0.5)
|
||
return {"success": True, "muted": enable}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def clear_chat_history(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""清空聊天记录 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
self.click_text("清空聊天记录") or self.click_contains("清空")
|
||
self.sleep(0.5)
|
||
self.click_text("确定") or self.click_text("清空")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 十七、收藏管理（新增2个）
|
||
# 控制形式: u2 UI自动化 | Hook可读(get_favorites走SQLite Favorite.db)
|
||
# ======================================================================
|
||
|
||
def add_to_favorites(self, content_desc: str = None, **kwargs) -> Dict[str, Any]:
|
||
"""收藏消息 [u2, ~4s]"""
|
||
try:
|
||
H = self.d.info["displayHeight"]
|
||
W = self.d.info["displayWidth"]
|
||
self.long_click(int(W * 0.5), int(H * 0.5), duration=1.0)
|
||
self.sleep(0.5)
|
||
if self.click_text("收藏"):
|
||
return {"success": True}
|
||
return {"success": False, "error": "未找到收藏选项"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def delete_favorite(self, local_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""删除收藏 [u2, 默认进入收藏列表并停在确认前]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("收藏")
|
||
self.sleep(1.5)
|
||
if not bool(kwargs.get("confirm")):
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "已打开收藏列表，删除收藏需 confirm=true",
|
||
"local_id": local_id,
|
||
}
|
||
H = self.d.info["displayHeight"]
|
||
W = self.d.info["displayWidth"]
|
||
self.long_click(int(W * 0.5), int(H * 0.25), duration=1.0)
|
||
self.sleep(0.5)
|
||
self.click_text("删除") or self.click_contains("删除")
|
||
self.sleep(0.5)
|
||
self.click_text("删除") or self.click_text("确定")
|
||
return {"success": True, "local_id": local_id}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_favorites(self, limit: int = 20, **kwargs) -> Dict[str, Any]:
|
||
"""获取收藏列表 [Hook优先(SQLite Favorite.db, ~100ms), u2降级~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("收藏")
|
||
self.sleep(1.5)
|
||
items = []
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and len(t) > 2 and t not in ("收藏", "搜索", "最近使用"):
|
||
items.append({"content": t[:100]})
|
||
return {"success": True, "favorites": items[:limit]}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "favorites": []}
|
||
|
||
# ======================================================================
|
||
# 十八、小程序 & 公众号（新增2个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def open_mini_program(self, name: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""打开小程序 [u2, ~8s]"""
|
||
try:
|
||
name = name or kwargs.get("app_id", "")
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("小程序")
|
||
self.sleep(1)
|
||
if name:
|
||
self.click_text("搜索小程序") or self.click_desc("搜索")
|
||
self.sleep(0.5)
|
||
self.input_text(name)
|
||
self.sleep(1.5)
|
||
self.click_text(name) or self.click_contains(name)
|
||
self.sleep(2)
|
||
return {"success": True, "name": name}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_recent_mini_programs(self, limit: int = 20, **kwargs) -> Dict[str, Any]:
|
||
"""获取最近小程序列表 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("小程序")
|
||
self.sleep(1.5)
|
||
items = []
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and t not in ("小程序", "搜索", "最近使用", "我的小程序"):
|
||
items.append({"name": t[:80]})
|
||
return {"success": True, "mini_programs": items[:limit], "count": len(items[:limit])}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "mini_programs": []}
|
||
|
||
def share_mini_program(self, app_id: str = "", to_id: str = "", title: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""分享小程序 [u2, 默认打开小程序，不自动选择接收人]"""
|
||
if not bool(kwargs.get("confirm")):
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("小程序")
|
||
self.sleep(1)
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "已打开小程序入口，分享需 confirm=true 后搜索并发送",
|
||
"app_id": app_id,
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
res = self.open_mini_program(name=app_id or title, **kwargs)
|
||
if not res.get("success"):
|
||
return res
|
||
if not to_id or not bool(kwargs.get("confirm")):
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "已打开小程序，分享需 to_id + confirm=true 后继续",
|
||
"app_id": app_id,
|
||
}
|
||
try:
|
||
self.click_desc("更多") or self.click_text("...")
|
||
self.sleep(0.5)
|
||
self.click_text("转发") or self.click_text("发送给朋友")
|
||
self.sleep(0.8)
|
||
self.click_text(to_id) or self.click_contains(to_id)
|
||
self.sleep(0.5)
|
||
self.click_text("发送") or self.click_text("确定")
|
||
return {"success": True, "to_id": to_id, "app_id": app_id}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def follow_official_account(self, account_name: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""关注公众号 [u2, ~8s]"""
|
||
try:
|
||
account_name = account_name or kwargs.get("account_id", "")
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(account_name)
|
||
self.sleep(1)
|
||
self.click_text(account_name) or self.click_contains(account_name)
|
||
self.sleep(2)
|
||
if self.click_text("关注") or self.click_contains("关注"):
|
||
return {"success": True, "account": account_name}
|
||
return {"success": True, "note": "可能已关注"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def unfollow_official_account(self, account_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""取消关注公众号 [u2, 默认导航到公众号资料页，人工/二次确认后取消]"""
|
||
account_name = account_id or kwargs.get("account_name", "")
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(account_name)
|
||
self.sleep(1)
|
||
self.click_text(account_name) or self.click_contains(account_name)
|
||
self.sleep(1)
|
||
if not bool(kwargs.get("confirm")):
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "已打开公众号页面，取消关注需 confirm=true",
|
||
"account": account_name,
|
||
}
|
||
self.click_desc("更多") or self.click_text("...")
|
||
self.sleep(0.5)
|
||
self.click_text("不再关注") or self.click_contains("取消关注")
|
||
self.sleep(0.5)
|
||
self.click_text("不再关注") or self.click_text("确定")
|
||
return {"success": True, "account": account_name}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_official_account_articles(self, account_id: str = "", limit: int = 10, **kwargs) -> Dict[str, Any]:
|
||
"""获取/打开公众号文章列表 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(account_id)
|
||
self.sleep(1)
|
||
self.click_text(account_id) or self.click_contains(account_id)
|
||
self.sleep(2)
|
||
articles = []
|
||
try:
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(self.d.dump_hierarchy())
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and t not in ("消息", "服务", "发消息"):
|
||
articles.append({"title": t[:120]})
|
||
except Exception:
|
||
pass
|
||
return {
|
||
"success": True,
|
||
"account_id": account_id,
|
||
"articles": articles[:limit],
|
||
"count": len(articles[:limit]),
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "articles": []}
|
||
|
||
# ======================================================================
|
||
# 十九、视频号（新增6个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def _enter_video_channel(self):
|
||
"""进入视频号（内部方法）"""
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("视频号") or self.click_contains("视频号")
|
||
self.sleep(2)
|
||
|
||
def open_video_channel(self, **kwargs) -> Dict[str, Any]:
|
||
"""打开视频号 [u2, ~3s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_video_list(self, limit: int = 10, **kwargs) -> Dict[str, Any]:
|
||
"""获取视频号列表 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
videos = []
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and len(t) > 3:
|
||
videos.append({"title": t[:100]})
|
||
return {"success": True, "videos": videos[:limit]}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "videos": []}
|
||
|
||
def like_video(self, index: int = 0, **kwargs) -> Dict[str, Any]:
|
||
"""点赞视频 [u2, ~4s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
for _ in range(index):
|
||
self.swipe("up", scale=0.8)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
H = self.d.info["displayHeight"]
|
||
self.click(int(W * 0.93), int(H * 0.45))
|
||
self.sleep(0.5)
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def comment_video(self, index: int = 0, comment: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""评论视频 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
for _ in range(index):
|
||
self.swipe("up", scale=0.8)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
H = self.d.info["displayHeight"]
|
||
self.click(int(W * 0.93), int(H * 0.55))
|
||
self.sleep(1)
|
||
if comment:
|
||
self.input_text(comment)
|
||
self.sleep(0.3)
|
||
self.click_text("发送")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def reply_comment(self, video_id: str = "", comment_id: str = "", content: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""回复当前评论 [u2, 默认打开评论区并停在发送前]"""
|
||
try:
|
||
res = self.comment_video(index=int(kwargs.get("index", 0) or 0), comment=content if bool(kwargs.get("confirm")) else "")
|
||
if res.get("success") and not bool(kwargs.get("confirm")):
|
||
res.update({
|
||
"dry_run": True,
|
||
"confirm_required": bool(content),
|
||
"video_id": video_id,
|
||
"comment_id": comment_id,
|
||
"note": "已打开评论区，回复发送需 confirm=true",
|
||
})
|
||
return res
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def follow_video_creator(self, index: int = 0, **kwargs) -> Dict[str, Any]:
|
||
"""关注视频号创作者 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
for _ in range(index):
|
||
self.swipe("up", scale=0.8)
|
||
self.sleep(1)
|
||
self.click(int(self.d.info["displayWidth"] * 0.08),
|
||
int(self.d.info["displayHeight"] * 0.85))
|
||
self.sleep(1)
|
||
self.click_text("关注") or self.click_contains("关注")
|
||
self.sleep(0.5)
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def unfollow_video_creator(self, index: int = 0, **kwargs) -> Dict[str, Any]:
|
||
"""取消关注视频号创作者 [u2, 默认停在确认前]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
for _ in range(index):
|
||
self.swipe("up", scale=0.8)
|
||
self.sleep(1)
|
||
self.click(int(self.d.info["displayWidth"] * 0.08),
|
||
int(self.d.info["displayHeight"] * 0.85))
|
||
self.sleep(1)
|
||
if not bool(kwargs.get("confirm")):
|
||
return {
|
||
"success": True,
|
||
"dry_run": True,
|
||
"confirm_required": True,
|
||
"note": "已打开视频号创作者入口，取消关注需 confirm=true",
|
||
}
|
||
self.click_text("已关注") or self.click_contains("已关注")
|
||
self.sleep(0.5)
|
||
self.click_text("不再关注") or self.click_text("确定")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def share_video(self, index: int = 0, to_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""分享视频 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self._enter_video_channel()
|
||
for _ in range(index):
|
||
self.swipe("up", scale=0.8)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
H = self.d.info["displayHeight"]
|
||
self.click(int(W * 0.93), int(H * 0.65))
|
||
self.sleep(1)
|
||
if to_id:
|
||
self.click_text(to_id) or self.click_contains(to_id)
|
||
self.sleep(0.5)
|
||
self.click_text("发送") or self.click_text("确定")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十、扫一扫（新增4个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def scan_qr_code(self, **kwargs) -> Dict[str, Any]:
|
||
"""扫描二维码 [u2, ~3s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_desc("+") or self.click_text("+")
|
||
self.sleep(0.5)
|
||
self.click_text("扫一扫")
|
||
self.sleep(2)
|
||
return {"success": True, "note": "已打开扫一扫"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def scan_add_friend(self, **kwargs) -> Dict[str, Any]:
|
||
"""扫码加好友 [u2, ~3s]"""
|
||
return self.scan_qr_code(**kwargs)
|
||
|
||
def show_my_qr(self, **kwargs) -> Dict[str, Any]:
|
||
"""显示我的二维码并截图返回 [u2, ~6s]
|
||
|
||
WX-13：不仅导航到二维码页，还截图返回完整 base64 PNG（真正的二维码素材，
|
||
供 SDK/hub 展示或下游识别）。截图失败不影响导航成功标记。
|
||
"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self._goto_personal_info()
|
||
self.click_text("二维码名片") or self.click_text("我的二维码") or self.click_contains("二维码")
|
||
self.sleep(1.5)
|
||
result = {"success": True, "note": "已显示二维码"}
|
||
try:
|
||
import base64
|
||
from io import BytesIO
|
||
img = self.d.screenshot()
|
||
buf = BytesIO()
|
||
img.save(buf, format="PNG")
|
||
result["qr_image_b64"] = base64.b64encode(buf.getvalue()).decode()
|
||
result["qr_image_format"] = "png"
|
||
except Exception as _se:
|
||
result["qr_image_error"] = str(_se)
|
||
return result
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def show_group_qr(self, group_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""显示群二维码并截图返回 [u2, ~8s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(group_id)
|
||
self.click_text(group_id) or self.click_contains(group_id)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
self.click_text("群二维码") or self.click_contains("二维码")
|
||
self.sleep(1.5)
|
||
result = {"success": True, "note": "已显示群二维码", "group_id": group_id}
|
||
try:
|
||
import base64
|
||
from io import BytesIO
|
||
img = self.d.screenshot()
|
||
buf = BytesIO()
|
||
img.save(buf, format="PNG")
|
||
result["qr_image_b64"] = base64.b64encode(buf.getvalue()).decode()
|
||
result["qr_image_format"] = "png"
|
||
except Exception as _se:
|
||
result["qr_image_error"] = str(_se)
|
||
return result
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def _save_qr_image_to_album(self, image_base64: str = "", image_url: str = "") -> str:
|
||
"""WP-WX-04：二维码图落盘相册并触发媒体扫描，失败返回空串。"""
|
||
import base64
|
||
import shutil
|
||
import subprocess
|
||
import tempfile
|
||
|
||
dst = f"/sdcard/Pictures/wp_qr_{int(time.time())}.png"
|
||
try:
|
||
self.d.shell(f"mkdir -p {shlex.quote(os.path.dirname(dst))}", timeout=8)
|
||
|
||
def adb_push_fallback(local_path: str) -> bool:
|
||
adb_bin = shutil.which("adb")
|
||
serial = (
|
||
getattr(self.d, "serial", "")
|
||
or getattr(self, "device_id", "")
|
||
or os.environ.get("ANDROID_SERIAL", "")
|
||
)
|
||
if not adb_bin or not serial:
|
||
return False
|
||
try:
|
||
subprocess.run(
|
||
[adb_bin, "-s", serial, "push", local_path, dst],
|
||
check=True,
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=30,
|
||
)
|
||
return True
|
||
except Exception as adb_err:
|
||
logger.warning("save_qr_image_to_album adb push fallback failed: %s", adb_err)
|
||
return False
|
||
|
||
if image_url:
|
||
self.d.shell(
|
||
f"curl -sL -o {shlex.quote(dst)} {shlex.quote(image_url)}",
|
||
timeout=30,
|
||
)
|
||
elif image_base64:
|
||
b64 = image_base64.split(",", 1)[-1].strip()
|
||
raw = base64.b64decode(b64)
|
||
with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as f:
|
||
f.write(raw)
|
||
local = f.name
|
||
try:
|
||
pushed = False
|
||
try:
|
||
self.d.push(local, dst)
|
||
pushed = True
|
||
except Exception as push_err:
|
||
logger.warning("save_qr_image_to_album u2 push failed: %s", push_err)
|
||
if not pushed and not adb_push_fallback(local):
|
||
return ""
|
||
finally:
|
||
os.unlink(local)
|
||
else:
|
||
return ""
|
||
size_out = (self.d.shell(f"wc -c < {shlex.quote(dst)}").output or "").strip()
|
||
if not size_out or size_out == "0":
|
||
return ""
|
||
self.d.shell(
|
||
"am broadcast -a android.intent.action.MEDIA_SCANNER_SCAN_FILE "
|
||
f"-d file://{dst}",
|
||
timeout=8,
|
||
)
|
||
self.sleep(1)
|
||
return dst
|
||
except Exception as e:
|
||
logger.warning("save_qr_image_to_album failed: %s", e)
|
||
return ""
|
||
|
||
def _open_scan_pick_album_first(self) -> Dict[str, Any]:
|
||
"""微信扫一扫 → 相册 → 点第一张图。
|
||
|
||
真机验证（2026-06-27 Redmi K70 1080x2400）：
|
||
- u2 dump_hierarchy 不到微信节点（SurfaceView/自绘限制）
|
||
- u2 click 被 MIUI INJECT_EVENTS 拦截
|
||
- 改用 root am start + root input tap 坐标点击
|
||
- 权限弹窗与 AlbumPreviewUI 用多坐标兜底
|
||
"""
|
||
try:
|
||
def current_focus() -> str:
|
||
try:
|
||
return self.d.shell("dumpsys window | grep mCurrentFocus").output or ""
|
||
except Exception:
|
||
return ""
|
||
|
||
def tap(x: int, y: int, wait_sec: float = 1.5) -> None:
|
||
self.d.shell(f"su -c 'input tap {x} {y}'")
|
||
self.sleep(wait_sec)
|
||
|
||
# 1. root am start 直接打开扫一扫（绕过 u2 click "+" 菜单）
|
||
self.d.shell("su -c 'am start -n com.tencent.mm/.plugin.scanner.ui.BaseScanUI'")
|
||
self.sleep(2)
|
||
|
||
# 2. 点相册按钮（937,2012 · 真机验证）
|
||
tap(937, 2012, 2)
|
||
|
||
# 3. 处理 MIUI/微信自有权限弹窗；焦点有时不叫 GrantPermissions
|
||
focus = current_focus()
|
||
focus_norm = focus.lower()
|
||
if any(key in focus_norm for key in ("grantpermissions", "permission", "permis")):
|
||
for allow_x, allow_y in ((830, 1710), (540, 1710)):
|
||
tap(allow_x, allow_y, 1.5)
|
||
self.d.shell("su -c 'am start -n com.tencent.mm/.plugin.scanner.ui.BaseScanUI'")
|
||
self.sleep(1.5)
|
||
tap(937, 2012, 2)
|
||
focus_norm = current_focus().lower()
|
||
if "gallery" in focus_norm or "album" in focus_norm:
|
||
break
|
||
|
||
# 4. 检查是否进入相册（焦点应含 AlbumPreviewUI / AlbumPickerUI / gallery）
|
||
focus2 = current_focus()
|
||
focus2_norm = focus2.lower()
|
||
if "gallery" not in focus2_norm and "album" not in focus2_norm:
|
||
return {"success": False, "error": f"未进入相册界面（焦点={focus2[:120]})"}
|
||
|
||
# 5. 点第一张图（270,400 · 3列网格第一张 · 真机验证）
|
||
tap(270, 400, 2)
|
||
|
||
# 6. 在 AlbumPreviewUI 点"完成"按钮
|
||
ok = self.click_text("完成") or self.click_contains("完成")
|
||
if not ok:
|
||
# `/tmp/find_btn_out.txt` 像素分析显示底部中心高亮明显，优先点底部中心，再试右下/右上。
|
||
for finish_x, finish_y in ((528, 2330), (950, 2300), (950, 120)):
|
||
tap(finish_x, finish_y, 1.5)
|
||
xml_try = self.d.dump_hierarchy(compressed=True)
|
||
if any(flag in xml_try for flag in ("添加到通讯录", "发消息", "二维码已过期", "二维码无效")):
|
||
ok = True
|
||
break
|
||
|
||
xml = self.d.dump_hierarchy(compressed=True)
|
||
texts = re.findall(r'text="([^"]+)"', xml)[:15]
|
||
if any(flag in xml for flag in ("添加到通讯录", "发消息", "二维码已过期", "二维码无效")):
|
||
return {"success": True, "note": "已从相册选取图片识别二维码", "page_texts": texts}
|
||
return {
|
||
"success": False,
|
||
"error": "已选取图片但未进入二维码结果页",
|
||
"page_texts": texts,
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": f"扫一扫相册流程异常: {e}"}
|
||
|
||
def extract_qr_from_image(
|
||
self, image_base64: str = "", image_url: str = "", **kwargs
|
||
) -> Dict[str, Any]:
|
||
"""从图片识别二维码（支持 image_base64 / image_url 落盘后再扫） [u2, ~8s]"""
|
||
try:
|
||
saved = ""
|
||
if image_base64 or image_url:
|
||
saved = self._save_qr_image_to_album(
|
||
image_base64=image_base64, image_url=image_url
|
||
)
|
||
if not saved:
|
||
return {
|
||
"success": False,
|
||
"error": "图片落盘相册失败（image_base64/image_url 无效或写入失败）",
|
||
}
|
||
result = self._open_scan_pick_album_first()
|
||
if saved:
|
||
result["saved_path"] = saved
|
||
return result
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def add_friend_from_image(
|
||
self,
|
||
image_base64: str = "",
|
||
image_url: str = "",
|
||
verify_message: str = "",
|
||
**kwargs,
|
||
) -> Dict[str, Any]:
|
||
"""WP-WX-04：二维码图 → 扫一扫相册识别 → 发送好友申请。"""
|
||
try:
|
||
ext = self.extract_qr_from_image(
|
||
image_base64=image_base64, image_url=image_url
|
||
)
|
||
if not ext.get("success"):
|
||
return ext
|
||
self.sleep(2)
|
||
xml = self.d.dump_hierarchy(compressed=True)
|
||
if "添加到通讯录" in xml:
|
||
self.click_text("添加到通讯录") or self.click_contains("添加到通讯录")
|
||
self.sleep(1)
|
||
if verify_message:
|
||
xml2 = self.d.dump_hierarchy(compressed=True)
|
||
if "发送添加朋友申请" in xml2 or "申请添加朋友" in xml2:
|
||
(
|
||
self.click_text("发送添加朋友申请")
|
||
or self.click_contains("申请添加朋友")
|
||
)
|
||
self.sleep(0.5)
|
||
self.human_type(verify_message, clear=True)
|
||
self.sleep(0.3)
|
||
self.click_text("发送") or self.click_text("完成")
|
||
return {
|
||
"success": True,
|
||
"note": "已识别二维码并发送好友请求",
|
||
"verified": True,
|
||
"verify_message": verify_message,
|
||
}
|
||
if "该二维码已过期" in xml or "二维码无效" in xml:
|
||
return {"success": False, "error": "二维码已过期/无效"}
|
||
if "已是好友" in xml or ("发消息" in xml and "添加到通讯录" not in xml):
|
||
return {"success": True, "note": "对方已是好友", "already_friend": True}
|
||
texts = re.findall(r'text="([^"]+)"', xml)[:10]
|
||
return {
|
||
"success": False,
|
||
"error": "未识别到好友二维码或未进入添加页",
|
||
"page_texts": texts,
|
||
}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十一、通话（新增2个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def voice_call(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""语音通话 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("语音通话") or self.click_contains("语音通话")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已发起语音通话"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def video_call(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""视频通话 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("视频通话") or self.click_contains("视频通话")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已发起视频通话"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十二、群发/搜索/发现（新增3个）
|
||
# 控制形式: u2 UI自动化 | Hook可加速(wechat_search走SQLite)
|
||
# ======================================================================
|
||
|
||
def mass_send(self, content: str = "", user_ids: List[str] = None, **kwargs) -> Dict[str, Any]:
|
||
"""群发消息（微信自带群发助手）[u2, ~10s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("通用")
|
||
self.sleep(1)
|
||
self.click_text("辅助功能") or self.click_contains("辅助")
|
||
self.sleep(1)
|
||
self.click_text("群发助手")
|
||
self.sleep(1)
|
||
self.click_text("新建群发") or self.click_text("开始群发")
|
||
self.sleep(1)
|
||
if user_ids:
|
||
for uid in user_ids:
|
||
self.click_text(uid) or self.click_contains(uid)
|
||
self.sleep(0.3)
|
||
self.click_text("下一步") or self.click_text("完成")
|
||
self.sleep(1)
|
||
if content:
|
||
self.input_text(content)
|
||
self.sleep(0.3)
|
||
self.click_text("发送")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def wechat_search(self, keyword: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""搜一搜 [Hook优先(SQLite多表联查, ~100ms), u2降级~3s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("搜一搜") or self.click_contains("搜索")
|
||
self.sleep(1)
|
||
if keyword:
|
||
self.input_text(keyword)
|
||
self.sleep(2)
|
||
return {"success": True, "keyword": keyword}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def top_stories(self, **kwargs) -> Dict[str, Any]:
|
||
"""看一看 [u2, ~3s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("发现")
|
||
self.sleep(1)
|
||
self.click_text("看一看") or self.click_contains("看一看")
|
||
self.sleep(2)
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十三、微信运动（新增2个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def get_steps(self, **kwargs) -> Dict[str, Any]:
|
||
"""获取步数 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search("微信运动")
|
||
self.click_text("微信运动") or self.click_contains("微信运动")
|
||
self.sleep(2)
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
steps = None
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and t.isdigit() and len(t) >= 2:
|
||
steps = int(t)
|
||
break
|
||
return {"success": True, "steps": steps}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def like_steps(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""点赞步数 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search("微信运动")
|
||
self.click_text("微信运动") or self.click_contains("微信运动")
|
||
self.sleep(2)
|
||
self.click_text("步数排行榜") or self.click_contains("排行")
|
||
self.sleep(1)
|
||
if user_id:
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(0.5)
|
||
self.click_text("赞") or self.click_contains("赞")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十四、位置分享（新增2个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def send_location(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""发送位置 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
if user_id:
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("位置") or self.click_contains("位置")
|
||
self.sleep(0.5)
|
||
self.click_text("发送位置")
|
||
self.sleep(2)
|
||
self.click_text("发送")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def share_real_time_location(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""共享实时位置 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
if user_id:
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("位置") or self.click_contains("位置")
|
||
self.sleep(0.5)
|
||
self.click_text("共享实时位置")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已发起实时位置共享"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十五、表情管理（新增2个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def send_emoji(self, user_id: str = "", emoji_name: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""发送表情 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
if user_id:
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
H = self.d.info["displayHeight"]
|
||
self.click(int(W * 0.12), int(H * 0.955))
|
||
self.sleep(0.5)
|
||
if emoji_name:
|
||
self.click_text(emoji_name) or self.click_contains(emoji_name)
|
||
else:
|
||
self.click(int(W * 0.1), int(H * 0.88))
|
||
self.sleep(0.3)
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def add_custom_emoji(self, image_path: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""添加自定义表情 [u2, 默认打开添加入口]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("表情")
|
||
self.sleep(1)
|
||
self.click_text("设置") or self.click_desc("设置")
|
||
self.sleep(0.8)
|
||
self.click_text("添加的单个表情") or self.click_contains("单个表情")
|
||
self.sleep(0.8)
|
||
self.click_desc("+") or self.click_text("+")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开添加自定义表情入口", "image_path": image_path}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def get_sticker_list(self, **kwargs) -> Dict[str, Any]:
|
||
"""获取表情包列表 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("表情")
|
||
self.sleep(1)
|
||
stickers = []
|
||
ui = self.d.dump_hierarchy()
|
||
import xml.etree.ElementTree as ET
|
||
root = ET.fromstring(ui)
|
||
for elem in root.iter():
|
||
t = elem.get("text", "").strip()
|
||
if t and len(t) > 0 and t not in ("表情", "搜索", "我的表情", "更多"):
|
||
stickers.append({"name": t})
|
||
return {"success": True, "stickers": stickers[:50]}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e), "stickers": []}
|
||
|
||
# ======================================================================
|
||
# 二十六、语音/文件（新增3个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def send_voice_message(self, user_id: str = "", duration: int = 3, **kwargs) -> Dict[str, Any]:
|
||
"""发送语音消息 [u2, ~duration+3s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
if user_id:
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
W = self.d.info["displayWidth"]
|
||
H = self.d.info["displayHeight"]
|
||
if self.click_text("按住 说话") or self.click_contains("按住"):
|
||
pass
|
||
else:
|
||
self.click(int(W * 0.06), int(H * 0.955))
|
||
self.sleep(0.5)
|
||
btn_x = int(W * 0.5)
|
||
btn_y = int(H * 0.955)
|
||
self.long_click(btn_x, btn_y, duration=float(duration))
|
||
self.sleep(0.5)
|
||
return {"success": True, "duration": duration}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def send_file_from_chat(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""发送文件 [u2, ~6s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
if user_id:
|
||
self.search(user_id)
|
||
self.click_text(user_id) or self.click_contains(user_id)
|
||
self.sleep(1)
|
||
self.click_text("+") or self.click_desc("更多功能")
|
||
self.sleep(0.5)
|
||
self.click_text("文件") or self.click_contains("文件")
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已打开文件选择"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def download_file(self, user_id: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""下载文件 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.sleep(1)
|
||
if self.click_text("文件") or self.click_contains("[文件]"):
|
||
self.sleep(1)
|
||
return {"success": True, "note": "已点击文件"}
|
||
return {"success": False, "error": "未找到可下载的文件"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十七、设置管理（新增5个）
|
||
# 控制形式: u2 UI自动化
|
||
# ======================================================================
|
||
|
||
def toggle_do_not_disturb(self, enable: bool = True, **kwargs) -> Dict[str, Any]:
|
||
"""勿扰模式 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("新消息通知") or self.click_text("消息通知")
|
||
self.sleep(1)
|
||
self.click_text("勿扰模式") or self.click_contains("勿扰")
|
||
self.sleep(0.5)
|
||
return {"success": True, "dnd": enable}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def add_to_float(self, wxid: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""添加聊天浮窗 [u2, 默认进入聊天更多菜单]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.search(wxid)
|
||
self.click_text(wxid) or self.click_contains(wxid)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
self.click_text("浮窗") or self.click_contains("浮窗")
|
||
return {"success": True, "wxid": wxid}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def remove_from_float(self, wxid: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""移除浮窗 [u2]"""
|
||
return self.add_to_float(wxid=wxid, **kwargs)
|
||
|
||
def set_notification(self, type: str = "all", enable: bool = True, **kwargs) -> Dict[str, Any]:
|
||
"""设置新消息通知 [u2]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("新消息通知") or self.click_text("消息通知")
|
||
self.sleep(1)
|
||
return {"success": True, "type": type, "enable": enable, "note": "已打开新消息通知设置"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def set_chat_background(self, wxid: str = "", image_path: str = "", **kwargs) -> Dict[str, Any]:
|
||
"""设置聊天背景 [u2, 默认打开背景设置入口]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
if wxid:
|
||
self.search(wxid)
|
||
self.click_text(wxid) or self.click_contains(wxid)
|
||
self.sleep(1)
|
||
self.d.xpath('//*[@content-desc="聊天信息"]').click_exists(timeout=3)
|
||
self.sleep(1)
|
||
self.click_text("设置当前聊天背景") or self.click_contains("聊天背景")
|
||
self.sleep(1)
|
||
return {"success": True, "wxid": wxid, "image_path": image_path, "note": "已打开聊天背景入口"}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def clear_cache(self, **kwargs) -> Dict[str, Any]:
|
||
"""清理缓存 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("通用")
|
||
self.sleep(1)
|
||
self.click_text("存储空间") or self.click_contains("存储")
|
||
self.sleep(2)
|
||
self.click_text("清理") or self.click_contains("清理")
|
||
self.sleep(0.5)
|
||
self.click_text("确定")
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def check_for_update(self, **kwargs) -> Dict[str, Any]:
|
||
"""检查更新 [u2, ~5s]"""
|
||
try:
|
||
self.launch()
|
||
self.wait_for_app_ready()
|
||
self.click_text("我")
|
||
self.sleep(1)
|
||
self.click_text("设置")
|
||
self.sleep(1)
|
||
self.click_text("关于微信")
|
||
self.sleep(1)
|
||
self.click_text("检查新版本") or self.click_contains("版本")
|
||
self.sleep(2)
|
||
return {"success": True}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def logout(self, **kwargs) -> Dict[str, Any]:
|
||
"""退出登录 [root am start 深链 SettingsUI + root tap]
|
||
|
||
WeChat 8.0.69 UI 非无障碍（click_text 失效），故：
|
||
1. `su -c am start SettingsUI` 直达设置页（绕非无障碍底栏 tab 导航）；
|
||
2. root swipe 滚到底（退出登录在底部）；
|
||
3. 截图返回（qr/screenshot 供人工/下游确认 退出登录 位置）；
|
||
4. **默认非破坏**：仅当 confirm=True 才 root tap 退出登录 + 确认（真退登录，需再登凭证）。
|
||
"""
|
||
import base64
|
||
from io import BytesIO
|
||
confirm = bool(kwargs.get("confirm"))
|
||
try:
|
||
# 1) root 深链直达设置页
|
||
self.d.shell("su -c 'am start -n com.tencent.mm/.plugin.setting.ui.setting.SettingsUI'")
|
||
self.sleep(2.0)
|
||
cur = ""
|
||
try:
|
||
cur = self.d.app_current().get("activity", "")
|
||
except Exception:
|
||
pass
|
||
if "SettingsUI" not in cur:
|
||
# 回退：常规我→设置（老版本/可访问时）
|
||
self.launch(); self.wait_for_app_ready()
|
||
self.click_text("我"); self.sleep(1); self.click_text("设置"); self.sleep(1)
|
||
# 2) 滚到底（退出登录在底部）
|
||
info = self.d.info
|
||
W, H = info.get("displayWidth", 1080), info.get("displayHeight", 2400)
|
||
for _ in range(4):
|
||
if self._root_input_available():
|
||
self._root_swipe(W // 2, int(H * 0.8), W // 2, int(H * 0.25), 250)
|
||
else:
|
||
self.swipe("up", scale=0.6)
|
||
self.sleep(0.4)
|
||
# 3) 截图返回（供确认 退出登录 位置）
|
||
result = {"success": True, "navigated_to": "SettingsUI", "confirm": confirm}
|
||
try:
|
||
img = self.d.screenshot(); buf = BytesIO(); img.save(buf, format="PNG")
|
||
result["screenshot_b64"] = base64.b64encode(buf.getvalue()).decode()
|
||
except Exception as _se:
|
||
result["screenshot_error"] = str(_se)
|
||
# 4) 默认非破坏；confirm=True 才真退
|
||
if not confirm:
|
||
result["note"] = "已到设置底部(退出登录区域)，未点击退出（confirm=False，非破坏）；传 confirm=true 真退（需再登凭证）"
|
||
return result
|
||
# 真退：root tap「退出」（底部按钮，截图校准 ~H*0.89）+ 确认对话框
|
||
if self._root_input_available():
|
||
self._root_tap(W // 2, int(H * 0.89)) # 8.0.69 设置底部「退出」按钮（截图校准）
|
||
self.sleep(1.2)
|
||
self._root_tap(W // 2, int(H * 0.60)) # 确认对话框「退出登录」按钮（居中偏下）
|
||
else:
|
||
self.click_text("退出登录") or self.click_text("退出")
|
||
self.sleep(0.5)
|
||
self.click_text("退出登录") or self.click_text("确定")
|
||
self.sleep(2.0)
|
||
result["note"] = "已尝试退出登录（confirm=True）"
|
||
return result
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
def switch_account(self, **kwargs) -> Dict[str, Any]:
|
||
"""切换账号 [root am start 深链 SettingsUI + root tap「切换账号」]
|
||
|
||
WeChat 8.0.69 非无障碍：root 深链直达设置 + root swipe 滚到底 + root tap
|
||
「切换账号」(截图校准 ~H*0.79)。打开账号选择页（非破坏，不实际切换）。
|
||
"""
|
||
import base64
|
||
from io import BytesIO
|
||
try:
|
||
self.d.shell("su -c 'am start -n com.tencent.mm/.plugin.setting.ui.setting.SettingsUI'")
|
||
self.sleep(2.0)
|
||
cur = ""
|
||
try:
|
||
cur = self.d.app_current().get("activity", "")
|
||
except Exception:
|
||
pass
|
||
if "SettingsUI" not in cur:
|
||
self.launch(); self.wait_for_app_ready()
|
||
self.click_text("我"); self.sleep(1); self.click_text("设置"); self.sleep(1)
|
||
info = self.d.info
|
||
W, H = info.get("displayWidth", 1080), info.get("displayHeight", 2400)
|
||
for _ in range(4):
|
||
if self._root_input_available():
|
||
self._root_swipe(W // 2, int(H * 0.8), W // 2, int(H * 0.25), 250)
|
||
else:
|
||
self.swipe("up", scale=0.6)
|
||
self.sleep(0.4)
|
||
if self._root_input_available():
|
||
self._root_tap(W // 2, int(H * 0.79)) # 设置底部「切换账号」按钮（截图校准）
|
||
else:
|
||
self.click_text("切换账号") or self.click_contains("切换")
|
||
self.sleep(1.5)
|
||
result = {"success": True, "note": "已 root 深链导航并点击「切换账号」(打开账号选择页)"}
|
||
try:
|
||
img = self.d.screenshot(); buf = BytesIO(); img.save(buf, format="PNG")
|
||
result["screenshot_b64"] = base64.b64encode(buf.getvalue()).decode()
|
||
except Exception:
|
||
pass
|
||
return result
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|
||
|
||
# ======================================================================
|
||
# 二十八、截图（新增1个）
|
||
# 控制形式: u2直接调用设备截图API
|
||
# ======================================================================
|
||
|
||
def screenshot(self, **kwargs) -> Dict[str, Any]:
|
||
"""截图 [u2直接API, ~500ms]"""
|
||
try:
|
||
import base64
|
||
from io import BytesIO
|
||
img = self.d.screenshot()
|
||
buf = BytesIO()
|
||
img.save(buf, format="PNG")
|
||
b64 = base64.b64encode(buf.getvalue()).decode()
|
||
return {"success": True, "screenshot": b64[:100] + "...", "size": len(b64)}
|
||
except Exception as e:
|
||
return {"success": False, "error": str(e)}
|