diff --git a/Entity/Variables.py b/Entity/Variables.py index ed852e1..c02f801 100644 --- a/Entity/Variables.py +++ b/Entity/Variables.py @@ -10,6 +10,8 @@ WdaAppBundleId = "com.vv.wda.xctrunner" anchorList: list[AnchorModel] = [] # 线程锁 anchorListLock = threading.Lock() +# 账号token +accountToken = None # 安全删除数据 def removeModelFromAnchorList(model: AnchorModel): @@ -21,4 +23,4 @@ def addModelToAnchorList(models: list[Dict[str, Any]]): with anchorListLock: for dic in models: obj = AnchorModel.dictToModel(dic) - anchorList.append(obj) \ No newline at end of file + anchorList.append(obj) diff --git a/Flask/FlaskService.py b/Flask/FlaskService.py index c7cb3a9..3660597 100644 --- a/Flask/FlaskService.py +++ b/Flask/FlaskService.py @@ -14,7 +14,7 @@ from Entity.ResultData import ResultData from Utils.ControlUtils import ControlUtils from Utils.ThreadManager import ThreadManager from script.ScriptManager import ScriptManager -from Entity.AnchorModel import AnchorModel +from Entity.Variables import accountToken from Entity.Variables import anchorList, addModelToAnchorList app = Flask(__name__) @@ -68,6 +68,12 @@ def start_socket_listener(): listener_thread = threading.Thread(target=start_socket_listener, daemon=True) listener_thread.start() +@app.route('/passToken', methods=['POST']) +def passToken(): + data = request.get_json() + accountToken = data['token'] + return ResultData(data="").toJson() + # 获取设备列表 @app.route('/deviceList', methods=['GET']) def deviceList(): @@ -173,6 +179,19 @@ def growAccount(): ThreadManager.add(udid, thread, event) return ResultData(data="").toJson() +# 观看直播 +@app.route("/watchLiveForGrowth", methods=['POST']) +def watchLiveForGrowth(): + body = request.get_json() + udid = body.get("udid") + manager = ScriptManager() + event = threading.Event() + thread = threading.Thread(target=manager.watchLiveForGrowth, args=(udid, event)) + thread.start() + # 添加到线程管理 + ThreadManager.add(udid, thread, event) + return ResultData(data="").toJson() + # 停止脚本 @app.route("/stopScript", methods=['POST']) def stopScript(): diff --git a/Module/Main.py b/Module/Main.py index 9a7d82c..07b442b 100644 --- a/Module/Main.py +++ b/Module/Main.py @@ -4,15 +4,14 @@ from Module.DeviceInfo import Deviceinfo from Module.FlaskSubprocessManager import FlaskSubprocessManager from Utils.LogManager import LogManager +# 项目入口 if __name__ == "__main__": - + # 清空日志 LogManager.clearLogs() time.sleep(1) - print("启动flask") manager = FlaskSubprocessManager.get_instance() manager.start() - print("启动主线程") info = Deviceinfo() info.startDeviceListener() \ No newline at end of file diff --git a/Utils/AiUtils.py b/Utils/AiUtils.py index 53c982a..661442f 100644 --- a/Utils/AiUtils.py +++ b/Utils/AiUtils.py @@ -5,6 +5,7 @@ import cv2 import numpy as np import wda from Utils.LogManager import LogManager +import xml.etree.ElementTree as ET # 工具类 class AiUtils(object): @@ -209,10 +210,10 @@ class AiUtils(object): # 查找关闭按钮 @classmethod - def findCloseButton(cls,udid="eca000fcb6f55d7ed9b4c524055214c26a7de7aa"): + def findLiveCloseButton(cls,udid="eca000fcb6f55d7ed9b4c524055214c26a7de7aa"): client = wda.USBClient(udid) session = client.session() - session.appium_settings({"snapshotMaxDepth": 0}) + session.appium_settings({"snapshotMaxDepth": 10}) r = session.xpath("//XCUIElementTypeButton[@name='关闭屏幕']") try: if r.label == "关闭屏幕": @@ -223,4 +224,22 @@ class AiUtils(object): print(e) return None -# AiUtils.screenshot() \ No newline at end of file + # 获取直播间窗口数量 + @classmethod + def count_add_by_xml(cls, session): + xml = session.source() + root = ET.fromstring(xml) + return sum( + 1 for e in root.iter() + if e.get('type') in ('XCUIElementTypeButton', 'XCUIElementTypeImage') + and (e.get('name') == '添加' or e.get('label') == '添加') + and (e.get('visible') in (None, 'true')) + ) + + # 获取当前屏幕上的节点 + @classmethod + def getCurrentScreenSource(cls): + client = wda.USBClient("eca000fcb6f55d7ed9b4c524055214c26a7de7aa") + print(client.source()) + +# AiUtils.getCurrentScreenSource() \ No newline at end of file diff --git a/Utils/ControlUtils.py b/Utils/ControlUtils.py index a7bc6f0..751faaa 100644 --- a/Utils/ControlUtils.py +++ b/Utils/ControlUtils.py @@ -52,34 +52,34 @@ class ControlUtils(object): # 返回 @classmethod def clickBack(cls, session: Client): - x, y = AiUtils.findImageInScreen("back") - if x > 0: - r = session.swipe_right() - if r.status == 0: - return True + try: + x, y = AiUtils.findImageInScreen("back") + if x > 0: + r = session.swipe_right() + if r.status == 0: + return True + else: + return False else: + print("本页面无法返回") return False - else: - print("本页面无法返回") + except Exception as e: + print(e) return False # 点赞 @classmethod def clickLike(cls, session: Client, udid): scale = session.scale - x, y = AiUtils.findImageInScreen(udid,"add") - print(x,y) + x, y = AiUtils.findImageInScreen("add",udid) + print(x, y) if x > -1: - print("找到目标了") - r = session.click(x // scale, y // scale + 50) - if r.status == 0: - return True - else: - return False + print("点赞了") + session.click(x // scale, y // scale + 50) + return True else: print("没有找到目标") return False - # 点击评论 # @classmethod # def clickComment(cls, session: Client): @@ -103,11 +103,46 @@ class ControlUtils(object): def clickSearch(cls, session: Client): obj = session.xpath("//*[@name='搜索']") try: - if obj.label == "搜索": - return obj - else: - return None + if obj.exists: + obj.click() + return True except Exception as e: print(e) - return None + return False + + # 获取主播详情页的第一个视频 + @classmethod + def clickFirstVideoFromDetailPage(cls, session: Client): + videoCell = session.xpath('//Window/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[2]/Other[1]/ScrollView[1]/Other[1]/CollectionView[1]/Cell[2]') + if videoCell.exists: + videoCell.click() + # 点击视频 + return True + else: + return False + + # 获取关注按钮 + @classmethod + def clickFollowButton(cls, session: Client): + followButton = session.xpath('//Window[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[2]/Other[2]/Other[1]/Other[1]/Other[3]/Other[1]/Other[1]/Button[1]') + if followButton.exists: + print("找到了") + followButton.click() + return True + else: + print("没找到") + return False + + # 查找发消息按钮 + @classmethod + def clickSendMesageButton(cls, session: Client): + msgButton = session.xpath("//Window[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[2]/Other[2]/Other[1]/Other[1]/Other[3]/Other[1]/Other[1]") + if msgButton.exists: + print("找到了") + print(msgButton.name) + msgButton.click() + return True + else: + print("没找到") + return False diff --git a/Utils/Requester.py b/Utils/Requester.py new file mode 100644 index 0000000..9be95be --- /dev/null +++ b/Utils/Requester.py @@ -0,0 +1,7 @@ +from Entity.Variables import accountToken + +class Requester(): + + @classmethod + def printToken(cls): + print(accountToken) diff --git a/resources/eca000fcb6f55d7ed9b4c524055214c26a7de7aa/bgv.png b/resources/eca000fcb6f55d7ed9b4c524055214c26a7de7aa/bgv.png index f00ba37..f6c50dc 100644 Binary files a/resources/eca000fcb6f55d7ed9b4c524055214c26a7de7aa/bgv.png and b/resources/eca000fcb6f55d7ed9b4c524055214c26a7de7aa/bgv.png differ diff --git a/script/ScriptManager.py b/script/ScriptManager.py index f59fa0d..43085f6 100644 --- a/script/ScriptManager.py +++ b/script/ScriptManager.py @@ -60,6 +60,7 @@ class ScriptManager(): try: # 判断视频类型 addX, addY = AiUtils.findImageInScreen("add", udid) + # 多次获取结果是否一致，如果有不一致的结果就切换视频 isSame = False for i in range(2): tx, ty = AiUtils.findImageInScreen("add", udid) @@ -102,10 +103,82 @@ class ScriptManager(): # 观看直播 - def viewLive(self): - pass + def watchLiveForGrowth(self, udid, event): + client = wda.USBClient(udid) + session = client.session() - # 关注打招呼 + session.appium_settings({"snapshotMaxDepth": 15}) + # 先关闭Tik Tok + ControlUtils.closeTikTok(session, udid) + time.sleep(1) + + # 重新打开Tik Tok + ControlUtils.openTikTok(session, udid) + time.sleep(3) + # 进入直播 + live_button = session(xpath='//XCUIElementTypeButton[@name="直播"]') + if live_button.exists: + live_button.click() + time.sleep(20) + + size = session.window_size() + width, height = size.width, size.height + print(f"屏幕的宽高是：{width},{height}") + + # 可选：重新拉起 session，规避偶发 Stale 会话 + session = client.session() + # session.appium_settings({"snapshotMaxDepth": 25}) + + while not event.is_set(): + try: + time.sleep(3) + + # 如果处于 PK（分数条），直接划走 + if session(xpath='//XCUIElementTypeOther[@name="kGBLInteractionViewMatchScoreBar"]').exists: + print("✅ 当前是 PK，跳过") + session.swipe_up() + continue + + # 数直播显示区域窗口（主画面 + 连麦小窗） + count = AiUtils.count_add_by_xml(session) + print(f"检测到直播显示区域窗口数：{count}") + + if count > 1: + print("❌ 多窗口（有人连麦/分屏），划走") + session.swipe_up() + continue + else: + print("✅ 单窗口（只有一个主播），(目前是20%概率)开始点赞") + + # 点赞（仍保留中途转PK的保护） + if random.random() >= 0.89: + print("开始点赞") + for _ in range(random.randint(10, 30)): + if session(xpath='//XCUIElementTypeOther[@name="kGBLInteractionViewMatchScoreBar"]').exists: + print("❗ 中途开始 PK，停止点赞并跳过") + session.swipe_up() + break + x = width // 3 + random.randint(-10, 10) + y = height // 3 + random.randint(10, 20) + print("双击坐标：", x, y) + session.double_tap(x, y) + + print("--------------------------------------------") + time.sleep(random.randint(100, 300)) + session.swipe_up() + + except Exception as e: + print("循环异常，重试：", repr(e)) + # 轻量恢复：重新获取 session，避免因为快照或元素句柄失效卡死 + try: + session = client.session() + except Exception: + time.sleep(2) + session = client.session() + + + + # 关注打招呼以及回复主播消息 def greetNewFollowers(self, udid, needReply, event): client = wda.USBClient(udid) session = client.session() @@ -120,65 +193,123 @@ class ScriptManager(): time.sleep(3) # 点击搜索按钮 - searchButton = ControlUtils.clickSearch(session) - if searchButton is not None: - searchButton.click() - else: - print("没找到搜索按钮") - return - + ControlUtils.clickSearch(session) # 搜索框 input = session.xpath('//XCUIElementTypeSearchField') - - # 获取一个主播 - anchor = anchorList[0] - aid = anchor.anchorId - # 如果找到了输入框，就点击并且输入内容 - if input.exists: + input.click() + # 稍作停顿 + time.sleep(1) + + # 返回上一步 + def goBack(): + ControlUtils.clickBack(session) + # 搜索框 + input = session.xpath('//XCUIElementTypeSearchField') + # 如果找到了输入框，就点击并且输入内容 input.click() + # 稍作停顿 time.sleep(1) + # 删除数据 + removeModelFromAnchorList(anchor) + + + # 循环条件。1、 循环关闭 2、 数据处理完毕 + while not event.is_set() or len(anchorList) > 0: + # 获取一个主播 + anchor = anchorList[0] + aid = anchor.anchorId + + try: + input.clear_text() + except Exception as e: + print(e) + + time.sleep(2) + # 输入主播id input.set_text(aid + "\n") # 切换UI查找深度 session.appium_settings({"snapshotMaxDepth": 25}) - # 点击进入主播首页 - session.xpath( - '(//XCUIElementTypeCollectionView[@name="TTKSearchCollectionComponent"]' - '//XCUIElementTypeCell' - '//XCUIElementTypeButton[following-sibling::XCUIElementTypeButton[@name="关注" or @label="关注"]])[1]' - ).get(timeout=5).tap() + # 定位 "关注" 按钮 通过关注按钮的位置点击主播首页 + follow_button = session.xpath( + '//XCUIElementTypeCell[@index="1"]//XCUIElementTypeButton[@index="1"]').get() + if follow_button: + print("找到关注按钮！") + x = follow_button.bounds.x - 200 + y = follow_button.bounds.y + client.click(x, y) + else: + print("未找到关注按钮") time.sleep(3) - # 向上划一点 - r = client.swipe_up() - print(r) - - # 分析页面节点 - # print(session.source()) + # 找到并点击第一个视频 + cellClickResult = ControlUtils.clickFirstVideoFromDetailPage(session) # 观看主播视频 def viewAnchorVideo(): - pass + print("开始查看视频") + count = 5 + while count > 1: + print("条件满足，继续查看") + img = client.screenshot() + time.sleep(1) + filePath = f"resources/{udid}/bgv.png" + img.save(filePath) + LogManager.info("保存屏幕图像成功", udid) + print("保存了背景图") + time.sleep(1) + addX, addY = AiUtils.findImageInScreen("add", udid) + if addX != -1: + r = ControlUtils.clickLike(session, udid) + # 如果点了赞，总数量-1 否则划下一个视频，无法点赞的原因很多。指不定是什么原因。所以直接下一个视频比较好 + if r: + count -= 1 + else: + client.swipe_up() + else: + client.swipe_up() + continue + + # 假装看几秒视频 + time.sleep(10) + client.swipe_up() + if count == 1: + break + + # 点击第一个视频后的逻辑 + if cellClickResult: + print("点击了视频") + session.appium_settings({"snapshotMaxDepth": 5}) + print("重新设置了匹配深度") + viewAnchorVideo() + + # 视频看完需要点关注。 + ControlUtils.clickFollowButton(session) + time.sleep(1) + ControlUtils.clickFollowButton(session) + + # 发送一条信息 + chatInput = session.xpath("//*[className='XCUIElementTypeTextView']") + if chatInput.exists: + print("找到了") + else: + print("没找到") + + # 清除数据 + removeModelFromAnchorList(anchor) + # 流程结束 + client.swipe_left() + time.sleep(1) + else: + print("获取该主播第一个视频失败") + goBack() - viewAnchorVideo() - else: - pass - # 功能待完善 - # while not event.is_set() and len(anchorList) > 0: - # anchor = anchorList[0] - # aid = anchor.anchorId - # - # ControlUtils.clickSearch(session, udid) - # - # # 删除数据 - # removeModelFromAnchorList(anchor) - # print(len(anchorList)) # manager = ScriptManager() # manager.growAccount("eca000fcb6f55d7ed9b4c524055214c26a7de7aa")