增加切换账号功能
This commit is contained in:
6
.idea/workspace.xml
generated
6
.idea/workspace.xml
generated
@@ -4,11 +4,7 @@
|
||||
<option name="autoReloadType" value="SELECTIVE" />
|
||||
</component>
|
||||
<component name="ChangeListManager">
|
||||
<list default="true" id="eceeff5e-51c1-459c-a911-d21ec090a423" name="Changes" comment="20250904-初步功能已完成">
|
||||
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
|
||||
<change beforePath="$PROJECT_DIR$/Module/DeviceInfo.py" beforeDir="false" afterPath="$PROJECT_DIR$/Module/DeviceInfo.py" afterDir="false" />
|
||||
<change beforePath="$PROJECT_DIR$/build.bat" beforeDir="false" afterPath="$PROJECT_DIR$/build.bat" afterDir="false" />
|
||||
</list>
|
||||
<list default="true" id="eceeff5e-51c1-459c-a911-d21ec090a423" name="Changes" comment="20250904-初步功能已完成" />
|
||||
<option name="SHOW_DIALOG" value="false" />
|
||||
<option name="HIGHLIGHT_CONFLICTS" value="true" />
|
||||
<option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
|
||||
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -151,6 +151,7 @@ def deviceList():
|
||||
LogManager.error("获取设备列表失败:", e)
|
||||
return ResultData(data=[]).toJson()
|
||||
|
||||
|
||||
# 传递token
|
||||
@app.route('/passToken', methods=['POST'])
|
||||
def passToken():
|
||||
@@ -158,6 +159,7 @@ def passToken():
|
||||
print(data)
|
||||
return ResultData(data="").toJson()
|
||||
|
||||
|
||||
# 获取设备应用列表
|
||||
@app.route('/deviceAppList', methods=['POST'])
|
||||
def deviceAppList():
|
||||
@@ -309,6 +311,7 @@ def passAnchorData():
|
||||
LogManager.error(e)
|
||||
return ResultData(data="", code=1001).toJson()
|
||||
|
||||
|
||||
@app.route('/followAndGreetUnion', methods=['POST'])
|
||||
def followAndGreetUnion():
|
||||
try:
|
||||
@@ -401,11 +404,13 @@ def getChatTextInfo():
|
||||
# 监控消息
|
||||
@app.route("/replyMessages", methods=['POST'])
|
||||
def monitorMessages():
|
||||
LogManager.method_info("开始监控消息,监控消息脚本启动", "监控消息")
|
||||
body = request.get_json()
|
||||
udid = body.get("udid")
|
||||
manager = ScriptManager()
|
||||
event = threading.Event()
|
||||
thread = threading.Thread(target=manager.replyMessages, args=(udid, event))
|
||||
LogManager.method_info("创建监控消息脚本线程成功", "监控消息")
|
||||
# 添加到线程管理
|
||||
ThreadManager.add(udid, thread, event)
|
||||
return ResultData(data="").toJson()
|
||||
@@ -444,7 +449,6 @@ def queryAnchorList():
|
||||
|
||||
|
||||
# 修改当前的主播列表数据
|
||||
|
||||
@app.route("/updateAnchorList", methods=['POST'])
|
||||
def updateAnchorList():
|
||||
"""
|
||||
@@ -548,7 +552,7 @@ def update_last_message():
|
||||
updated_count = JsonUtils.update_json_items(
|
||||
match={"sender": sender, "text": text}, # 匹配条件
|
||||
patch={"state": 1}, # 修改内容
|
||||
filename="log/last_message.json", # 要修改的文件
|
||||
filename="last_message.json", # 要修改的文件
|
||||
multi=False # 只改第一条匹配的
|
||||
)
|
||||
if updated_count > 0:
|
||||
@@ -566,7 +570,7 @@ def delete_last_message():
|
||||
|
||||
updated_count = JsonUtils.delete_json_items(
|
||||
match={"sender": sender, "text": text}, # 匹配条件
|
||||
filename="log/last_message.json", # 要修改的文件
|
||||
filename="last_message.json", # 要修改的文件
|
||||
multi=False # 只改第一条匹配的
|
||||
)
|
||||
if updated_count > 0:
|
||||
@@ -582,18 +586,24 @@ def stopAllTask():
|
||||
return ResultData(code, [], msg).toJson()
|
||||
|
||||
|
||||
# @app.route("/killWda", methods=['POST'])
|
||||
# def killWda():
|
||||
# data = request.get_json() # 解析 JSON
|
||||
# udid = data.get("device")
|
||||
# print(udid)
|
||||
#
|
||||
#
|
||||
# AiUtils.kill_wda(udid)
|
||||
# time.sleep(10)
|
||||
# AiUtils.launch_wda(udid)
|
||||
#
|
||||
# return ResultData(data="", msg="WDA重新启动").toJson()
|
||||
# 切换账号
|
||||
@app.route('/changeAccount', methods=['POST'])
|
||||
def changeAccount():
|
||||
body = request.get_json()
|
||||
udid = body.get("udid")
|
||||
account_id = body.get("account_id")
|
||||
|
||||
IOSAIStorage.save(account_id, f"{udid}/accountId.json")
|
||||
|
||||
# 存储到本地
|
||||
manager = ScriptManager()
|
||||
event = threading.Event()
|
||||
|
||||
# 启动脚本
|
||||
thread = threading.Thread(target=manager.changeAccount, args=(udid, event))
|
||||
# 添加到线程管理
|
||||
code, msg = ThreadManager.add(udid, thread, event)
|
||||
return ResultData(data="", code=code, message=msg).toJson()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
@@ -36,7 +36,7 @@ def main(arg):
|
||||
if __name__ == "__main__":
|
||||
|
||||
# 获取启动时候传递的参数
|
||||
# main(sys.argv)
|
||||
main(sys.argv)
|
||||
|
||||
# 添加iOS开发包到电脑上
|
||||
deployer = DevDiskImageDeployer(verbose=True)
|
||||
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
375
Utils/AiUtils.py
375
Utils/AiUtils.py
@@ -198,13 +198,13 @@ class AiUtils(object):
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
session.appium_settings({"snapshotMaxDepth": 10})
|
||||
homeButton = session.xpath("//*[@label='首页']")
|
||||
homeButton = session.xpath( "//XCUIElementTypeButton[@name='a11y_vo_home' or @label='Home' or @label='首页']")
|
||||
try:
|
||||
if homeButton.label == "首页":
|
||||
print("1.找到了")
|
||||
if homeButton.exists:
|
||||
print("找到首页了")
|
||||
return homeButton
|
||||
else:
|
||||
print("1.没找到")
|
||||
print("没找到首页")
|
||||
return None
|
||||
except Exception as e:
|
||||
print(e)
|
||||
@@ -259,11 +259,20 @@ class AiUtils(object):
|
||||
@classmethod
|
||||
def getSendMesageButton(cls, session: Client):
|
||||
|
||||
# msgButton = session.xpath(
|
||||
# '//XCUIElementTypeButton['
|
||||
# '(@name="发消息" or @label="发消息" or '
|
||||
# '@name="发送 👋" or @label="发送 👋" or '
|
||||
# '@name="消息" or @label="消息")'
|
||||
# ' and @visible="true"]'
|
||||
# )
|
||||
|
||||
msgButton = session.xpath(
|
||||
'//XCUIElementTypeButton['
|
||||
'(@name="发消息" or @label="发消息" or '
|
||||
'@name="发送 👋" or @label="发送 👋" or '
|
||||
'@name="消息" or @label="消息")'
|
||||
'@name="消息" or @label="消息" or '
|
||||
'@name="Message" or @label="Message")'
|
||||
' and @visible="true"]'
|
||||
)
|
||||
|
||||
@@ -300,6 +309,222 @@ class AiUtils(object):
|
||||
return cls.findNumber(btn.label)
|
||||
|
||||
|
||||
# # 识别当前页面的消息
|
||||
# @classmethod
|
||||
# def extract_messages_from_xml(cls, xml: str):
|
||||
# """
|
||||
# 解析 TikTok 聊天 XML,返回当前屏幕可见的消息与时间分隔:
|
||||
# [{"type":"time","text":"..."}, {"type":"msg","dir":"in|out","text":"..."}]
|
||||
# 兼容 Table / CollectionView / ScrollView;过滤系统提示/底部工具栏;可见性使用“重叠可视+容差”。
|
||||
# """
|
||||
# if not isinstance(xml, str) or not xml.strip():
|
||||
# return []
|
||||
#
|
||||
# try:
|
||||
# root = etree.fromstring(xml.encode("utf-8"))
|
||||
# except Exception:
|
||||
# return []
|
||||
#
|
||||
# # ---------- 小工具 ----------
|
||||
# def get_text(el):
|
||||
# s = (el.get('label') or el.get('name') or el.get('value') or '') or ''
|
||||
# return html.unescape(s.strip())
|
||||
#
|
||||
# def is_visible(el):
|
||||
# """无 visible 属性按可见处理;有且为 'false' 才视为不可见。"""
|
||||
# v = el.get('visible')
|
||||
# return (v is None) or (v.lower() == 'true')
|
||||
#
|
||||
# # ---------- 屏幕尺寸 ----------
|
||||
# app = root.xpath('/XCUIElementTypeApplication')
|
||||
# screen_w = cls.parse_float(app[0], 'width', 414.0) if app else 414.0
|
||||
# screen_h = cls.parse_float(app[0], 'height', 736.0) if app else 736.0
|
||||
#
|
||||
# # ---------- 主容器探测(评分选择最像聊天区的容器) ----------
|
||||
#
|
||||
# def pick_container():
|
||||
# cands = []
|
||||
# for xp, ctype in (
|
||||
# ('//XCUIElementTypeTable', 'table'),
|
||||
# ('//XCUIElementTypeCollectionView', 'collection'),
|
||||
# ('//XCUIElementTypeScrollView', 'scroll'),
|
||||
# ):
|
||||
# nodes = [n for n in root.xpath(xp) if is_visible(n)]
|
||||
# for n in nodes:
|
||||
# y = cls.parse_float(n, 'y', 0.0)
|
||||
# h = cls.parse_float(n, 'height', screen_h)
|
||||
# # Cell 数越多越像聊天列表;越靠中间越像
|
||||
# cells = n.xpath('.//XCUIElementTypeCell')
|
||||
# score = len(cells) * 10 - abs((y + h / 2) - screen_h / 2)
|
||||
# cands.append((score, n, ctype))
|
||||
# if cands:
|
||||
# cands.sort(key=lambda t: t[0], reverse=True)
|
||||
# return cands[0][1], cands[0][2]
|
||||
# return None, None
|
||||
#
|
||||
# container, container_type = pick_container()
|
||||
#
|
||||
# # ---------- 可视区(area_top, area_bot) ----------
|
||||
# if container is not None:
|
||||
# area_top = cls.parse_float(container, 'y', 0.0)
|
||||
# area_h = cls.parse_float(container, 'height', screen_h)
|
||||
# area_bot = area_top + area_h
|
||||
# else:
|
||||
# # 顶栏底缘作为上边界(选最靠上的宽>200的块)
|
||||
# blocks = [n for n in root.xpath('//XCUIElementTypeOther[@y and @height and @width>="200"]') if
|
||||
# is_visible(n)]
|
||||
# area_top = 0.0
|
||||
# if blocks:
|
||||
# blocks.sort(key=lambda n: cls.parse_float(n, 'y', 0.0))
|
||||
# b = blocks[0]
|
||||
# area_top = cls.parse_float(b, 'y', 0.0) + cls.parse_float(b, 'height', 0.0)
|
||||
# # 输入框 TextView 顶边作为下边界
|
||||
# tvs = [n for n in root.xpath('//XCUIElementTypeTextView') if is_visible(n)]
|
||||
# if tvs:
|
||||
# tvs.sort(key=lambda n: cls.parse_float(n, 'y', 0.0))
|
||||
# area_bot = cls.parse_float(tvs[-1], 'y', screen_h)
|
||||
# else:
|
||||
# area_bot = screen_h
|
||||
# if area_bot - area_top < 100:
|
||||
# area_top, area_bot = 0.0, screen_h
|
||||
#
|
||||
# def in_view(el) -> bool:
|
||||
# if not is_visible(el):
|
||||
# return False
|
||||
# y = cls.parse_float(el, 'y', -1e9)
|
||||
# h = cls.parse_float(el, 'height', 0.0)
|
||||
# by = y + h
|
||||
# tol = 8.0 # 容差,避免边缘误判
|
||||
# return not (by <= area_top + tol or y >= area_bot - tol)
|
||||
#
|
||||
# # ---------- 时间分隔(Header) ----------
|
||||
# items = []
|
||||
# for t in root.xpath('//XCUIElementTypeStaticText[contains(@traits, "Header")]'):
|
||||
# if not in_view(t):
|
||||
# continue
|
||||
# txt = get_text(t)
|
||||
# if txt:
|
||||
# items.append({'type': 'time', 'text': txt, 'y': cls.parse_float(t, 'y', 0.0)})
|
||||
#
|
||||
# # ---------- 系统提示/横幅过滤 ----------
|
||||
# EXCLUDES_LITERAL = {
|
||||
# 'Heart', 'Lol', 'ThumbsUp',
|
||||
# '分享发布内容', '视频贴纸标签页', '双击发送表情', '贴纸',
|
||||
# }
|
||||
# SYSTEM_PATTERNS = [
|
||||
# r"(消息请求已被接受|你开始了和.*的聊天|你打开了这个与.*的聊天).*"
|
||||
# r"回复时接收通知", r"开启(私信)?通知", r"开启通知",
|
||||
# r"你打开了这个与 .* 的聊天。.*隐私",
|
||||
# r"在此用户接受你的消息请求之前,你最多只能发送 ?\d+ 条消息。?",
|
||||
# r"聊天消息条数已达上限,你将无法向该用户发送消息。?",
|
||||
# r"未发送$",
|
||||
# r"Turn on (DM|message|direct message)?\s*notifications",
|
||||
# r"Enable notifications",
|
||||
# r"Get notified when .* replies",
|
||||
# r"You opened this chat .* privacy",
|
||||
# r"Only \d+ message can be sent .* accepts .* request",
|
||||
# ]
|
||||
# SYSTEM_RE = re.compile("|".join(SYSTEM_PATTERNS), re.IGNORECASE)
|
||||
#
|
||||
# # 排除底部贴纸/GIF/分享栏(通常是位于底部、较矮的一排 CollectionView)
|
||||
# def is_toolbar_like(o) -> bool:
|
||||
# txt = get_text(o)
|
||||
# if txt in EXCLUDES_LITERAL:
|
||||
# return True
|
||||
# y = cls.parse_float(o, 'y', 0.0)
|
||||
# h = cls.parse_float(o, 'height', 0.0)
|
||||
# near_bottom = (area_bot - (y + h)) < 48
|
||||
# is_short = h <= 40
|
||||
# return near_bottom and is_short
|
||||
#
|
||||
# # ---------- 收集消息候选 ----------
|
||||
# msg_nodes = []
|
||||
# if container is not None:
|
||||
# # 容器内优先找 Cell 下的文本节点(Other/StaticText/TextView)
|
||||
# cand = container.xpath(
|
||||
# './/XCUIElementTypeCell//*[self::XCUIElementTypeOther or self::XCUIElementTypeStaticText or self::XCUIElementTypeTextView]'
|
||||
# '[@y and (@name or @label or @value)]'
|
||||
# )
|
||||
# for o in cand:
|
||||
# if not in_view(o):
|
||||
# continue
|
||||
# if is_toolbar_like(o):
|
||||
# continue
|
||||
# txt = get_text(o)
|
||||
# if not txt or SYSTEM_RE.search(txt):
|
||||
# continue
|
||||
# msg_nodes.append(o)
|
||||
# else:
|
||||
# # 全局兜底:排除直接挂在 CollectionView(底部工具栏)下的节点
|
||||
# cand = root.xpath(
|
||||
# '//XCUIElementTypeOther[@y and (@name or @label or @value)]'
|
||||
# ' | //XCUIElementTypeStaticText[@y and (@name or @label or @value)]'
|
||||
# ' | //XCUIElementTypeTextView[@y and (@name or @label or @value)]'
|
||||
# )
|
||||
# for o in cand:
|
||||
# p = o.getparent()
|
||||
# if p is not None and p.get('type') == 'XCUIElementTypeCollectionView':
|
||||
# continue
|
||||
# if not in_view(o) or is_toolbar_like(o):
|
||||
# continue
|
||||
# txt = get_text(o)
|
||||
# if not txt or SYSTEM_RE.search(txt):
|
||||
# continue
|
||||
# msg_nodes.append(o)
|
||||
#
|
||||
# # ---------- 方向判定 & 组装 ----------
|
||||
# for o in msg_nodes:
|
||||
# txt = get_text(o)
|
||||
# if not txt or txt in EXCLUDES_LITERAL:
|
||||
# continue
|
||||
#
|
||||
# # 找所在 Cell(用于查头像)
|
||||
# cell = o.getparent()
|
||||
# while cell is not None and cell.get('type') != 'XCUIElementTypeCell':
|
||||
# cell = cell.getparent()
|
||||
#
|
||||
# x = cls.parse_float(o, 'x', 0.0)
|
||||
# y = cls.parse_float(o, 'y', 0.0)
|
||||
# w = cls.parse_float(o, 'width', 0.0)
|
||||
# right_edge = x + w
|
||||
#
|
||||
# direction = None
|
||||
# if cell is not None:
|
||||
# avatars = [a for a in cell.xpath(
|
||||
# './/XCUIElementTypeButton[@visible="true" and (@name="图片头像" or @label="图片头像")]'
|
||||
# ) if is_visible(a)]
|
||||
# if avatars:
|
||||
# ax = cls.parse_float(avatars[0], 'x', 0.0)
|
||||
# direction = 'in' if ax < (screen_w / 2) else 'out'
|
||||
# if direction is None:
|
||||
# direction = 'out' if right_edge > (screen_w * 0.75) else 'in'
|
||||
#
|
||||
# items.append({'type': 'msg', 'dir': direction, 'text': txt, 'y': y})
|
||||
#
|
||||
# # ---------- 排序 & 收尾 ----------
|
||||
# if items:
|
||||
# items.sort(key=lambda i: i.get('y', 0.0))
|
||||
# for it in items:
|
||||
# it.pop('y', None)
|
||||
# return items
|
||||
#
|
||||
#
|
||||
# @classmethod
|
||||
# def parse_float(cls, el, attr, default=0.0):
|
||||
# try:
|
||||
# v = el.get(attr)
|
||||
# if v is None:
|
||||
# return default
|
||||
# return float(v)
|
||||
# except Exception:
|
||||
# return default
|
||||
|
||||
@classmethod
|
||||
def parse_float(cls, el, attr, default=0.0):
|
||||
try:
|
||||
return float(el.get(attr, default))
|
||||
except Exception:
|
||||
return default
|
||||
|
||||
@classmethod
|
||||
def extract_messages_from_xml(cls, xml: str):
|
||||
@@ -310,6 +535,7 @@ class AiUtils(object):
|
||||
"""
|
||||
if not isinstance(xml, str) or not xml.strip():
|
||||
return []
|
||||
|
||||
try:
|
||||
root = etree.fromstring(xml.encode("utf-8"))
|
||||
except Exception:
|
||||
@@ -321,7 +547,6 @@ class AiUtils(object):
|
||||
return html.unescape(s.strip())
|
||||
|
||||
def is_visible(el):
|
||||
"""无 visible 属性按可见处理;有且为 'false' 才视为不可见。"""
|
||||
v = el.get('visible')
|
||||
return (v is None) or (v.lower() == 'true')
|
||||
|
||||
@@ -330,8 +555,7 @@ class AiUtils(object):
|
||||
screen_w = cls.parse_float(app[0], 'width', 414.0) if app else 414.0
|
||||
screen_h = cls.parse_float(app[0], 'height', 736.0) if app else 736.0
|
||||
|
||||
# ---------- 主容器探测(评分选择最像聊天区的容器) ----------
|
||||
|
||||
# ---------- 主容器探测 ----------
|
||||
def pick_container():
|
||||
cands = []
|
||||
for xp, ctype in (
|
||||
@@ -343,7 +567,6 @@ class AiUtils(object):
|
||||
for n in nodes:
|
||||
y = cls.parse_float(n, 'y', 0.0)
|
||||
h = cls.parse_float(n, 'height', screen_h)
|
||||
# Cell 数越多越像聊天列表;越靠中间越像
|
||||
cells = n.xpath('.//XCUIElementTypeCell')
|
||||
score = len(cells) * 10 - abs((y + h / 2) - screen_h / 2)
|
||||
cands.append((score, n, ctype))
|
||||
@@ -354,13 +577,12 @@ class AiUtils(object):
|
||||
|
||||
container, container_type = pick_container()
|
||||
|
||||
# ---------- 可视区(area_top, area_bot) ----------
|
||||
# ---------- 可视区 ----------
|
||||
if container is not None:
|
||||
area_top = cls.parse_float(container, 'y', 0.0)
|
||||
area_h = cls.parse_float(container, 'height', screen_h)
|
||||
area_bot = area_top + area_h
|
||||
else:
|
||||
# 顶栏底缘作为上边界(选最靠上的宽>200的块)
|
||||
blocks = [n for n in root.xpath('//XCUIElementTypeOther[@y and @height and @width>="200"]') if
|
||||
is_visible(n)]
|
||||
area_top = 0.0
|
||||
@@ -368,7 +590,6 @@ class AiUtils(object):
|
||||
blocks.sort(key=lambda n: cls.parse_float(n, 'y', 0.0))
|
||||
b = blocks[0]
|
||||
area_top = cls.parse_float(b, 'y', 0.0) + cls.parse_float(b, 'height', 0.0)
|
||||
# 输入框 TextView 顶边作为下边界
|
||||
tvs = [n for n in root.xpath('//XCUIElementTypeTextView') if is_visible(n)]
|
||||
if tvs:
|
||||
tvs.sort(key=lambda n: cls.parse_float(n, 'y', 0.0))
|
||||
@@ -384,10 +605,10 @@ class AiUtils(object):
|
||||
y = cls.parse_float(el, 'y', -1e9)
|
||||
h = cls.parse_float(el, 'height', 0.0)
|
||||
by = y + h
|
||||
tol = 8.0 # 容差,避免边缘误判
|
||||
tol = 8.0
|
||||
return not (by <= area_top + tol or y >= area_bot - tol)
|
||||
|
||||
# ---------- 时间分隔(Header) ----------
|
||||
# ---------- 时间分隔 ----------
|
||||
items = []
|
||||
for t in root.xpath('//XCUIElementTypeStaticText[contains(@traits, "Header")]'):
|
||||
if not in_view(t):
|
||||
@@ -402,8 +623,9 @@ class AiUtils(object):
|
||||
'分享发布内容', '视频贴纸标签页', '双击发送表情', '贴纸',
|
||||
}
|
||||
SYSTEM_PATTERNS = [
|
||||
r"(消息请求已被接受|你开始了和.*的聊天|你打开了这个与.*的聊天).*"
|
||||
r"回复时接收通知", r"开启(私信)?通知", r"开启通知",
|
||||
r"消息请求已被接受。你们可以开始聊天了。",
|
||||
r"(消息请求已被接受|你开始了和.*的聊天|你打开了这个与.*的聊天).*",
|
||||
r"开启(私信)?通知", r"开启通知",
|
||||
r"你打开了这个与 .* 的聊天。.*隐私",
|
||||
r"在此用户接受你的消息请求之前,你最多只能发送 ?\d+ 条消息。?",
|
||||
r"聊天消息条数已达上限,你将无法向该用户发送消息。?",
|
||||
@@ -413,10 +635,13 @@ class AiUtils(object):
|
||||
r"Get notified when .* replies",
|
||||
r"You opened this chat .* privacy",
|
||||
r"Only \d+ message can be sent .* accepts .* request",
|
||||
|
||||
r"此消息可能违反.*",
|
||||
r"无法发送",
|
||||
r"请告知我们"
|
||||
]
|
||||
SYSTEM_RE = re.compile("|".join(SYSTEM_PATTERNS), re.IGNORECASE)
|
||||
|
||||
# 排除底部贴纸/GIF/分享栏(通常是位于底部、较矮的一排 CollectionView)
|
||||
def is_toolbar_like(o) -> bool:
|
||||
txt = get_text(o)
|
||||
if txt in EXCLUDES_LITERAL:
|
||||
@@ -430,7 +655,6 @@ class AiUtils(object):
|
||||
# ---------- 收集消息候选 ----------
|
||||
msg_nodes = []
|
||||
if container is not None:
|
||||
# 容器内优先找 Cell 下的文本节点(Other/StaticText/TextView)
|
||||
cand = container.xpath(
|
||||
'.//XCUIElementTypeCell//*[self::XCUIElementTypeOther or self::XCUIElementTypeStaticText or self::XCUIElementTypeTextView]'
|
||||
'[@y and (@name or @label or @value)]'
|
||||
@@ -445,7 +669,6 @@ class AiUtils(object):
|
||||
continue
|
||||
msg_nodes.append(o)
|
||||
else:
|
||||
# 全局兜底:排除直接挂在 CollectionView(底部工具栏)下的节点
|
||||
cand = root.xpath(
|
||||
'//XCUIElementTypeOther[@y and (@name or @label or @value)]'
|
||||
' | //XCUIElementTypeStaticText[@y and (@name or @label or @value)]'
|
||||
@@ -465,10 +688,9 @@ class AiUtils(object):
|
||||
# ---------- 方向判定 & 组装 ----------
|
||||
for o in msg_nodes:
|
||||
txt = get_text(o)
|
||||
if not txt or txt in EXCLUDES_LITERAL:
|
||||
if not txt or txt in EXCLUDES_LITERAL or SYSTEM_RE.search(txt):
|
||||
continue
|
||||
|
||||
# 找所在 Cell(用于查头像)
|
||||
cell = o.getparent()
|
||||
while cell is not None and cell.get('type') != 'XCUIElementTypeCell':
|
||||
cell = cell.getparent()
|
||||
@@ -481,12 +703,17 @@ class AiUtils(object):
|
||||
direction = None
|
||||
if cell is not None:
|
||||
avatars = [a for a in cell.xpath(
|
||||
'.//XCUIElementTypeButton[@visible="true" and (@name="图片头像" or @label="图片头像")]'
|
||||
'.//XCUIElementTypeButton[@visible="true" and (@name="Profile photo" or @label="Profile photo")]'
|
||||
) if is_visible(a)]
|
||||
if not avatars and SYSTEM_RE.search(txt):
|
||||
continue # 没头像且系统消息,直接跳过
|
||||
if avatars:
|
||||
ax = cls.parse_float(avatars[0], 'x', 0.0)
|
||||
direction = 'in' if ax < (screen_w / 2) else 'out'
|
||||
|
||||
if direction is None:
|
||||
if w > screen_w * 0.8 and SYSTEM_RE.search(txt):
|
||||
continue
|
||||
direction = 'out' if right_edge > (screen_w * 0.75) else 'in'
|
||||
|
||||
items.append({'type': 'msg', 'dir': direction, 'text': txt, 'y': y})
|
||||
@@ -498,21 +725,16 @@ class AiUtils(object):
|
||||
it.pop('y', None)
|
||||
return items
|
||||
|
||||
@classmethod
|
||||
def parse_float(cls, el, attr, default=0.0):
|
||||
try:
|
||||
v = el.get(attr)
|
||||
if v is None:
|
||||
return default
|
||||
return float(v)
|
||||
except Exception:
|
||||
return default
|
||||
|
||||
# 从导航栏读取主播名称。找不到时返回空字符串
|
||||
|
||||
|
||||
|
||||
|
||||
@classmethod
|
||||
def get_navbar_anchor_name(cls, session, timeout: float = 2.0) -> str:
|
||||
"""只从导航栏读取主播名称。找不到时返回空字符串。"""
|
||||
# 可选:限制快照深度,提升解析速度/稳定性
|
||||
def get_navbar_anchor_name(cls, session, timeout: float = 5) -> str:
|
||||
"""从聊天页导航栏读取主播名称;找不到返回空字符串。"""
|
||||
|
||||
# 限制快照深度(可选)
|
||||
try:
|
||||
session.appium_settings({"snapshotMaxDepth": 22})
|
||||
except Exception:
|
||||
@@ -523,50 +745,81 @@ class AiUtils(object):
|
||||
return (info.get("label") or info.get("name") or info.get("value") or "").strip()
|
||||
|
||||
def _clean_tail(s: str) -> str:
|
||||
# 去掉末尾中英标点和空白(TikTok 昵称右侧常带一个顿号/逗号,比如 “Alina,”)
|
||||
return re.sub(r"[,、,。.\s]+$", "", s).strip()
|
||||
|
||||
# 导航栏容器:从“返回”按钮向上找最近祖先,且该祖先内包含“更多/举报”(多语言兜底)
|
||||
NAV_CONTAINER = (
|
||||
# ---- 关键修复:导航容器 ----
|
||||
# 1) “返回”可能是 Button 也可能是 Other;同时页面右上角有 “更多/举报” 按钮
|
||||
BACK_ELEM = (
|
||||
"//*[@type='XCUIElementTypeButton' or @type='XCUIElementTypeOther']"
|
||||
"[@name='返回' or @label='返回' or @name='Back' or @label='Back' "
|
||||
" or @name='戻る' or @label='戻る']"
|
||||
)
|
||||
RIGHT_MENU_BTN = (
|
||||
"//XCUIElementTypeButton"
|
||||
"[@name='返回' or @label='返回' or @name='Back' or @label='Back' or @name='戻る' or @label='戻る']"
|
||||
"/ancestor::XCUIElementTypeOther"
|
||||
"[ .//XCUIElementTypeButton"
|
||||
" [@name='更多' or @label='更多' or @name='More' or @label='More' or "
|
||||
" @name='その他' or @label='その他' or @name='詳細' or @label='詳細' or "
|
||||
" @name='举报' or @label='举报' or @name='Report' or @label='Report' or "
|
||||
" @name='報告' or @label='報告']"
|
||||
"][1]"
|
||||
"[@name='更多' or @label='更多' or @name='More' or @label='More' or "
|
||||
" @name='その他' or @label='その他' or @name='詳細' or @label='詳細' or "
|
||||
" @name='举报' or @label='举报' or @name='Report' or @label='Report' or "
|
||||
" @name='報告' or @label='報告']"
|
||||
)
|
||||
# 从“返回”向上找到最近祖先 Other,且该祖先内包含“更多/举报”按钮
|
||||
NAV_CONTAINER = (
|
||||
BACK_ELEM +
|
||||
"/ancestor::XCUIElementTypeOther[ .//XCUIElementTypeOther or .//XCUIElementTypeButton ][ ."
|
||||
+ RIGHT_MENU_BTN +
|
||||
"][1]"
|
||||
)
|
||||
|
||||
# ① 优先:可访问的 Other(自身有文本,且不含子 Button),更贴近 TikTok 的实现
|
||||
# ① 优先:在导航容器里找“可访问的 Other(有文本且不包含子 Button)”
|
||||
XPATH_TITLE_OTHER = (
|
||||
NAV_CONTAINER +
|
||||
"//XCUIElementTypeOther[@accessible='true' and count(.//XCUIElementTypeButton)=0 "
|
||||
" and (string-length(@name)>0 or string-length(@label)>0 or string-length(@value)>0)"
|
||||
"][1]"
|
||||
" and (string-length(@name)>0 or string-length(@label)>0 or string-length(@value)>0)][1]"
|
||||
)
|
||||
# ② 退路:第一个 StaticText
|
||||
|
||||
# ② 退路:导航容器内第一个有文本的 StaticText
|
||||
XPATH_TITLE_STATIC = (
|
||||
NAV_CONTAINER +
|
||||
"//XCUIElementTypeStaticText[string-length(@value)>0 or string-length(@label)>0 or string-length(@name)>0][1]"
|
||||
)
|
||||
|
||||
# 尝试 ①
|
||||
q = session.xpath(XPATH_TITLE_OTHER)
|
||||
if q.wait(timeout):
|
||||
t = _clean_tail(_text_of(q.get()))
|
||||
if t:
|
||||
return t
|
||||
# ③ 兜底:直接在“返回”与右侧菜单同一层级/附近范围内找可读的 Other(适配部分机型结构差异)
|
||||
XPATH_FALLBACK_NEAR_BACK = (
|
||||
BACK_ELEM +
|
||||
"/ancestor::XCUIElementTypeOther[1]"
|
||||
"//XCUIElementTypeOther[@accessible='true' and "
|
||||
" (string-length(@name)>0 or string-length(@label)>0 or string-length(@value)>0)]"
|
||||
"[count(.//XCUIElementTypeButton)=0][1]"
|
||||
)
|
||||
|
||||
# 尝试 ②
|
||||
q2 = session.xpath(XPATH_TITLE_STATIC)
|
||||
if q2.wait(1.0):
|
||||
t = _clean_tail(_text_of(q2.get()))
|
||||
if t:
|
||||
return t
|
||||
# ④ 兜底:昵称往往以顿号/逗号结尾(例如 “Alina,”);利用这个规律匹配
|
||||
XPATH_HINT_COMMA_END = (
|
||||
NAV_CONTAINER +
|
||||
"//*[ (contains(@name,',') or contains(@label,',') or contains(@value,',')) "
|
||||
" and string-length(@name)+string-length(@label)+string-length(@value) < 64 ]"
|
||||
"[1]"
|
||||
)
|
||||
|
||||
# ---- 查询顺序:① -> ② -> ③ -> ④ ----
|
||||
for xp, wait_s in [
|
||||
(XPATH_TITLE_OTHER, timeout),
|
||||
(XPATH_TITLE_STATIC, 1.0),
|
||||
(XPATH_FALLBACK_NEAR_BACK, 1.0),
|
||||
(XPATH_HINT_COMMA_END, 1.0),
|
||||
]:
|
||||
try:
|
||||
q = session.xpath(xp)
|
||||
if q.wait(wait_s):
|
||||
txt = _clean_tail(_text_of(q.get()))
|
||||
if txt:
|
||||
return txt
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
return ""
|
||||
|
||||
|
||||
|
||||
# 检查字符串中是否包含中文
|
||||
@classmethod
|
||||
def contains_chinese(cls, text):
|
||||
|
||||
@@ -58,14 +58,29 @@ class ControlUtils(object):
|
||||
@classmethod
|
||||
def clickBack(cls, session: Client):
|
||||
try:
|
||||
# back = session.xpath(
|
||||
# "//*[@label='返回']"
|
||||
# " | "
|
||||
# "//*[@label='返回上一屏幕']"
|
||||
# " | "
|
||||
# "//XCUIElementTypeButton[@visible='true' and @name='TTKProfileNavBarBaseItemComponent' and @label='IconChevronLeftOffsetLTR']"
|
||||
# )
|
||||
|
||||
back = session.xpath(
|
||||
"//*[@label='返回']"
|
||||
# ① 常见中文文案
|
||||
"//*[@label='返回' or @label='返回上一屏幕']"
|
||||
" | "
|
||||
"//*[@label='返回上一屏幕']"
|
||||
" | "
|
||||
"//XCUIElementTypeButton[@visible='true' and @name='TTKProfileNavBarBaseItemComponent' and @label='IconChevronLeftOffsetLTR']"
|
||||
# ② 英文 / 内部 name / 图标 label 的按钮(仅限 Button,且可见)
|
||||
"//XCUIElementTypeButton[@visible='true' and ("
|
||||
"@name='Back' or @label='Back' or " # 英文
|
||||
"@name='返回' or @label='返回' or " # 中文
|
||||
"@label='返回上一屏幕' or " # 中文另一种
|
||||
"@name='nav_bar_start_back' or " # 内部常见 name
|
||||
"(@name='TTKProfileNavBarBaseItemComponent' and @label='IconChevronLeftOffsetLTR')" # 你给的特例
|
||||
")]"
|
||||
)
|
||||
|
||||
|
||||
if back.exists:
|
||||
back.click()
|
||||
return True
|
||||
@@ -104,7 +119,8 @@ class ControlUtils(object):
|
||||
# 点击搜索
|
||||
@classmethod
|
||||
def clickSearch(cls, session: Client):
|
||||
obj = session.xpath("//*[@name='搜索']")
|
||||
# obj = session.xpath("//*[@name='搜索']")
|
||||
obj = session(xpath='//*[@name="搜索" or @label="搜索" or @name="Search" or @label="Search"]')
|
||||
try:
|
||||
if obj.exists:
|
||||
obj.click()
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
import json
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
@@ -12,18 +13,52 @@ class IOSAIStorage:
|
||||
iosai_dir.mkdir(parents=True, exist_ok=True)
|
||||
return iosai_dir
|
||||
|
||||
# @classmethod
|
||||
# def save(cls, data: dict | list, filename: str = "data.json") -> Path:
|
||||
# """
|
||||
# 存储数据到 C:/Users/<用户名>/IOSAI/filename
|
||||
# """
|
||||
# file_path = cls._get_iosai_dir() / filename
|
||||
# try:
|
||||
# with open(file_path, "w", encoding="utf-8") as f:
|
||||
# json.dump(data, f, ensure_ascii=False, indent=2)
|
||||
# print(f"[IOSAIStorage] 已保存到: {file_path}")
|
||||
# except Exception as e:
|
||||
# print(f"[IOSAIStorage] 写入失败: {e}")
|
||||
# return file_path
|
||||
|
||||
@classmethod
|
||||
def save(cls, data: dict | list, filename: str = "data.json") -> Path:
|
||||
"""
|
||||
存储数据到 C:/Users/<用户名>/IOSAI/filename
|
||||
"""
|
||||
def save(cls, data: dict | list, filename: str = "data.json", mode: str = "overwrite") -> Path:
|
||||
file_path = cls._get_iosai_dir() / filename
|
||||
try:
|
||||
with open(file_path, "w", encoding="utf-8") as f:
|
||||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||||
print(f"[IOSAIStorage] 已保存到: {file_path}")
|
||||
except Exception as e:
|
||||
print(f"[IOSAIStorage] 写入失败: {e}")
|
||||
file_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
def _load_json():
|
||||
try:
|
||||
return json.loads(file_path.read_text("utf-8"))
|
||||
except Exception:
|
||||
return {} if isinstance(data, dict) else []
|
||||
|
||||
if mode == "merge" and isinstance(data, dict):
|
||||
old = _load_json()
|
||||
if not isinstance(old, dict):
|
||||
old = {}
|
||||
old.update(data)
|
||||
to_write = old
|
||||
elif mode == "append" and isinstance(data, list):
|
||||
old = _load_json()
|
||||
if not isinstance(old, list):
|
||||
old = []
|
||||
old.extend(data)
|
||||
to_write = old
|
||||
else:
|
||||
to_write = data # 覆盖
|
||||
|
||||
# 原子写入
|
||||
tmp = file_path.with_suffix(file_path.suffix + ".tmp")
|
||||
with open(tmp, "w", encoding="utf-8") as f:
|
||||
json.dump(to_write, f, ensure_ascii=False, indent=2)
|
||||
os.replace(tmp, file_path)
|
||||
print(f"[IOSAIStorage] 已写入: {file_path}")
|
||||
return file_path
|
||||
|
||||
@classmethod
|
||||
|
||||
@@ -201,7 +201,10 @@ class JsonUtils:
|
||||
if not isinstance(data, list):
|
||||
return []
|
||||
# 过滤 sender 为空字符串的项
|
||||
return [item for item in data if isinstance(item, dict) and item.get("sender", "").strip()]
|
||||
# return [item for item in data if isinstance(item, dict) and item.get("sender", "").strip()]
|
||||
return [item for item in data if isinstance(item, dict)]
|
||||
|
||||
|
||||
|
||||
@classmethod
|
||||
def delete_json_items(cls,
|
||||
|
||||
@@ -26,7 +26,7 @@ def _force_utf8_everywhere():
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# _force_utf8_everywhere()
|
||||
_force_utf8_everywhere()
|
||||
|
||||
class LogManager:
|
||||
"""
|
||||
|
||||
@@ -53,6 +53,34 @@ class Requester():
|
||||
except Exception as e:
|
||||
LogManager.method_error(f"翻译失败,报错的原因:{e}", "翻译失败异常")
|
||||
|
||||
|
||||
|
||||
|
||||
# 翻译
|
||||
@classmethod
|
||||
def translationToChinese(cls, msg):
|
||||
try:
|
||||
param = {
|
||||
"msg": msg,
|
||||
}
|
||||
url = "https://ai.yolozs.com/translationToChinese"
|
||||
result = requests.post(url=url, json=param, verify=False)
|
||||
|
||||
LogManager.info(f"翻译 请求的参数:{param}", "翻译")
|
||||
LogManager.info(f"翻译,状态码:{result.status_code},服务器返回的内容:{result.text}", "翻译")
|
||||
|
||||
if result.status_code != 200:
|
||||
LogManager.error(f"翻译失败,状态码:{result.status_code},服务器返回的内容:{result.text}")
|
||||
return None
|
||||
|
||||
json = result.json()
|
||||
data = json.get("data")
|
||||
return data
|
||||
except Exception as e:
|
||||
LogManager.method_error(f"翻译失败,报错的原因:{e}", "翻译失败异常")
|
||||
|
||||
|
||||
|
||||
# ai聊天
|
||||
@classmethod
|
||||
def chatToAi(cls, param):
|
||||
@@ -75,10 +103,13 @@ class Requester():
|
||||
try:
|
||||
url = "https://ai.yolozs.com/chat"
|
||||
result = requests.post(url=url, json=param, verify=False)
|
||||
|
||||
LogManager.method_info(f"ai聊天的参数:{param}", "ai聊天")
|
||||
|
||||
json = result.json()
|
||||
data = json.get("answer", {})
|
||||
session_id = json.get("conversation_id", {})
|
||||
LogManager.method_info(f"ai聊天的参数:{param},ai聊天返回的内容:{result.json()}", "ai聊天")
|
||||
LogManager.method_info(f"ai聊天返回的内容:{result.json()}", "ai聊天")
|
||||
|
||||
return data, session_id
|
||||
except Exception as e:
|
||||
|
||||
@@ -20,6 +20,7 @@ class ThreadManager:
|
||||
@classmethod
|
||||
def add(cls, udid: str, thread: threading.Thread, event: threading.Event) -> Tuple[int, str]:
|
||||
LogManager.method_info(f"准备创建任务:{udid}", "task")
|
||||
LogManager.method_info("创建线程成功","监控消息")
|
||||
with cls._lock:
|
||||
# 判断当前设备是否有任务
|
||||
if cls._tasks.get(udid, None) is not None:
|
||||
@@ -61,13 +62,19 @@ class ThreadManager:
|
||||
@classmethod
|
||||
def _kill_thread(cls, tid: int) -> bool:
|
||||
"""向原生线程 ID 抛 KeyboardInterrupt,强制跳出"""
|
||||
res = ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid),
|
||||
ctypes.py_object(KeyboardInterrupt))
|
||||
if res == 0: # 线程已不存在
|
||||
print("线程不存在")
|
||||
return False
|
||||
if res > 1: # 命中多个线程,重置
|
||||
print("命中了多个线程")
|
||||
ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid), 0)
|
||||
print("杀死线程成功")
|
||||
return True
|
||||
try:
|
||||
res = ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid),
|
||||
ctypes.py_object(KeyboardInterrupt))
|
||||
# LogManager.method_info(f"向原生线程 {tid} 抛 KeyboardInterrupt,强制跳出", "task")
|
||||
if res == 0: # 线程已不存在
|
||||
print("线程不存在")
|
||||
return False
|
||||
if res > 1: # 命中多个线程,重置
|
||||
print("命中了多个线程")
|
||||
ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid), 0)
|
||||
LogManager.method_info("杀死线程创建成功", "监控消息")
|
||||
|
||||
print("杀死线程成功")
|
||||
return True
|
||||
except Exception as e:
|
||||
print("杀死线程出现问题 错误的原因:",e)
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -1,5 +1,13 @@
|
||||
facebook_wda==1.5.1
|
||||
easyocr==1.7.2
|
||||
facebook_wda==1.5.4
|
||||
Flask==3.1.2
|
||||
flask_cors==6.0.1
|
||||
lxml==6.0.2
|
||||
numpy==2.3.3
|
||||
opencv_python==4.12.0.88
|
||||
opencv_python_headless==4.12.0.88
|
||||
portalocker==3.2.0
|
||||
psutil==7.1.0
|
||||
Requests==2.32.5
|
||||
tidevice==0.12.10
|
||||
torch==2.8.0
|
||||
|
||||
@@ -1,5 +1,7 @@
|
||||
import atexit
|
||||
import random
|
||||
import re
|
||||
import subprocess
|
||||
import threading
|
||||
import time
|
||||
from enum import Enum
|
||||
@@ -11,6 +13,7 @@ from Utils.IOSAIStorage import IOSAIStorage
|
||||
from Utils.JsonUtils import JsonUtils
|
||||
from Utils.LogManager import LogManager
|
||||
from Entity.Variables import anchorList, removeModelFromAnchorList, anchorWithSession
|
||||
# from Utils.OCRUtils import OCRUtils
|
||||
from Utils.Requester import Requester
|
||||
import Entity.Variables as ev
|
||||
|
||||
@@ -42,8 +45,11 @@ class ScriptManager():
|
||||
|
||||
# 关闭并重新打开 TikTok
|
||||
ControlUtils.closeTikTok(session, udid)
|
||||
|
||||
event.wait(timeout=1)
|
||||
|
||||
ControlUtils.openTikTok(session, udid)
|
||||
|
||||
event.wait(timeout=3)
|
||||
LogManager.method_info("养号重启tiktok", "养号", udid)
|
||||
AiUtils.makeUdidDir(udid)
|
||||
@@ -124,7 +130,7 @@ class ScriptManager():
|
||||
ControlUtils.clickLike(session, udid)
|
||||
|
||||
LogManager.method_info("继续观看视频", "养号", udid)
|
||||
videoTime = random.randint(10, 30)
|
||||
videoTime = random.randint(25, 40)
|
||||
for _ in range(videoTime):
|
||||
if event.is_set():
|
||||
break
|
||||
@@ -137,9 +143,9 @@ class ScriptManager():
|
||||
else:
|
||||
nextTime = random.randint(1, 5)
|
||||
for _ in range(nextTime):
|
||||
if event.is_set():
|
||||
break
|
||||
event.wait(timeout=1)
|
||||
if event.is_set():
|
||||
break
|
||||
event.wait(timeout=1)
|
||||
client.swipe_up()
|
||||
|
||||
except Exception as e:
|
||||
@@ -172,7 +178,9 @@ class ScriptManager():
|
||||
|
||||
# 2) 进入直播 (使用英文)
|
||||
live_button = session(
|
||||
xpath='//XCUIElementTypeButton[@name="直播"] | //XCUIElementTypeOther[@name="直播"]')
|
||||
xpath='//XCUIElementTypeButton[@name="LIVE" or @label="LIVE" or @name="直播" or @label="直播"] '
|
||||
'| //XCUIElementTypeOther[@name="LIVE" or @label="LIVE" or @name="直播" or @label="直播"]'
|
||||
)
|
||||
|
||||
if live_button.exists:
|
||||
live_button.click()
|
||||
@@ -186,7 +194,12 @@ class ScriptManager():
|
||||
break
|
||||
event.wait(timeout=1)
|
||||
|
||||
live_button = session(xpath='//XCUIElementTypeButton[@name="直播"]')
|
||||
# live_button = session(xpath='//XCUIElementTypeButton[@name="直播"]')
|
||||
live_button = session(
|
||||
xpath='//XCUIElementTypeButton[@name="LIVE" or @label="LIVE" or @name="直播" or @label="直播"] '
|
||||
'| //XCUIElementTypeOther[@name="LIVE" or @label="LIVE" or @name="直播" or @label="直播"]'
|
||||
)
|
||||
|
||||
if live_button.exists:
|
||||
continue
|
||||
|
||||
@@ -880,14 +893,26 @@ class ScriptManager():
|
||||
|
||||
print("greetNewFollowers方法执行完毕")
|
||||
|
||||
|
||||
|
||||
# 检测消息
|
||||
def replyMessages(self, udid, event):
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
|
||||
try:
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
except Exception as e:
|
||||
LogManager.method_error(f"创建wda会话异常: {e}", "检测消息", udid)
|
||||
return
|
||||
|
||||
LogManager.method_info("开始重启tiktok", "监控消息")
|
||||
ControlUtils.closeTikTok(session, udid)
|
||||
event.wait(timeout=2)
|
||||
# time.sleep(1)
|
||||
ControlUtils.openTikTok(session, udid)
|
||||
event.wait(timeout=3)
|
||||
# time.sleep(1)
|
||||
LogManager.method_info("重启tiktok成功", "监控消息")
|
||||
|
||||
while not event.is_set():
|
||||
try:
|
||||
@@ -895,16 +920,344 @@ class ScriptManager():
|
||||
self.monitorMessages(session, udid, event)
|
||||
except Exception as e:
|
||||
LogManager.method_error(f"监控消息 出现异常: {e},重新启动监控直播", "检测消息", udid)
|
||||
|
||||
LogManager.method_info(f"出现异常时,稍等再重启 TikTok 并重试 异常是: {e}", "监控消息", udid)
|
||||
|
||||
LogManager.method_info(f"出现异常,重新创建wda", "监控消息", udid)
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
|
||||
LogManager.method_info(f"重启 TikTok", "监控消息", udid)
|
||||
# 出现异常时,稍等再重启 TikTok 并重试
|
||||
ControlUtils.closeTikTok(session, udid)
|
||||
event.wait(timeout=2)
|
||||
# time.sleep(1)
|
||||
ControlUtils.openTikTok(session, udid)
|
||||
event.wait(timeout=3)
|
||||
# time.sleep(1)
|
||||
LogManager.method_info("TikTok 重启成功", "监控消息", udid)
|
||||
continue # 重新进入 while 循环,调用 monitorMessages
|
||||
|
||||
# 检查未读消息并回复
|
||||
# 此方法暂时只取最后一天进行发送给ai
|
||||
|
||||
# def monitorMessages(self, session, udid, event):
|
||||
#
|
||||
# LogManager.method_info("脚本开始执行中", "监控消息")
|
||||
#
|
||||
# # 调整节点的深度为 7
|
||||
# session.appium_settings({"snapshotMaxDepth": 7})
|
||||
#
|
||||
# el = session.xpath(
|
||||
# '//XCUIElementTypeButton[@name="a11y_vo_inbox"]'
|
||||
# ' | '
|
||||
# '//XCUIElementTypeButton[contains(@name,"收件箱")]'
|
||||
# ' | '
|
||||
# '//XCUIElementTypeButton[.//XCUIElementTypeStaticText[@value="收件箱"]]'
|
||||
# )
|
||||
#
|
||||
# # 如果收件箱有消息 则进行点击
|
||||
# if el.exists:
|
||||
#
|
||||
# try:
|
||||
# m = re.search(r'(\d+)', el.label) # 抓到的第一个数字串
|
||||
# except Exception as e:
|
||||
# LogManager.method_error(f"解析收件箱数量异常: {e}", "检测消息", udid)
|
||||
#
|
||||
# count = int(m.group(1)) if m else 0
|
||||
# if count:
|
||||
# el.click()
|
||||
# session.appium_settings({"snapshotMaxDepth": 25})
|
||||
# event.wait(timeout=3)
|
||||
# while True:
|
||||
# info_count = 0
|
||||
#
|
||||
# # 创建新的会话
|
||||
# el = session.xpath(
|
||||
# '//XCUIElementTypeButton[@name="a11y_vo_inbox"]'
|
||||
# ' | '
|
||||
# '//XCUIElementTypeButton[contains(@name,"收件箱")]'
|
||||
# ' | '
|
||||
# '//XCUIElementTypeButton[.//XCUIElementTypeStaticText[@value="收件箱"]]'
|
||||
# )
|
||||
#
|
||||
# print("el", el)
|
||||
# if not el.exists:
|
||||
# LogManager.method_error(f"检测不到收件箱", "检测消息", udid)
|
||||
# raise Exception("当前页面找不到收件箱,重启")
|
||||
# # break
|
||||
#
|
||||
# # 支持中文“收件箱”和英文“Inbox”
|
||||
# xpath_query = (
|
||||
# "//XCUIElementTypeStaticText"
|
||||
# "[@value='收件箱' or @label='收件箱' or @name='收件箱'"
|
||||
# " or @value='Inbox' or @label='Inbox' or @name='Inbox']"
|
||||
# )
|
||||
#
|
||||
# # 查找所有收件箱节点
|
||||
# inbox_nodes = session.xpath(xpath_query).find_elements()
|
||||
#
|
||||
# if len(inbox_nodes) < 2:
|
||||
# LogManager.method_error(f"当前页面不再收件箱页面,重启", "检测消息", udid)
|
||||
# raise Exception("当前页面不再收件箱页面,重启")
|
||||
#
|
||||
# m = re.search(r'(\d+)', el.label) # 抓到的第一个数字串
|
||||
# count = int(m.group(1)) if m else 0
|
||||
#
|
||||
# if not count:
|
||||
# LogManager.method_info(f"当前收件箱的总数量{count}", "检测消息", udid)
|
||||
# break
|
||||
#
|
||||
# # 新粉丝
|
||||
# xp_new_fan_badge = (
|
||||
# "//XCUIElementTypeCell[.//XCUIElementTypeLink[@name='新粉丝']]"
|
||||
# "//XCUIElementTypeStaticText[string-length(@value)>0 and translate(@value,'0123456789','')='']"
|
||||
# )
|
||||
#
|
||||
# # 活动
|
||||
# xp_activity_badge = (
|
||||
# "//XCUIElementTypeCell[.//XCUIElementTypeLink[@name='活动']]"
|
||||
# "//XCUIElementTypeStaticText[string-length(@value)>0 and translate(@value,'0123456789','')='']"
|
||||
# )
|
||||
#
|
||||
# # 系统通知
|
||||
# xp_system_badge = (
|
||||
# "//XCUIElementTypeCell[.//XCUIElementTypeLink[@name='系统通知']]"
|
||||
# "//XCUIElementTypeStaticText[string-length(@value)>0 and translate(@value,'0123456789','')='']"
|
||||
# )
|
||||
#
|
||||
# # 消息请求
|
||||
# xp_request_badge = (
|
||||
# "//XCUIElementTypeCell"
|
||||
# "[.//*[self::XCUIElementTypeLink or self::XCUIElementTypeStaticText]"
|
||||
# " [@name='消息请求' or @label='消息请求' or @value='消息请求']]"
|
||||
# "//XCUIElementTypeStaticText[string-length(@value)>0 and translate(@value,'0123456789','')='']"
|
||||
# )
|
||||
#
|
||||
# # 用户消息
|
||||
# xp_badge_numeric = (
|
||||
# "("
|
||||
# # 你的两类未读容器组件 + 数字徽标(value 纯数字)
|
||||
# "//XCUIElementTypeOther["
|
||||
# " @name='AWEIMChatListCellUnreadCountViewComponent'"
|
||||
# " or @name='TikTokIMImpl.InboxCellUnreadCountViewBuilder'"
|
||||
# "]//XCUIElementTypeStaticText[@value and translate(@value,'0123456789','')='']"
|
||||
# ")/ancestor::XCUIElementTypeCell[1]"
|
||||
# " | "
|
||||
# # 兜底:任何在 CollectionView 下、value 纯数字的徽标 → 找其最近的 Cell
|
||||
# "//XCUIElementTypeCollectionView//XCUIElementTypeStaticText"
|
||||
# "[@value and translate(@value,'0123456789','')='']"
|
||||
# "/ancestor::XCUIElementTypeCell[1]"
|
||||
# )
|
||||
#
|
||||
# try:
|
||||
# # 如果 2 秒内找不到,会抛异常
|
||||
# user_text = session.xpath(xp_badge_numeric).get(timeout=2.0)
|
||||
# val = (user_text.info.get("value") or
|
||||
# user_text.info.get("label") or
|
||||
# user_text.info.get("name"))
|
||||
# LogManager.method_info(f"用户未读数量:{val}", "检测消息", udid)
|
||||
# except Exception:
|
||||
# LogManager.method_warning("当前屏幕没有找到 用户 未读徽标数字", "检测消息", udid)
|
||||
# print("当前屏幕没有找到 用户消息 未读徽标数字", udid)
|
||||
# user_text = None
|
||||
# info_count += 1
|
||||
#
|
||||
# if user_text:
|
||||
#
|
||||
# user_text.tap()
|
||||
# event.wait(timeout=3)
|
||||
#
|
||||
# xml = session.source()
|
||||
# msgs = AiUtils.extract_messages_from_xml(xml)
|
||||
#
|
||||
# text_list = ['What do you think of my live stream?',
|
||||
# 'What do you think makes my streams special?',
|
||||
# 'Do you think I’m one of the most engaging streamers you’ve seen?']
|
||||
#
|
||||
# # 检测出对方发的最后一条信息
|
||||
#
|
||||
#
|
||||
# last_msg = next((item['text'] for item in reversed(msgs) if item['type'] == 'msg'),
|
||||
# random.choice(text_list))
|
||||
#
|
||||
# LogManager.method_info(f"检测到对方最后发送的消息:{last_msg}", "检测消息", udid)
|
||||
#
|
||||
# isLanguage = AiUtils.is_language(last_msg)
|
||||
# if isLanguage:
|
||||
# # LogManager.method_info(f":{last_msg}", "检测消息", udid)
|
||||
#
|
||||
# last_msg_text = last_msg
|
||||
# else:
|
||||
# LogManager.method_info(f"对方发送的消息不是语言,随机挑选作为最后一条进行回复:{last_msg}",
|
||||
# "检测消息", udid)
|
||||
# last_msg_text = random.choice(text_list)
|
||||
#
|
||||
# if AiUtils.contains_chinese(last_msg_text):
|
||||
# LogManager.method_info(f"需要翻译:{last_msg_text}, 即将进行翻译", "检测消息", udid)
|
||||
# last_msg_text = Requester.translation(last_msg_text)
|
||||
# LogManager.method_info(f"翻译成功:{last_msg_text}, ", "检测消息", udid)
|
||||
#
|
||||
# # 向ai发送信息
|
||||
# # 获取主播的名称
|
||||
# anchor_name = AiUtils.get_navbar_anchor_name(session)
|
||||
#
|
||||
# LogManager.method_info(f"获取主播的名称:{anchor_name}", "检测消息", udid)
|
||||
# LogManager.method_info(f"获取主播最后发送的消息 进行翻译:{last_msg}", "检测消息", udid)
|
||||
# last_msg = Requester.translationToChinese(last_msg)
|
||||
# LogManager.method_info(f"翻译后的内容:{last_msg}", "检测消息", udid)
|
||||
#
|
||||
# # 找到输入框
|
||||
# last_data = [{
|
||||
# "sender": anchor_name,
|
||||
# "device": udid,
|
||||
# "text": last_msg,
|
||||
# "status": 0
|
||||
# }]
|
||||
# print(last_data)
|
||||
#
|
||||
# LogManager.method_info(f"主播最后发送的数据,传递给前端进行记录:{last_data}", "检测消息", udid)
|
||||
# JsonUtils.append_json_items(last_data, "log/last_message.json")
|
||||
#
|
||||
# # 从C盘中读取数据
|
||||
# anchorWithSession = IOSAIStorage.load()
|
||||
#
|
||||
# sel = session.xpath("//TextView")
|
||||
# if anchor_name not in anchorWithSession:
|
||||
# # 如果是第一次发消息(没有sessionId的情况)
|
||||
# LogManager.method_info(f"第一次发消息:{anchor_name},没有记忆 开始请求ai", "检测消息", udid)
|
||||
# LogManager.method_info(f"向ai发送的参数", "检测消息", udid)
|
||||
# aiResult, sessionId = Requester.chatToAi({"query": last_msg_text, "user": "1"})
|
||||
# IOSAIStorage.save({anchor_name: sessionId})
|
||||
#
|
||||
# # 找到输入框,输入ai返回出来的消息
|
||||
#
|
||||
# if sel.exists:
|
||||
# sel.click() # 聚焦
|
||||
# event.wait(timeout=1)
|
||||
# sel.clear_text()
|
||||
# sel.set_text(f"{aiResult or '暂无数据'}\n")
|
||||
# else:
|
||||
# LogManager.method_error("找不到输入框,重启", "检测消息", udid)
|
||||
# raise Exception("找不到输入框,重启")
|
||||
# else:
|
||||
# LogManager.method_info(f"不是一次发消息:{anchor_name},有记忆", "检测消息", udid)
|
||||
# # 如果不是第一次发消息(证明存储的有sessionId)
|
||||
# sessionId = anchorWithSession[anchor_name]
|
||||
#
|
||||
# # TODO: user后续添加,暂时写死
|
||||
#
|
||||
# aiResult, sessionId = Requester.chatToAi(
|
||||
# {"query": last_msg_text, "conversation_id": sessionId, "user": "1"})
|
||||
# # aiResult = response['result']
|
||||
# if sel.exists:
|
||||
# sel.click() # 聚焦
|
||||
# event.wait(timeout=1)
|
||||
# sel.clear_text()
|
||||
# sel.set_text(f"{aiResult or '暂无数据'}\n")
|
||||
#
|
||||
# LogManager.method_info(f"存储的sessionId:{anchorWithSession}", "检测消息", udid)
|
||||
# event.wait(timeout=1)
|
||||
# # 返回
|
||||
# ControlUtils.clickBack(session)
|
||||
#
|
||||
# # 重新回到收件箱页面后,强制刷新节点
|
||||
# session.appium_settings({"snapshotMaxDepth": 25})
|
||||
# event.wait(timeout=1)
|
||||
# try:
|
||||
# # 如果 2 秒内找不到,会抛异常
|
||||
# badge_text = session.xpath(xp_new_fan_badge).get(timeout=2.0)
|
||||
# val = (badge_text.info.get("value") or
|
||||
# badge_text.info.get("label") or
|
||||
# badge_text.info.get("name"))
|
||||
#
|
||||
# LogManager.method_info(f"新粉丝未读数量:{val}", "检测消息", udid)
|
||||
# if badge_text:
|
||||
# badge_text.tap()
|
||||
# event.wait(timeout=1)
|
||||
# ControlUtils.clickBack(session)
|
||||
# event.wait(timeout=1)
|
||||
# except Exception:
|
||||
# LogManager.method_warning("当前屏幕没有找到 新粉丝 未读徽标数字", "检测消息", udid)
|
||||
# print("当前屏幕没有找到 新粉丝 未读徽标数字", udid)
|
||||
# badge_text = None
|
||||
# info_count += 1
|
||||
#
|
||||
# try:
|
||||
#
|
||||
# # 如果 2 秒内找不到,会抛异常
|
||||
# badge_text = session.xpath(xp_activity_badge).get(timeout=2.0)
|
||||
# val = (badge_text.info.get("value") or
|
||||
# badge_text.info.get("label") or
|
||||
# badge_text.info.get("name"))
|
||||
# LogManager.method_info(f"活动未读数量:{val}", "检测消息", udid)
|
||||
# if badge_text:
|
||||
# badge_text.tap()
|
||||
# event.wait(timeout=1)
|
||||
# ControlUtils.clickBack(session)
|
||||
# event.wait(timeout=1)
|
||||
# except Exception:
|
||||
# LogManager.method_warning("当前屏幕没有找到 活动 未读徽标数字", "检测消息", udid)
|
||||
# print("当前屏幕没有找到 活动 未读徽标数字", udid)
|
||||
# badge_text = None
|
||||
# info_count += 1
|
||||
#
|
||||
# try:
|
||||
# # 如果 2 秒内找不到,会抛异常
|
||||
# badge_text = session.xpath(xp_system_badge).get(timeout=2.0)
|
||||
# val = (badge_text.info.get("value") or
|
||||
# badge_text.info.get("label") or
|
||||
# badge_text.info.get("name"))
|
||||
# LogManager.method_info(f"系统通知未读数量:{val}", "检测消息", udid)
|
||||
# if badge_text:
|
||||
# badge_text.tap()
|
||||
# event.wait(timeout=1)
|
||||
# ControlUtils.clickBack(session)
|
||||
# event.wait(timeout=1)
|
||||
# except Exception:
|
||||
# LogManager.method_warning("当前屏幕没有找到 系统通知 未读徽标数字", "检测消息", udid)
|
||||
# print("当前屏幕没有找到 系统通知 未读徽标数字", udid)
|
||||
# badge_text = None
|
||||
# info_count += 1
|
||||
#
|
||||
# try:
|
||||
# # 如果 2 秒内找不到,会抛异常
|
||||
# badge_text = session.xpath(xp_request_badge).get(timeout=2.0)
|
||||
# val = (badge_text.info.get("value") or
|
||||
# badge_text.info.get("label") or
|
||||
# badge_text.info.get("name"))
|
||||
# LogManager.method_info(f"消息请求未读数量:{val}", "检测消息", udid)
|
||||
# if badge_text:
|
||||
# badge_text.tap()
|
||||
# event.wait(timeout=1)
|
||||
# ControlUtils.clickBack(session)
|
||||
# event.wait(timeout=1)
|
||||
# except Exception:
|
||||
# LogManager.method_warning("当前屏幕没有找到 消息请求 未读徽标数字", "检测消息", udid)
|
||||
# print("当前屏幕没有找到 消息请求 未读徽标数字", udid)
|
||||
# badge_text = None
|
||||
# info_count += 1
|
||||
#
|
||||
# # 双击收件箱 定位到消息的位置
|
||||
# if info_count == 5:
|
||||
# r = el.bounds # 可能是命名属性,也可能是 tuple
|
||||
# cx = int((r.x + r.width / 2) if hasattr(r, "x") else (r[0] + r[2] / 2))
|
||||
# cy = int((r.y + r.height / 2) if hasattr(r, "y") else (r[1] + r[3] / 2))
|
||||
# session.double_tap(cx, cy) # 可能抛异常:方法不存在
|
||||
# LogManager.method_info(f"双击收件箱 定位到信息", "检测消息", udid)
|
||||
#
|
||||
# else:
|
||||
# return
|
||||
# else:
|
||||
# LogManager.method_error(f"检测不到收件箱", "检测消息", udid)
|
||||
# raise Exception("当前页面找不到收件箱,重启")
|
||||
|
||||
# 检测消息进行优化,检测直到我发送的内容,把这些发送给ai
|
||||
|
||||
def monitorMessages(self, session, udid, event):
|
||||
|
||||
LogManager.method_info("脚本开始执行中", "监控消息")
|
||||
|
||||
# 调整节点的深度为 7
|
||||
session.appium_settings({"snapshotMaxDepth": 7})
|
||||
|
||||
@@ -918,7 +1271,12 @@ class ScriptManager():
|
||||
|
||||
# 如果收件箱有消息 则进行点击
|
||||
if el.exists:
|
||||
m = re.search(r'(\d+)', el.label) # 抓到的第一个数字串
|
||||
|
||||
try:
|
||||
m = re.search(r'(\d+)', el.label) # 抓到的第一个数字串
|
||||
except Exception as e:
|
||||
LogManager.method_error(f"解析收件箱数量异常: {e}", "检测消息", udid)
|
||||
|
||||
count = int(m.group(1)) if m else 0
|
||||
if count:
|
||||
el.click()
|
||||
@@ -1019,6 +1377,7 @@ class ScriptManager():
|
||||
info_count += 1
|
||||
|
||||
if user_text:
|
||||
|
||||
user_text.tap()
|
||||
event.wait(timeout=3)
|
||||
|
||||
@@ -1028,46 +1387,46 @@ class ScriptManager():
|
||||
text_list = ['What do you think of my live stream?',
|
||||
'What do you think makes my streams special?',
|
||||
'Do you think I’m one of the most engaging streamers you’ve seen?']
|
||||
# 检测出对方发的最后一条信息
|
||||
|
||||
# 检测出对方发的最后一条信息,
|
||||
|
||||
# 获取了最后一条消息
|
||||
last_msg = next((item['text'] for item in reversed(msgs) if item['type'] == 'msg'),
|
||||
random.choice(text_list))
|
||||
|
||||
LogManager.method_info(f"检测到对方最后发送的消息:{last_msg}", "检测消息", udid)
|
||||
|
||||
isLanguage = AiUtils.is_language(last_msg)
|
||||
if isLanguage:
|
||||
# LogManager.method_info(f":{last_msg}", "检测消息", udid)
|
||||
# 如果最后一条消息不是文字,随机取出一条当做最后一条消息,最后一条消息是last_msg_text
|
||||
|
||||
isLanguage = AiUtils.is_language(last_msg)
|
||||
|
||||
if isLanguage:
|
||||
last_msg_text = last_msg
|
||||
else:
|
||||
LogManager.method_info(f"对方发送的消息不是语言,随机挑选作为最后一条进行回复:{last_msg}",
|
||||
"检测消息", udid)
|
||||
last_msg_text = random.choice(text_list)
|
||||
|
||||
if AiUtils.contains_chinese(last_msg_text):
|
||||
LogManager.method_info(f"需要翻译:{last_msg_text}, 即将进行翻译", "检测消息", udid)
|
||||
last_msg_text = Requester.translation(last_msg_text)
|
||||
LogManager.method_info(f"翻译成功:{last_msg_text}, ", "检测消息", udid)
|
||||
|
||||
# 向ai发送信息
|
||||
# 获取主播的名称
|
||||
anchor_name = AiUtils.get_navbar_anchor_name(session)
|
||||
|
||||
LogManager.method_info(f"获取主播的名称:{anchor_name}", "检测消息", udid)
|
||||
LogManager.method_info(f"获取主播最后发送的消息 进行翻译:{last_msg}", "检测消息", udid)
|
||||
last_msg = Requester.translation(last_msg, "中国")
|
||||
LogManager.method_info(f"翻译后的内容:{last_msg}", "检测消息", udid)
|
||||
chinese_last_msg_text = Requester.translationToChinese(last_msg_text)
|
||||
LogManager.method_info(f"翻译中文后的内容,交给前端进行展示:{chinese_last_msg_text}", "检测消息",
|
||||
udid)
|
||||
|
||||
# 找到输入框
|
||||
last_data = [{
|
||||
"sender": anchor_name,
|
||||
"device": udid,
|
||||
"text": last_msg,
|
||||
"text": chinese_last_msg_text,
|
||||
"status": 0
|
||||
}]
|
||||
print(last_data)
|
||||
|
||||
LogManager.method_info(f"主播最后发送的数据,传递给前端进行记录:{last_data}", "检测消息", udid)
|
||||
LogManager.method_info(f"主播最后发送的数据,传递给前端进行记录:{chinese_last_msg_text}",
|
||||
"检测消息", udid)
|
||||
JsonUtils.append_json_items(last_data, "log/last_message.json")
|
||||
|
||||
# 从C盘中读取数据
|
||||
@@ -1075,11 +1434,12 @@ class ScriptManager():
|
||||
|
||||
sel = session.xpath("//TextView")
|
||||
if anchor_name not in anchorWithSession:
|
||||
|
||||
# 如果是第一次发消息(没有sessionId的情况)
|
||||
LogManager.method_info(f"第一次发消息:{anchor_name},没有记忆 开始请求ai", "检测消息", udid)
|
||||
LogManager.method_info(f"向ai发送的参数", "检测消息", udid)
|
||||
LogManager.method_info(f"向ai发送的参数: 文本为:{last_msg_text}", "检测消息", udid)
|
||||
aiResult, sessionId = Requester.chatToAi({"query": last_msg_text, "user": "1"})
|
||||
IOSAIStorage.save({anchor_name: sessionId})
|
||||
IOSAIStorage.save({anchor_name: sessionId}, mode="merge")
|
||||
|
||||
# 找到输入框,输入ai返回出来的消息
|
||||
|
||||
@@ -1098,6 +1458,8 @@ class ScriptManager():
|
||||
|
||||
# TODO: user后续添加,暂时写死
|
||||
|
||||
LogManager.method_info(f"向ai发送的参数: 文本为:{last_msg_text}", "检测消息", udid)
|
||||
|
||||
aiResult, sessionId = Requester.chatToAi(
|
||||
{"query": last_msg_text, "conversation_id": sessionId, "user": "1"})
|
||||
# aiResult = response['result']
|
||||
@@ -1115,6 +1477,7 @@ class ScriptManager():
|
||||
# 重新回到收件箱页面后,强制刷新节点
|
||||
session.appium_settings({"snapshotMaxDepth": 25})
|
||||
event.wait(timeout=1)
|
||||
|
||||
try:
|
||||
# 如果 2 秒内找不到,会抛异常
|
||||
badge_text = session.xpath(xp_new_fan_badge).get(timeout=2.0)
|
||||
@@ -1203,7 +1566,6 @@ class ScriptManager():
|
||||
LogManager.method_error(f"检测不到收件箱", "检测消息", udid)
|
||||
raise Exception("当前页面找不到收件箱,重启")
|
||||
|
||||
|
||||
# 放在 ScriptManager 类外面或 utils 里
|
||||
def interruptible_sleep(self, event: threading.Event, seconds: float, slice_: float = 1.0):
|
||||
"""把一次长 sleep 拆成 1 秒一片,随时响应 event"""
|
||||
@@ -1214,3 +1576,82 @@ class ScriptManager():
|
||||
left -= timeout
|
||||
return not event.is_set() # 返回 True 表示正常睡完,False 被中断
|
||||
|
||||
# 切换账号
|
||||
def changeAccount(self, udid, event):
|
||||
|
||||
LogManager.method_info("开始进行切换账号", "切换账号", udid)
|
||||
|
||||
client = wda.USBClient(udid)
|
||||
session = client.session()
|
||||
|
||||
# 重启进行切换账号
|
||||
ControlUtils.closeTikTok(session, udid)
|
||||
# event.wait(timeout=2)
|
||||
time.sleep(1)
|
||||
ControlUtils.openTikTok(session, udid)
|
||||
|
||||
# 使用pop取出列表中的第一个元素,
|
||||
account_ids = IOSAIStorage.load(f"{udid}/accountId.json")
|
||||
account_id = account_ids.pop(0)
|
||||
# 再放到末尾
|
||||
account_ids.append(account_id)
|
||||
# 重新进行保存覆盖
|
||||
IOSAIStorage.save(account_ids, f"{udid}/accountId.json")
|
||||
|
||||
LogManager.method_info("重启进行切换账号", "切换账号", udid)
|
||||
|
||||
session.appium_settings({"snapshotMaxDepth": 15})
|
||||
|
||||
user_home = session.xpath('//XCUIElementTypeButton[@name="a11y_vo_profile" or @label="主页"]')
|
||||
LogManager.method_info("检测主页按钮", "切换账号", udid)
|
||||
if user_home.exists:
|
||||
LogManager.method_info("检测主页按钮成功,点击主页", "切换账号", udid)
|
||||
user_home.click()
|
||||
|
||||
session.appium_settings({"snapshotMaxDepth": 25})
|
||||
|
||||
LogManager.method_info("检测切换账号按钮", "切换账号", udid)
|
||||
check_account_btn = session.xpath('//XCUIElementTypeButton[@name="切换账号" or @label="切换账号"]')
|
||||
if check_account_btn.exists:
|
||||
LogManager.method_info("检测切换账号按钮成功,点击切换账号", "切换账号", udid)
|
||||
check_account_btn.click()
|
||||
|
||||
# 使用截图进行保存,保存进行图像的识别
|
||||
|
||||
try:
|
||||
img = client.screenshot()
|
||||
base_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
|
||||
resource_dir = os.path.join(base_dir, "resources", udid)
|
||||
os.makedirs(resource_dir, exist_ok=True)
|
||||
filePath = os.path.join(resource_dir, "home.png")
|
||||
img.save(filePath)
|
||||
LogManager.method_info(f"保存屏幕图像成功 -> {filePath}", "养号", udid)
|
||||
print("保存了背景图:", filePath)
|
||||
event.wait(timeout=1)
|
||||
except Exception as e:
|
||||
LogManager.method_info(f"截图或保存失败,失败原因:{e}", "养号", udid)
|
||||
raise Exception("截图或保存失败,重启养号功能")
|
||||
|
||||
# 使用EasyOcr进行识别,
|
||||
|
||||
# 取出图片路径
|
||||
image_path = AiUtils.imagePathWithName(udid, "home") # 替换为你的图像路径
|
||||
|
||||
# ocr = OCRUtils(langs=['ch_sim', 'en'], force_gpu=None)
|
||||
# print(f"image_path:{image_path}")
|
||||
#
|
||||
# # 进行识别,切换
|
||||
# hit = ocr.find_best(image_path, account_id, match_mode='fuzzy', weight_sim=0.75)
|
||||
#
|
||||
# if hit:
|
||||
# print(f"[BEST] {hit.text} | sim={hit.sim:.2f} conf={hit.conf:.2f} score={hit.score:.2f}")
|
||||
# print(f"bbox={hit.bbox}, center={hit.center}")
|
||||
# # 做一次缩放映射
|
||||
# mapped_hit = hit.mapped(1 / 6, 1 / 6)
|
||||
# print(f"center(/6)={mapped_hit.center}")
|
||||
#
|
||||
# # 传给 session.tap
|
||||
# session.tap(mapped_hit.center[0], mapped_hit.center[1])
|
||||
#
|
||||
# else:
|
||||
# print("未找到目标名称")
|
||||
|
||||
Binary file not shown.
Reference in New Issue
Block a user