iOSAI/Utils/ControlUtils.py

import math
import random
import re
import time

import tidevice
import wda
from wda import Client
from Utils.AiUtils import AiUtils
from Utils.LogManager import LogManager


# 页面控制工具类
class ControlUtils(object):

    # 获取设备上的app列表
    @classmethod
    def getDeviceAppList(self, udid):
        device = tidevice.Device(udid)
        # 获取已安装的应用列表
        apps = []
        for app in device.installation.iter_installed():
            apps.append({
                "name": app.get("CFBundleDisplayName", "Unknown"),
                "bundleId": app.get("CFBundleIdentifier", "Unknown"),
                "version": app.get("CFBundleShortVersionString", "Unknown"),
                "path": app.get("Path", "Unknown")
            })

        # 筛选非系统级应用（过滤掉以 com.apple 开头的系统应用）
        noSystemApps = [app for app in apps if not app["bundleId"].startswith("com.apple")]
        return noSystemApps

    # 打开Tik Tok
    @classmethod
    def openTikTok(cls, session: Client, udid):
        apps = cls.getDeviceAppList(udid)
        tk = ""
        for app in apps:
            if app.get("name", "") == "TikTok":
                tk = app.get("bundleId", "")

        currentApp = session.app_current()
        if currentApp != tk:
            session.app_start(tk)

    # 关闭Tik Tok
    @classmethod
    def closeTikTok(cls, session: Client, udid):
        apps = cls.getDeviceAppList(udid)
        tk = ""
        for app in apps:
            if app.get("name", "") == "TikTok":
                tk = app.get("bundleId", "")
        session.app_stop(tk)

    # 返回
    @classmethod
    def clickBack(cls, session: Client):
        try:
            back = session.xpath(
                "//*[@label='返回']"
                " | "
                "//*[@label='返回上一屏幕']"
                " | "
                "//XCUIElementTypeButton[@visible='true' and @name='TTKProfileNavBarBaseItemComponent' and @label='IconChevronLeftOffsetLTR']"
            )

            if back.exists:
                back.click()
                return True
            elif session.xpath("//*[@name='nav_bar_start_back']").exists:
                back = session.xpath("//*[@name='nav_bar_start_back']")
                back.click()
                return True
            elif session.xpath(
                    "//Window[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]").exists:
                back = session.xpath(
                    "//Window[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]/Other[1]")
                back.click()
                return True
            else:
                return False
        except Exception as e:
            print(e)
            return False

    # 点赞
    @classmethod
    def clickLike(cls, session: Client, udid):
        scale = session.scale
        x, y = AiUtils.findImageInScreen("add", udid)
        print(x, y)
        if x > -1:
            LogManager.info("点赞了", udid)
            session.click(x // scale, y // scale + 50)
            return True
        else:
            LogManager.info("没有找到目标", udid)
            return False

    # 点击搜索
    @classmethod
    def clickSearch(cls, session: Client):
        obj = session.xpath("//*[@name='搜索']")
        try:
            if obj.exists:
                obj.click()
                return True
        except Exception as e:
            print(e)
            return False

    # 点击收件箱按钮
    @classmethod
    def clickMsgBox(cls, session: Client):
        box = session.xpath("//XCUIElementTypeButton[name='a11y_vo_inbox']")
        if box.exists:
            box.click()
            return True
        else:
            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]')

        videoCell = session.xpath(
            '(//XCUIElementTypeCollectionView//XCUIElementTypeCell[.//XCUIElementTypeImage[@name="profile_video"]])[1]')

        tab = session.xpath(
            '//XCUIElementTypeButton[@name="TTKProfileTabVideoButton_0" or contains(@label,"作品") or contains(@name,"作品")]'
        ).get(timeout=5)        # 某些版本 tab.value 可能就是数量；或者 tab.label 类似 “作品 7”
        m = re.search(r"\d+", tab.label)

        num = 0

        if m:
            # 判断当前的作品的数量
            num = int(m.group())
            print("作品数量为：", num)

        if videoCell.exists:
            videoCell.click()
            # 点击视频
            print("找到主页的第一个视频")
            return True, num
        else:
            print("没有找到主页的第一个视频")
            return False, num

    @classmethod
    def clickFollow(cls, session, aid):
        # 1) 含“关注/已关注/Follow/Following”的首个 cell
        cell_xpath = (
            '(//XCUIElementTypeCollectionView[@name="TTKSearchCollectionComponent"]'
            '//XCUIElementTypeCell[.//XCUIElementTypeButton[@name="关注" or @name="Follow" or @name="已关注" or @name="Following"]])[1]'
        )
        cell = session.xpath(cell_xpath).get(timeout=5)

        # 2) 先试“用户信息 Button”（label/name 里包含 aid）
        profile_btn_xpath = (
            f'{cell_xpath}//XCUIElementTypeButton[contains(@label, "{aid}") or contains(@name, "{aid}")]'
        )

        try:
            profile_btn = session.xpath(profile_btn_xpath).get(timeout=3)
            profile_btn.click()
        except wda.WDAElementNotFoundError:
            # 3) 兜底：用“关注”按钮做锚点，向左偏移点击头像/用户名区域
            follow_btn_xpath = (
                f'{cell_xpath}//XCUIElementTypeButton[@name="关注" or @name="Follow" or @name="已关注" or @name="Following"]'
            )
            follow_btn = session.xpath(follow_btn_xpath).get(timeout=5)
            rect = follow_btn.bounds
            left_x = max(1, rect.x - 20)
            center_y = rect.y + rect.height // 2
            session.tap(left_x, center_y)