diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index ee97e09..a47552f 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,11 +4,7 @@
-
-
-
-
-
+
diff --git a/Entity/__pycache__/AnchorModel.cpython-312.pyc b/Entity/__pycache__/AnchorModel.cpython-312.pyc
index 7f4f147..18285a9 100644
Binary files a/Entity/__pycache__/AnchorModel.cpython-312.pyc and b/Entity/__pycache__/AnchorModel.cpython-312.pyc differ
diff --git a/Entity/__pycache__/DeviceModel.cpython-312.pyc b/Entity/__pycache__/DeviceModel.cpython-312.pyc
index 719dd17..a451030 100644
Binary files a/Entity/__pycache__/DeviceModel.cpython-312.pyc and b/Entity/__pycache__/DeviceModel.cpython-312.pyc differ
diff --git a/Entity/__pycache__/ResultData.cpython-312.pyc b/Entity/__pycache__/ResultData.cpython-312.pyc
index 2079761..9b90579 100644
Binary files a/Entity/__pycache__/ResultData.cpython-312.pyc and b/Entity/__pycache__/ResultData.cpython-312.pyc differ
diff --git a/Entity/__pycache__/Variables.cpython-312.pyc b/Entity/__pycache__/Variables.cpython-312.pyc
index 111dc13..eff3e48 100644
Binary files a/Entity/__pycache__/Variables.cpython-312.pyc and b/Entity/__pycache__/Variables.cpython-312.pyc differ
diff --git a/Module/FlaskService.py b/Module/FlaskService.py
index 78a293d..7cdc8ed 100644
--- a/Module/FlaskService.py
+++ b/Module/FlaskService.py
@@ -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__':
diff --git a/Module/Main.py b/Module/Main.py
index a964353..8b8a9f3 100644
--- a/Module/Main.py
+++ b/Module/Main.py
@@ -36,7 +36,7 @@ def main(arg):
if __name__ == "__main__":
# 获取启动时候传递的参数
- # main(sys.argv)
+ main(sys.argv)
# 添加iOS开发包到电脑上
deployer = DevDiskImageDeployer(verbose=True)
diff --git a/Module/__pycache__/DeviceInfo.cpython-312.pyc b/Module/__pycache__/DeviceInfo.cpython-312.pyc
index 2998d25..f2ddf59 100644
Binary files a/Module/__pycache__/DeviceInfo.cpython-312.pyc and b/Module/__pycache__/DeviceInfo.cpython-312.pyc differ
diff --git a/Module/__pycache__/FlaskService.cpython-312.pyc b/Module/__pycache__/FlaskService.cpython-312.pyc
index 81b06d9..8a3cbf3 100644
Binary files a/Module/__pycache__/FlaskService.cpython-312.pyc and b/Module/__pycache__/FlaskService.cpython-312.pyc differ
diff --git a/Module/__pycache__/FlaskSubprocessManager.cpython-312.pyc b/Module/__pycache__/FlaskSubprocessManager.cpython-312.pyc
index c6d2d10..13aec83 100644
Binary files a/Module/__pycache__/FlaskSubprocessManager.cpython-312.pyc and b/Module/__pycache__/FlaskSubprocessManager.cpython-312.pyc differ
diff --git a/Module/__pycache__/Main.cpython-312.pyc b/Module/__pycache__/Main.cpython-312.pyc
index 81e8d2f..d403ad4 100644
Binary files a/Module/__pycache__/Main.cpython-312.pyc and b/Module/__pycache__/Main.cpython-312.pyc differ
diff --git a/Utils/AiUtils.py b/Utils/AiUtils.py
index e64a658..aa71610 100644
--- a/Utils/AiUtils.py
+++ b/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):
diff --git a/Utils/ControlUtils.py b/Utils/ControlUtils.py
index a333504..08a19b8 100644
--- a/Utils/ControlUtils.py
+++ b/Utils/ControlUtils.py
@@ -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()
diff --git a/Utils/IOSAIStorage.py b/Utils/IOSAIStorage.py
index 8f0a605..946b87b 100644
--- a/Utils/IOSAIStorage.py
+++ b/Utils/IOSAIStorage.py
@@ -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
diff --git a/Utils/JsonUtils.py b/Utils/JsonUtils.py
index 6dd7bab..50b0f78 100644
--- a/Utils/JsonUtils.py
+++ b/Utils/JsonUtils.py
@@ -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,
diff --git a/Utils/LogManager.py b/Utils/LogManager.py
index 0c8bb94..3981133 100644
--- a/Utils/LogManager.py
+++ b/Utils/LogManager.py
@@ -26,7 +26,7 @@ def _force_utf8_everywhere():
except Exception:
pass
-# _force_utf8_everywhere()
+_force_utf8_everywhere()
class LogManager:
"""
diff --git a/Utils/Requester.py b/Utils/Requester.py
index 62ff328..78f37cb 100644
--- a/Utils/Requester.py
+++ b/Utils/Requester.py
@@ -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:
diff --git a/Utils/ThreadManager.py b/Utils/ThreadManager.py
index 845dd47..d77e383 100644
--- a/Utils/ThreadManager.py
+++ b/Utils/ThreadManager.py
@@ -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
\ No newline at end of file
+ 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)
\ No newline at end of file
diff --git a/Utils/__pycache__/AiUtils.cpython-312.pyc b/Utils/__pycache__/AiUtils.cpython-312.pyc
index e520de6..36e8dfc 100644
Binary files a/Utils/__pycache__/AiUtils.cpython-312.pyc and b/Utils/__pycache__/AiUtils.cpython-312.pyc differ
diff --git a/Utils/__pycache__/ControlUtils.cpython-312.pyc b/Utils/__pycache__/ControlUtils.cpython-312.pyc
index e99a525..75ffa3b 100644
Binary files a/Utils/__pycache__/ControlUtils.cpython-312.pyc and b/Utils/__pycache__/ControlUtils.cpython-312.pyc differ
diff --git a/Utils/__pycache__/DevDiskImageDeployer.cpython-312.pyc b/Utils/__pycache__/DevDiskImageDeployer.cpython-312.pyc
index 6626122..984f6db 100644
Binary files a/Utils/__pycache__/DevDiskImageDeployer.cpython-312.pyc and b/Utils/__pycache__/DevDiskImageDeployer.cpython-312.pyc differ
diff --git a/Utils/__pycache__/JsonUtils.cpython-312.pyc b/Utils/__pycache__/JsonUtils.cpython-312.pyc
index 5a87e50..64e9fcf 100644
Binary files a/Utils/__pycache__/JsonUtils.cpython-312.pyc and b/Utils/__pycache__/JsonUtils.cpython-312.pyc differ
diff --git a/Utils/__pycache__/LogManager.cpython-312.pyc b/Utils/__pycache__/LogManager.cpython-312.pyc
index 3eda893..f1c24fa 100644
Binary files a/Utils/__pycache__/LogManager.cpython-312.pyc and b/Utils/__pycache__/LogManager.cpython-312.pyc differ
diff --git a/Utils/__pycache__/Requester.cpython-312.pyc b/Utils/__pycache__/Requester.cpython-312.pyc
index b41658f..3d86a4f 100644
Binary files a/Utils/__pycache__/Requester.cpython-312.pyc and b/Utils/__pycache__/Requester.cpython-312.pyc differ
diff --git a/Utils/__pycache__/ThreadManager.cpython-312.pyc b/Utils/__pycache__/ThreadManager.cpython-312.pyc
index c24066d..790ff8f 100644
Binary files a/Utils/__pycache__/ThreadManager.cpython-312.pyc and b/Utils/__pycache__/ThreadManager.cpython-312.pyc differ
diff --git a/requirements.txt b/requirements.txt
index cfac345..95e0e58 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -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
diff --git a/script/ScriptManager.py b/script/ScriptManager.py
index ea40c30..de798d8 100644
--- a/script/ScriptManager.py
+++ b/script/ScriptManager.py
@@ -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("未找到目标名称")
diff --git a/script/__pycache__/ScriptManager.cpython-312.pyc b/script/__pycache__/ScriptManager.cpython-312.pyc
index 1618b9c..a495cb4 100644
Binary files a/script/__pycache__/ScriptManager.cpython-312.pyc and b/script/__pycache__/ScriptManager.cpython-312.pyc differ