mangoautomation 1.1.21__py3-none-any.whl → 1.1.23__py3-none-any.whl

mangoautomation/__init__.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
 # @Project: 芒果测试平台
 # @Description: 
-# @Time   : 2024-09-07 22:15
+# @Time   : 2025-10-22 09:50
 # @Author : 毛鹏

mangoautomation/enums/__init__.py

@@ -3,11 +3,9 @@
 # @Description: 
 # @Time   : 2023-04-29 11:23
 # @Author : 毛鹏
-import sys
 
 from ..enums._ui_enum import ElementExpEnum, ElementOperationEnum, DeviceEnum, BrowserTypeEnum, DriveTypeEnum
 
-
 __all__ = [
     'ElementOperationEnum',
     'DeviceEnum',

mangoautomation/exceptions/__init__.py

@@ -3,10 +3,7 @@
 # @Description:
 # @Time   : 2023-07-07 10:14
 # @Author : 毛鹏
-import sys
 
 from ._exceptions import MangoAutomationError
 
-
-
 __all__ = ['MangoAutomationError']

mangoautomation/exceptions/error_msg.py

@@ -14,7 +14,7 @@ ERROR_MSG_0007 = (4007, '安卓设备被关闭，请检查设备')
 ERROR_MSG_0008 = (4008, '浏览器类型不正确，请联系管理员检查')
 ERROR_MSG_0009 = (4009, '浏览器路径不正确，请手动设置一个正确的浏览器路径，现在的路径：{}')
 ERROR_MSG_0010 = (4010, '浏览器被关闭，请不要手动关闭浏览器')
-ERROR_MSG_0011 = (4011, '操作元素【{}】超时，请检查元素可见但是是否可操作或调整定位方式')
+ERROR_MSG_0011 = (4011, '操作元素【{}】超时，请检查元素是否可见，或可以进行对应的操作')
 ERROR_MSG_0012 = (4012, '操作失败，请检查输入的值是否有空字符串，null或不满足传参的值')
 ERROR_MSG_0013 = (
     313, '打开url时超时，分为2个问题：1.启用的浏览器过多，应当减少浏览器并发个数；2.自行查看为啥60秒还未打开：{}')
@@ -33,7 +33,7 @@ ERROR_MSG_0025 = (4025, '无法元素对应的下标，请检查通过下标获
 ERROR_MSG_0026 = (4026, '')
 ERROR_MSG_0027 = (4027,
                   '您元素的操作内容中没有任何的数据，请检查：1.页面步骤详情中字段->元素操作值是否是空，是空可能是你删除了，也可能是执行器的操作选项没有同步需要点击执行器的同步发送缓存数据；2.元素表达式错误导致查询不到元素；')
-ERROR_MSG_0028 = (4028, '')
+ERROR_MSG_0028 = (4028, 'AI查找元素出现顶级异常，请联系管理员检查！')
 ERROR_MSG_0029 = (4029, '页面无元素【{}】，表达式：{}')
 ERROR_MSG_0030 = (4030, '断言时没有找到元素')
 ERROR_MSG_0031 = (4031, '元素未找到准备进行断言获取元素文本类容异常')
@@ -48,13 +48,13 @@ ERROR_MSG_0043 = (4043, '元素【{}】可能不存在，报错信息：{}')
 ERROR_MSG_0044 = (4044, '元素【{}】可能无法消失，报错信息：{}')
 ERROR_MSG_0045 = (4045, '设备启动超时！请检查设备是否已成功连接电脑，设备号：{}')
 ERROR_MSG_0046 = (4046, '您要测试的设备中，没有您输入的包名，请检查包名是否正确，或者设备是否安装了这个软件包')
-ERROR_MSG_0047 = (4047, '')
+ERROR_MSG_0047 = (4047, 'AI定位元素超过最大次数')
 ERROR_MSG_0048 = (4048, '选项发生错误，请在执行器点击发送缓存数据，然后重新修改步骤的操作！或联系管理员处理！')
 ERROR_MSG_0049 = (4049, 'url格式错误，请检查url或者是选择的测试环境不正确')
 ERROR_MSG_0050 = (4050, 'xpath定位未找到元素')
 ERROR_MSG_0051 = (4051, '')
 ERROR_MSG_0052 = (4052, '元素【{}】 进行断言时发生异常，请检查元素是否可以正常使用')
-ERROR_MSG_0053 = (4053, '')
+ERROR_MSG_0053 = (4053, '元素个数小于1，准备对元素的操作默认失败，请检查元素定位是否准确')
 ERROR_MSG_0054 = (4054, '步骤是操作类型，但是你没有输入操作内容，请查看文档中对元素操作的介绍')
 ERROR_MSG_0055 = (4055, '您的电脑未安装指定浏览器！')
 ERROR_MSG_0056 = (4056, '输入的参数类型不正确，请检查操作输入框的类型是否正确')

mangoautomation/mangos/__init__.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2025-10-27 17:09
+# @Author : 毛鹏
+from .mangos import *
+
+__all__ = [
+    'ElementMain','mango_send',
+    'AsyncWebAssertion', 'AsyncWebBrowser', 'AsyncWebCustomization', 'AsyncWebElement', 'AsyncWebDeviceInput', 'AsyncWebNewBrowser', 'AsyncWebPage',
+    'SyncWebAssertion', 'SyncWebBrowser', 'SyncWebCustomization', 'SyncWebElement', 'SyncWebDeviceInput', 'SyncWebNewBrowser', 'SyncWebPage',
+    'AndroidApplication','AndroidAssertion', 'AndroidCustomization','AndroidElement','AndroidEquipment', 'NewAndroid', 'AndroidPage',
+    'WebAIFinder'
+]

mangoautomation/models/__init__.py

@@ -2,11 +2,9 @@
 # @Project: 芒果测试平台
 # @Description:# @Time   : 2023-09-09 23:17
 # @Author : 毛鹏
-import sys
 
 from ..models._ui_model import ElementModel, ElementResultModel, ElementListResultModel
 
-
 __all__ = [
     'ElementModel',
     'ElementResultModel',

mangoautomation/models/_ui_model.py

@@ -22,7 +22,8 @@ class ElementListModel(BaseModel):
     exp: int
     loc: str
     sub: int | None = None
-    is_iframe: int | None
+    is_iframe: int | None = None
+    prompt: str | None = None
 
 
 class ElementModel(BaseModel):

mangoautomation/tools/__init__.py

@@ -3,10 +3,8 @@
 # @Description: 
 # @Time   : 2025-04-15 11:54
 # @Author : 毛鹏
-import sys
 
 from ..tools._mate import Meta
 from ..tools._uiautodev import start_uiautodev, stop_uiautodev
 
-
 __all__ = ['Meta', 'start_uiautodev', 'stop_uiautodev']

mangoautomation/tools/_uiautodev.py

@@ -6,8 +6,6 @@
 
 import subprocess
 
-import time
-
 uiautodev_process = None
 
 
@@ -39,7 +37,7 @@ def stop_uiautodev():
     try:
         uiautodev_process.terminate()
         uiautodev_process.wait(timeout=5)
-        return  "uiautodev 服务已关闭"
+        return "uiautodev 服务已关闭"
     except subprocess.TimeoutExpired:
         uiautodev_process.kill()
         return "uiautodev 服务被强制终止"

mangoautomation/uidrive/__init__.py

@@ -2,17 +2,15 @@
 # @Project: 芒果测试平台
 # @Description: # @Time   : 2023-07-15 11:57
 # @Author : 毛鹏
-import sys
-
-from ..uidrive._async_element import AsyncElement
-from ..uidrive._base_data import BaseData
-from ..uidrive._driver_object import DriverObject
-from ..uidrive._sync_element import SyncElement
-
+from ..uidrives import *
 
 __all__ = [
     'AsyncElement',
     'BaseData',
     'DriverObject',
     'SyncElement',
+    'AsyncWebDevice',
+    'SyncWebDevice',
+    'AsyncWebCustomization',
+    'SyncWebCustomization',
 ]

mangoautomation/uidrives/__init__.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: # @Time   : 2023-07-15 11:57
+# @Author : 毛鹏
+
+from .web import AsyncWebDevice, AsyncWebCustomization, SyncWebDevice, SyncWebCustomization
+from ..uidrives._async_element import AsyncElement
+from ..uidrives._base_data import BaseData
+from ..uidrives._driver_object import DriverObject
+from ..uidrives._sync_element import SyncElement
+
+__all__ = [
+    'BaseData',
+    'DriverObject',
+    'AsyncElement',
+    'SyncElement',
+    'AsyncWebDevice',
+    'SyncWebDevice',
+    'AsyncWebCustomization',
+    'SyncWebCustomization',
+]

mangoautomation/{uidrive → uidrives}/_async_element.py

@@ -5,12 +5,15 @@
 # @Author : 毛鹏
 import asyncio
 import os
-import random
 import traceback
 from typing import Optional
 
+import itertools
 from playwright._impl._errors import TargetClosedError, Error
 
+from mangoautomation.mangos import WebAIFinder
+from mangoautomation.uidrives.android import AndroidDriver
+from mangoautomation.uidrives.web import AsyncWebDevice, AsyncWebAssertion
 from mangotools.assertion import MangoAssertion
 from mangotools.decorator import async_retry
 from mangotools.enums import StatusEnum
@@ -18,8 +21,6 @@ from ..enums import ElementOperationEnum, DriveTypeEnum
 from ..exceptions import MangoAutomationError
 from ..exceptions.error_msg import *
 from ..models import ElementResultModel, ElementModel, ElementListResultModel
-from ..uidrive.android import AndroidDriver
-from ..uidrive.web.async_web import AsyncWebDevice, AsyncWebAssertion
 
 
 class AsyncElement(AsyncWebDevice, AndroidDriver):
@@ -31,6 +32,12 @@ class AsyncElement(AsyncWebDevice, AndroidDriver):
         self.element_result_model: Optional[ElementResultModel | None] = None
         self.element_list_model: list[ElementModel] = []
         self.test_data = self.base_data.test_data
+        self.elements = itertools.cycle([])
+        self.elements_count = 0
+        self.failed_expression = []
+        if self.base_data.is_ai:
+            self.agent = WebAIFinder(self.base_data.log, self.base_data.api_key, self.base_data.base_url,
+                                     self.base_data.model)
 
     async def open_device(self, is_open: bool = False):
         if self.drive_type == DriveTypeEnum.WEB.value:
@@ -45,8 +52,10 @@ class AsyncElement(AsyncWebDevice, AndroidDriver):
 
     async def element_main(self,
                            element_model: ElementModel,
-                           element_list_model: list[ElementModel]) -> ElementResultModel:
+                           element_list_model: list[ElementModel] | None = None) -> ElementResultModel:
         self.element_model = element_model
+        if self.element_model and self.element_model.elements:
+            self.elements = itertools.cycle(self.element_model.elements)
         self.element_list_model = element_list_model
         self.element_result_model = ElementResultModel(
             id=self.element_model.id,
@@ -195,7 +204,9 @@ class AsyncElement(AsyncWebDevice, AndroidDriver):
 
     async def __custom(self):
         for i in self.element_model.custom:
-            self.base_data.test_data.set_cache(i.get('key'), self.base_data.test_data.replace(i.get('value')))
+            value = self.base_data.test_data.replace(i.get('value'))
+            self.base_data.log.debug(f'开始执行自定义-1：key: {i.get("key")}, value: {value}')
+            self.base_data.test_data.set_cache(i.get('key'), value)
 
     async def __condition(self):
         error_list = []
@@ -222,54 +233,72 @@ class AsyncElement(AsyncWebDevice, AndroidDriver):
             ope_key = 'actual' if is_ass else 'locating'
             for i in self.element_model.ope_value:
                 if i.f == ope_key and self.element_model.elements:
-                    random_element = random.randint(0, len(self.element_model.elements) - 1)
-                    find_params = {
-                        'name': self.element_model.name,
-                        '_type': self.element_model.type,
-                        'exp': self.element_model.elements[random_element].exp,
-                        'loc': self.element_model.elements[random_element].loc,
-                        'sub': self.element_model.elements[random_element].sub
-                    }
-                    if self.drive_type == DriveTypeEnum.WEB.value:
-                        loc, ele_quantity, element_text = await self.web_find_ele(
-                            **find_params, is_iframe=self.element_model.elements[random_element].is_iframe)
-                    elif self.drive_type == DriveTypeEnum.ANDROID.value:
-                        loc, ele_quantity, element_text = self.a_find_ele(**find_params)
-                    else:
-                        loc, ele_quantity, element_text = None, 0, None
-                    new_element = ElementListResultModel(
-                        exp=self.element_model.elements[random_element].exp,
-                        loc=self.element_model.elements[random_element].loc,
-                        sub=self.element_model.elements[random_element].sub,
-                        ele_quantity=ele_quantity,
-                        element_text=element_text,
-                        is_iframe=self.element_model.elements[random_element].is_iframe
-                    )
-
-                    element_exists = any(
-                        existing.exp == new_element.exp and
-                        existing.loc == new_element.loc and
-                        existing.sub == new_element.sub and
-                        existing.is_iframe == new_element.is_iframe
-                        for existing in self.element_result_model.elements
-                    )
-
-                    if not element_exists:
-                        self.element_result_model.elements.append(new_element)
-                    if is_ass:
-                        if callable(getattr(AsyncWebAssertion, self.element_model.ope_key, None)):
-                            i.v = loc
-                        elif callable(getattr(MangoAssertion(), self.element_model.ope_key, None)):
-                            i.v = element_text
-                    else:
-                        i.v = loc
-                i.v = self.base_data.test_data.replace(i.v)
-
+                    await self.query_element(i, is_ass)
+                else:
+                    i.v = self.base_data.test_data.replace(i.v)
         except AttributeError as error:
             self.base_data.log.debug(
                 f'获取操作值失败-1，类型：{type(error)}，失败详情：{error}，失败明细：{traceback.format_exc()}')
             raise MangoAutomationError(*ERROR_MSG_0027)
 
+    async def query_element(self, i, is_ass):
+        random_element = next(self.elements)
+        self.elements_count += 1
+        find_params = {
+            'name': self.element_model.name,
+            '_type': self.element_model.type,
+            'exp': random_element.exp,
+            'loc': random_element.loc,
+            'sub': random_element.sub
+        }
+        loc_text = random_element.loc
+        try:
+            if self.drive_type == DriveTypeEnum.WEB.value:
+                loc, ele_quantity, element_text = await self.web_find_element(**find_params,
+                                                                              is_iframe=random_element.is_iframe)
+            elif self.drive_type == DriveTypeEnum.ANDROID.value:
+                loc, ele_quantity, element_text = self.a_find_ele(**find_params)
+            else:
+                loc, ele_quantity, element_text = None, 0, None
+        except MangoAutomationError as e:
+            if not self.base_data.is_ai:
+                raise e
+            if self.elements_count < len(self.element_model.elements):
+                raise e
+            self.failed_expression.append(find_params.get('loc'))
+            self.elements_count = 0
+            loc, loc_text = await self.agent.ai_find_element_async(self.base_data.page,
+                                                                   self.element_model.name,
+                                                                   random_element.prompt,
+                                                                   failed_expression=self.failed_expression)
+            loc, ele_quantity, element_text = await self.ai_element_info(loc)
+        if loc_text not in self.failed_expression:
+            self.failed_expression.append(loc_text)
+        element_exists = any(
+            existing.exp == random_element.exp and
+            existing.loc == random_element.loc and
+            existing.sub == random_element.sub and
+            existing.is_iframe == random_element.is_iframe
+            for existing in self.element_result_model.elements
+        )
+        if not element_exists:
+            new_element = ElementListResultModel(
+                exp=random_element.exp,
+                loc=random_element.loc,
+                sub=random_element.sub,
+                ele_quantity=ele_quantity,
+                element_text=element_text,
+                is_iframe=random_element.is_iframe
+            )
+            self.element_result_model.elements.append(new_element)
+        if is_ass:
+            if callable(getattr(AsyncWebAssertion, self.element_model.ope_key, None)):
+                i.v = loc
+            elif callable(getattr(MangoAssertion(), self.element_model.ope_key, None)):
+                i.v = element_text
+        else:
+            i.v = loc
+
     async def __error(self, msg: str, is_screenshot: bool = True):
         try:
             self.element_result_model.status = StatusEnum.FAIL.value
@@ -283,10 +312,9 @@ class AsyncElement(AsyncWebDevice, AndroidDriver):
                 """
             )
             if is_screenshot:
-                file_name = f'失败截图-{self.element_model.name}{self.base_data.test_data.get_time_for_min()}.jpg'
-                file_path = os.path.join(self.base_data.screenshot_path, file_name)
-                await self.__error_screenshot(file_path)
-                self.element_result_model.picture_path = file_path
+                file_name = f'失败截图-{self.element_model.name}{self.base_data.test_data.time_stamp()}.jpg'
+                await self.__error_screenshot(file_name)
+                self.element_result_model.picture_path = os.path.join(self.base_data.screenshot_path, file_name)
                 self.element_result_model.picture_name = file_name
         except MangoAutomationError as error:
             self.element_result_model.error_message += f'执行过程中发生失败，准备截图时截图失败，失败原因：{error.msg}'

mangoautomation/{uidrive → uidrives}/_base_data.py

@@ -7,8 +7,8 @@ from typing import Optional
 from unittest.mock import MagicMock
 
 import sys
-from playwright.async_api import Page as ap, BrowserContext as ac
-from playwright.sync_api import Page as p, BrowserContext as c
+from playwright.async_api import Page as AsyncPage, BrowserContext as AsyncBrowserContext
+from playwright.sync_api import Page as SyncPage, BrowserContext as SyncBrowserContext
 from uiautomator2 import Device
 
 from mangoautomation.enums import DriveTypeEnum
@@ -35,13 +35,18 @@ class BaseData:
         self.download_path: Optional[str | None] = None
         self.screenshot_path: Optional[str | None] = None
 
+        self.is_ai = False
+        self.api_key = None
+        self.base_url = None
+        self.model = None
+
         self.mysql_config: Optional[MysqlConingModel | None] = None
         self.mysql_connect: Optional[MysqlConnect | None] = None
 
         self.url: Optional[str | None] = None
         self.is_open_url = False
-        self.page: Optional[p | ap | None] = None
-        self.context: Optional[c | ac | None] = None
+        self.page: Optional[AsyncPage | SyncPage | None] = None
+        self.context: Optional[AsyncBrowserContext | SyncBrowserContext | None] = None
 
         self.package_name: Optional[str | None] = None
         self.android: Optional[Device | None] = None
@@ -116,3 +121,10 @@ class BaseData:
                 raise MangoAutomationError(*ERROR_MSG_0007)
         else:
             pass
+
+    def set_agent(self, is_ai: bool, api_key: str, base_url: str='https://api.siliconflow.cn/v1', model: str='THUDM/GLM-Z1-9B-0414'):
+        self.is_ai = is_ai
+        self.api_key = api_key
+        self.base_url = base_url
+        self.model = model
+        return self

mangoautomation/{uidrive → uidrives}/_driver_object.py

@@ -5,11 +5,9 @@
 # @Author : 毛鹏
 
 from typing import Optional
+from mangoautomation.mangos import NewAndroid, AsyncWebNewBrowser, SyncWebNewBrowser
+from ..uidrives.pc.new_windows import NewWindows
 
-from ..uidrive.android._new_android import NewAndroid
-from ..uidrive.pc.new_windows import NewWindows
-from ..uidrive.web.async_web._new_browser import AsyncWebNewBrowser
-from ..uidrive.web.sync_web._new_browser import SyncWebNewBrowser
 
 
 class DriverObject:

mangoautomation/{uidrive → uidrives}/_sync_element.py

@@ -4,22 +4,23 @@
 # @Time   : 2025-04-12 15:55
 # @Author : 毛鹏
 import os
-import random
 import traceback
 from typing import Optional
 
+import itertools
 import time
 from playwright._impl._errors import TargetClosedError, Error
 
 from mangotools.assertion import MangoAssertion
 from mangotools.decorator import sync_retry
 from mangotools.enums import StatusEnum
+from mangoautomation.mangos import WebAIFinder
 from ..enums import ElementOperationEnum, DriveTypeEnum
 from ..exceptions import MangoAutomationError
 from ..exceptions.error_msg import *
 from ..models import ElementResultModel, ElementModel, ElementListResultModel
-from ..uidrive.android import AndroidDriver
-from ..uidrive.web.sync_web import SyncWebDevice, SyncWebAssertion
+from ..uidrives.android import AndroidDriver
+from ..uidrives.web import SyncWebDevice, SyncWebAssertion
 
 
 class SyncElement(SyncWebDevice, AndroidDriver):
@@ -31,6 +32,12 @@ class SyncElement(SyncWebDevice, AndroidDriver):
         self.element_result_model: Optional[ElementResultModel | None] = None
         self.element_list_model: Optional[list[ElementModel] | None] = None
         self.test_data = self.base_data.test_data
+        self.elements = itertools.cycle([])
+        self.elements_count = 0
+        self.failed_expression = []
+        if self.base_data.is_ai:
+            self.agent = WebAIFinder(self.base_data.log, self.base_data.api_key, self.base_data.base_url,
+                                     self.base_data.model)
 
     def open_device(self, is_open: bool = False):
         if self.drive_type == DriveTypeEnum.WEB.value:
@@ -47,6 +54,8 @@ class SyncElement(SyncWebDevice, AndroidDriver):
                      element_model: ElementModel,
                      element_list_model: list[ElementModel] | None = None) -> ElementResultModel:
         self.element_model = element_model
+        if self.element_model and self.element_model.elements:
+            self.elements = itertools.cycle(self.element_model.elements)
         self.element_list_model = element_list_model
         self.element_result_model = ElementResultModel(
             id=self.element_model.id,
@@ -195,7 +204,9 @@ class SyncElement(SyncWebDevice, AndroidDriver):
 
     def __custom(self):
         for i in self.element_model.custom:
-            self.base_data.test_data.set_cache(i.get('key'), self.base_data.test_data.replace(i.get('value')))
+            value = self.base_data.test_data.replace(i.get('value'))
+            self.base_data.log.debug(f'开始执行自定义-1：key: {i.get("key")}, value: {value}')
+            self.base_data.test_data.set_cache(i.get('key'), value)
 
     def __condition(self):
         error_list = []
@@ -222,54 +233,72 @@ class SyncElement(SyncWebDevice, AndroidDriver):
             ope_key = 'actual' if is_ass else 'locating'
             for i in self.element_model.ope_value:
                 if i.f == ope_key and self.element_model.elements:
-                    random_element = random.randint(0, len(self.element_model.elements) - 1)
-                    find_params = {
-                        'name': self.element_model.name,
-                        '_type': self.element_model.type,
-                        'exp': self.element_model.elements[random_element].exp,
-                        'loc': self.element_model.elements[random_element].loc,
-                        'sub': self.element_model.elements[random_element].sub
-                    }
-                    if self.drive_type == DriveTypeEnum.WEB.value:
-                        loc, ele_quantity, element_text = self.web_find_ele(
-                            **find_params,is_iframe=self.element_model.is_iframe)
-                    elif self.drive_type == DriveTypeEnum.ANDROID.value:
-                        loc, ele_quantity, element_text = self.a_find_ele(**find_params)
-                    else:
-                        loc, ele_quantity, element_text = None, 0, None
-                    new_element = ElementListResultModel(
-                        exp=self.element_model.elements[random_element].exp,
-                        loc=self.element_model.elements[random_element].loc,
-                        sub=self.element_model.elements[random_element].sub,
-                        ele_quantity=ele_quantity,
-                        element_text=element_text,
-                        is_iframe=self.element_model.elements[random_element].is_iframe
-                    )
-
-                    element_exists = any(
-                        existing.exp == new_element.exp and
-                        existing.loc == new_element.loc and
-                        existing.sub == new_element.sub and
-                        existing.is_iframe == new_element.is_iframe
-                        for existing in self.element_result_model.elements
-                    )
-
-                    if not element_exists:
-                        self.element_result_model.elements.append(new_element)
-                    if is_ass:
-                        if callable(getattr(SyncWebAssertion, self.element_model.ope_key, None)):
-                            i.v = loc
-                        elif callable(getattr(MangoAssertion(), self.element_model.ope_key, None)):
-                            i.v = element_text
-                    else:
-                        i.v = loc
+                    self.query_element(i, is_ass)
+                else:
                     i.v = self.base_data.test_data.replace(i.v)
-
         except AttributeError as error:
             self.base_data.log.debug(
                 f'获取操作值失败-1，类型：{type(error)}，失败详情：{error}，失败明细：{traceback.format_exc()}')
             raise MangoAutomationError(*ERROR_MSG_0027)
 
+    def query_element(self, i, is_ass):
+        random_element = next(self.elements)
+        self.elements_count += 1
+        find_params = {
+            'name': self.element_model.name,
+            '_type': self.element_model.type,
+            'exp': random_element.exp,
+            'loc': random_element.loc,
+            'sub': random_element.sub
+        }
+        loc_text = random_element.loc
+        try:
+            if self.drive_type == DriveTypeEnum.WEB.value:
+                loc, ele_quantity, element_text = self.web_find_element(**find_params,
+                                                                        is_iframe=random_element.is_iframe)
+            elif self.drive_type == DriveTypeEnum.ANDROID.value:
+                loc, ele_quantity, element_text = self.a_find_ele(**find_params)
+            else:
+                loc, ele_quantity, element_text = None, 0, None
+        except MangoAutomationError as e:
+            if not self.base_data.is_ai:
+                raise e
+            if self.elements_count < len(self.element_model.elements):
+                raise e
+            self.failed_expression.append(find_params.get('loc'))
+            self.elements_count = 0
+            loc, loc_text = self.agent.ai_find_element_sync(self.base_data.page,
+                                                            self.element_model.name,
+                                                             random_element.prompt,
+                                                             failed_expression=self.failed_expression)
+            loc, ele_quantity, element_text = self.ai_element_info(loc)
+        if loc_text not in self.failed_expression:
+            self.failed_expression.append(loc_text)
+        element_exists = any(
+            existing.exp == random_element.exp and
+            existing.loc == random_element.loc and
+            existing.sub == random_element.sub and
+            existing.is_iframe == random_element.is_iframe
+            for existing in self.element_result_model.elements
+        )
+        if not element_exists:
+            new_element = ElementListResultModel(
+                exp=random_element.exp,
+                loc=random_element.loc,
+                sub=random_element.sub,
+                ele_quantity=ele_quantity,
+                element_text=element_text,
+                is_iframe=random_element.is_iframe
+            )
+            self.element_result_model.elements.append(new_element)
+        if is_ass:
+            if callable(getattr(SyncWebAssertion, self.element_model.ope_key, None)):
+                i.v = loc
+            elif callable(getattr(MangoAssertion(), self.element_model.ope_key, None)):
+                i.v = element_text
+        else:
+            i.v = loc
+
     def __error(self, msg: str, is_screenshot: bool = True):
         try:
             self.element_result_model.status = StatusEnum.FAIL.value
@@ -283,10 +312,9 @@ class SyncElement(SyncWebDevice, AndroidDriver):
                 """
             )
             if is_screenshot:
-                file_name = f'失败截图-{self.element_model.name}{self.base_data.test_data.get_time_for_min()}.jpg'
-                file_path = os.path.join(self.base_data.screenshot_path, file_name)
-                self.__error_screenshot(file_path)
-                self.element_result_model.picture_path = file_path
+                file_name = f'失败截图-{self.element_model.name}{self.base_data.test_data.time_stamp()}.jpg'
+                self.__error_screenshot(file_name)
+                self.element_result_model.picture_path = os.path.join(self.base_data.screenshot_path, file_name)
                 self.element_result_model.picture_name = file_name
         except MangoAutomationError as error:
             self.element_result_model.error_message += f'执行过程中发生失败，准备截图时截图失败，失败原因：{error.msg}'