ai 开始测试

Utils/ControlUtils.py

@@ -1,8 +1,11 @@
+import re
+
 import tidevice
 from wda import Client
 from Utils.AiUtils import AiUtils
 from Utils.LogManager import LogManager
 
+
 # 页面控制工具类
 class ControlUtils(object):
 
@@ -59,8 +62,10 @@ class ControlUtils(object):
                 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]")
+            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:
@@ -73,7 +78,7 @@ class ControlUtils(object):
     @classmethod
     def clickLike(cls, session: Client, udid):
         scale = session.scale
-        x, y = AiUtils.findImageInScreen("add",udid)
+        x, y = AiUtils.findImageInScreen("add", udid)
         print(x, y)
         if x > -1:
             print("点赞了")
@@ -108,13 +113,29 @@ class ControlUtils(object):
     # 获取主播详情页的第一个视频
     @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 = 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]').get(
+            timeout=5)
+
+        tab = session.xpath('//XCUIElementTypeButton[@name="TTKProfileTabVideoButton_0"]').get(timeout=2)
+        # 某些版本 tab.value 可能就是数量；或者 tab.label 类似 “作品 7”
+        m = re.search(r"\d+", tab.label)
+
+        num = 0
+
+        if m:
+            # 判断当前的作品的数量
+            num = int(m.group())
+            print("作品数量为：", num)
+
+        if videoCell is not None:
             videoCell.click()
             # 点击视频
-            return True
+            print("找到主页的第一个视频")
+            return True, num
         else:
-            return False
-
-
-
+            print("没有找到主页的第一个视频")
+            return False, num