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

mangoautomation/uidrives/web/_async_web.py

@@ -0,0 +1,166 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2025-04-04 21:42
+# @Author : 毛鹏
+import re
+import traceback
+
+from playwright.async_api import Locator, Error, TimeoutError
+
+from mangoautomation.mangos import AsyncWebAssertion, AsyncWebBrowser, AsyncWebCustomization, AsyncWebElement, \
+    AsyncWebDeviceInput, AsyncWebPage, ElementMain
+from mangotools.assertion import MangoAssertion
+from mangotools.enums import StatusEnum
+from mangotools.mangos import Mango
+from ...enums import ElementExpEnum
+from ...exceptions import MangoAutomationError
+from ...exceptions.error_msg import *
+
+re = re
+
+
+class AsyncWebDevice(AsyncWebBrowser,
+                     AsyncWebPage,
+                     AsyncWebElement,
+                     AsyncWebDeviceInput,
+                     AsyncWebCustomization):
+
+    def __init__(self, base_data):
+        super().__init__(base_data)
+
+    async def open_url(self, is_open: bool = False):
+        if not self.base_data.is_open_url or is_open:
+            self.base_data.log.debug(f'打开url，is_open_url：{self.base_data.is_open_url},url:{self.base_data.url}')
+            await self.w_goto(self.base_data.url)
+            self.base_data.is_open_url = True
+
+    async def web_action_element(self, name, ope_key, ope_value, ):
+        self.base_data.log.debug(f'操作元素{name}：,key:{ope_key},value:{ope_value}')
+        try:
+            await ElementMain.a_element(self, ope_key, ope_value)
+        except TimeoutError as error:
+            self.base_data.log.debug(f'WEB自动化操作失败-1，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0011, value=(name,))
+        except Error as error:
+            self.base_data.log.error(f'WEB自动化操作失败-2，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0032, value=(name,))
+        except ValueError as error:
+            self.base_data.log.error(f'WEB自动化操作失败-3，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0012)
+
+    async def web_assertion_element(self, name, ope_key, ope_value) -> str:
+        self.base_data.log.debug(f'断言元素，名称：{name},key:{ope_key},value:{ope_value}')
+        is_method = callable(getattr(AsyncWebAssertion, ope_key, None))
+        try:
+            if is_method:
+                if ope_value.get('actual', None) is None:
+                    raise MangoAutomationError(*ERROR_MSG_0031, value=(name,))
+                self.base_data.log.debug(f'开始断言-1，方法：{ope_key}，断言值：{ope_value}')
+                return await Mango.a_e(AsyncWebAssertion(self.base_data), ope_key, ope_value)
+            else:
+                self.base_data.log.debug(f'开始断言-2，方法：{ope_key}，断言值：{ope_value}')
+                return MangoAssertion(self.base_data.mysql_connect, self.base_data.test_data) \
+                    .ass(ope_key, ope_value.get('actual'), ope_value.get('expect'))
+        except AssertionError as error:
+            self.base_data.log.debug(f'WEB自动化断言失败-1，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0017, value=error.args)
+        except AttributeError as error:
+            self.base_data.log.error(f'WEB自动化断言失败-2，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0048)
+        except ValueError as error:
+            self.base_data.log.error(f'WEB自动化断言失败-3，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0005)
+        except Error as error:
+            self.base_data.log.error(f'WEB自动化断言失败-4，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0052, value=(name,), )
+
+    async def web_find_element(self, name, _type, exp, loc, sub, is_iframe) \
+            -> tuple[Locator, int, str] | tuple[list[Locator], int, str]:
+        self.base_data.log.debug(
+            f'查找元素-1，名称：{name},_type:{_type},exp:{exp},loc:{loc},sub:{sub},is_iframe:{is_iframe}')
+        if is_iframe != StatusEnum.SUCCESS.value:
+            locator: Locator = await self.__find_ele(self.base_data.page, exp, loc)
+            try:
+                return await self.__element_info(locator, sub)
+            except Error as error:
+                self.base_data.log.debug(
+                    f'WEB自动化查找元素失败-1，类型：{type(error)}，失败详情：{error}，失败明细：{traceback.format_exc()}')
+                raise MangoAutomationError(*ERROR_MSG_0041, value=(name, loc))
+        else:
+            return await self.__is_iframe(_type, exp, loc, sub)
+
+    async def __element_info(self, locator: Locator, sub) -> tuple[Locator, int, str]:
+        count = await locator.count()
+        if count < 1:
+            raise MangoAutomationError(*ERROR_MSG_0053)
+        if sub is not None:
+            loc_result = locator.nth(sub - 1) if sub else locator
+        else:
+            loc_result = locator.nth(0)
+
+        try:
+            text = await self.w_get_text(loc_result)
+        except Exception:
+            text = None
+        return loc_result, count, text
+
+    async def ai_element_info(self, locator: Locator) -> tuple[Locator, int, str]:
+        count = await locator.count()
+        if count < 1:
+            raise MangoAutomationError(*ERROR_MSG_0053)
+        try:
+            text = await self.w_get_text(locator.nth(0))
+        except Exception:
+            text = None
+        return locator.nth(0), count, text
+
+    async def __is_iframe(self, _type, exp, loc, sub):
+        ele_list: list[Locator] = []
+        for i in self.base_data.page.frames:
+            locator: Locator = await self.__find_ele(i, exp, loc)
+            try:
+                count = await locator.count()
+            except Error as error:
+                self.base_data.log.debug(
+                    f'WEB自动化查找元素失败-2，类型：{type(error)}，失败详情：{error}，失败明细：{traceback.format_exc()}')
+                raise MangoAutomationError(*ERROR_MSG_0041, )
+            if count > 0:
+                for nth in range(0, count):
+                    ele_list.append(locator.nth(nth))
+            else:
+                raise MangoAutomationError(*ERROR_MSG_0023)
+        try:
+            count = len(ele_list)
+            loc = ele_list[sub - 1] if sub else ele_list[0]
+            try:
+                text = await self.w_get_text(loc)
+            except Exception:
+                text = None
+            return loc, count, text
+        except IndexError:
+            raise MangoAutomationError(*ERROR_MSG_0025, value=(len(ele_list),))
+
+    async def __find_ele(self, page, exp, loc) -> Locator:
+        if exp == ElementExpEnum.LOCATOR.value:
+            try:
+                return eval(f"await page.{loc}")
+            except SyntaxError:
+                try:
+                    return eval(f"page.{loc}")
+                except SyntaxError as error:
+                    self.base_data.log.error(f'WEB自动化查找元素失败-3，类型：{type(error)}，失败详情：{error}')
+                    raise MangoAutomationError(*ERROR_MSG_0022)
+                except NameError as error:
+                    self.base_data.log.error(f'WEB自动化查找元素失败-4，类型：{type(error)}，失败详情：{error}')
+                    raise MangoAutomationError(*ERROR_MSG_0060)
+        elif exp == ElementExpEnum.XPATH.value:
+            return page.locator(f'xpath={loc}')
+        elif exp == ElementExpEnum.CSS.value:
+            return page.locator(loc)
+        elif exp == ElementExpEnum.TEXT.value:
+            return page.get_by_text(loc, exact=True)
+        elif exp == ElementExpEnum.PLACEHOLDER.value:
+            return page.get_by_placeholder(loc)
+        else:
+            raise MangoAutomationError(*ERROR_MSG_0020)

mangoautomation/uidrives/web/_sync_web.py

@@ -0,0 +1,166 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2025-04-04 21:42
+# @Author : 毛鹏
+import re
+import traceback
+
+from playwright.sync_api import Locator, TimeoutError, Error
+
+from mangoautomation.mangos import SyncWebAssertion, SyncWebBrowser, SyncWebCustomization, SyncWebElement, \
+    SyncWebDeviceInput, SyncWebPage, ElementMain
+from mangotools.assertion import MangoAssertion
+from mangotools.enums import StatusEnum
+from mangotools.mangos import Mango
+from ...enums import ElementExpEnum
+from ...exceptions import MangoAutomationError
+from ...exceptions.error_msg import *
+
+re = re
+
+
+class SyncWebDevice(SyncWebBrowser,
+                    SyncWebPage,
+                    SyncWebElement,
+                    SyncWebDeviceInput,
+                    SyncWebCustomization):
+
+    def __init__(self, base_data):
+        super().__init__(base_data)
+
+    def open_url(self, is_open: bool = False):
+        if not self.base_data.is_open_url or is_open:
+            self.base_data.log.debug(f'打开url，is_open_url：{self.base_data.is_open_url},url:{self.base_data.url}')
+            self.w_goto(self.base_data.url)
+            self.base_data.is_open_url = True
+
+    def web_action_element(self, name, ope_key, ope_value, ):
+        self.base_data.log.debug(f'操作元素，名称：{name},key:{ope_key},value:{ope_value}')
+        try:
+            ElementMain.s_element(self, ope_key, ope_value)
+        except TimeoutError as error:
+            self.base_data.log.debug(f'WEB自动化操作失败-1，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0011, value=(name,))
+        except Error as error:
+            self.base_data.log.error(f'WEB自动化操作失败-2，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0032, value=(name,))
+        except ValueError as error:
+            self.base_data.log.error(f'WEB自动化操作失败-3，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0012)
+
+    def web_assertion_element(self, name, ope_key, ope_value) -> str:
+        self.base_data.log.debug(f'断言元素，名称：{name},key:{ope_key},value:{ope_value}')
+        is_method = callable(getattr(SyncWebAssertion, ope_key, None))
+        try:
+            if is_method:
+                if ope_value.get('actual', None) is None:
+                    raise MangoAutomationError(*ERROR_MSG_0031, value=(name,))
+                self.base_data.log.debug(f'开始断言-1，方法：{ope_key}，断言值：{ope_value}')
+                return Mango.s_e(SyncWebAssertion(self.base_data), ope_key, ope_value)
+            else:
+                self.base_data.log.debug(f'开始断言-2，方法：{ope_key}，断言值：{ope_value}')
+                return MangoAssertion(self.base_data.mysql_connect, self.base_data.test_data) \
+                    .ass(ope_key, ope_value.get('actual'), ope_value.get('expect'))
+        except AssertionError as error:
+            self.base_data.log.debug(f'WEB自动化断言失败-1，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0017, value=error.args)
+        except AttributeError as error:
+            self.base_data.log.error(f'WEB自动化断言失败-2，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0048)
+        except ValueError as error:
+            self.base_data.log.error(f'WEB自动化断言失败-3，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0005)
+        except Error as error:
+            self.base_data.log.error(f'WEB自动化断言失败-4，类型：{type(error)}，失败详情：{error}')
+            raise MangoAutomationError(*ERROR_MSG_0052, value=(name,), )
+
+    def web_find_element(self, name, _type, exp, loc, sub, is_iframe) \
+            -> tuple[Locator, int, str] | tuple[list[Locator], int, str]:
+        self.base_data.log.debug(
+            f'查找元素，名称：{name},_type:{_type},exp:{exp},loc:{loc},sub:{sub},is_iframe:{is_iframe}')
+        if is_iframe != StatusEnum.SUCCESS.value:
+            locator: Locator = self.__find_ele(self.base_data.page, exp, loc)
+            try:
+                return self.__element_info(locator, sub)
+            except Error as error:
+                self.base_data.log.debug(
+                    f'WEB自动化查找元素失败-1，类型：{type(error)}，失败详情：{error}，失败明细：{traceback.format_exc()}')
+                raise MangoAutomationError(*ERROR_MSG_0041, )
+        else:
+            return self.__is_iframe(_type, exp, loc, sub)
+
+    def __element_info(self, locator: Locator, sub) -> tuple[Locator, int, str]:
+        count = locator.count()
+        if count < 1:
+            raise MangoAutomationError(*ERROR_MSG_0053)
+        if sub is not None:
+            loc_result = locator.nth(sub - 1) if sub else locator
+        else:
+            loc_result = locator.nth(0)
+
+        try:
+            text = self.w_get_text(loc_result)
+        except Exception:
+            text = None
+        return loc_result, count, text
+
+    def ai_element_info(self, locator: Locator) -> tuple[Locator, int, str]:
+        count = locator.count()
+        if count < 1:
+            raise MangoAutomationError(*ERROR_MSG_0053)
+        try:
+            text = self.w_get_text(locator.nth(0))
+        except Exception:
+            text = None
+        return locator.nth(0), count, text
+
+    def __is_iframe(self, _type, exp, loc, sub):
+        ele_list: list[Locator] = []
+        for i in self.base_data.page.frames:
+            locator: Locator = self.__find_ele(i, exp, loc)
+            try:
+                count = locator.count()
+            except Error as error:
+                self.base_data.log.debug(
+                    f'WEB自动化查找元素失败-2，类型：{type(error)}，失败详情：{error}，失败明细：{traceback.format_exc()}')
+                raise MangoAutomationError(*ERROR_MSG_0041, )
+            if count > 0:
+                for nth in range(0, count):
+                    ele_list.append(locator.nth(nth))
+            else:
+                raise MangoAutomationError(*ERROR_MSG_0023)
+        try:
+            count = len(ele_list)
+            loc = ele_list[sub - 1] if sub else ele_list[0]
+            try:
+                text = self.w_get_text(loc)
+            except Exception:
+                text = None
+            return loc, count, text
+        except IndexError:
+            raise MangoAutomationError(*ERROR_MSG_0025, value=(len(ele_list),))
+
+    def __find_ele(self, page, exp, loc) -> Locator:
+        if exp == ElementExpEnum.LOCATOR.value:
+            try:
+                return eval(f"page.{loc}")
+            except SyntaxError:
+                try:
+                    return eval(f"await page.{loc}")
+                except SyntaxError as error:
+                    self.base_data.log.error(f'WEB自动化查找元素失败-3，类型：{type(error)}，失败详情：{error}')
+                    raise MangoAutomationError(*ERROR_MSG_0022)
+                except NameError as error:
+                    self.base_data.log.error(f'WEB自动化查找元素失败-4，类型：{type(error)}，失败详情：{error}')
+                    raise MangoAutomationError(*ERROR_MSG_0060)
+        elif exp == ElementExpEnum.XPATH.value:
+            return page.locator(f'xpath={loc}')
+        elif exp == ElementExpEnum.CSS.value:
+            return page.locator(loc)
+        elif exp == ElementExpEnum.TEXT.value:
+            return page.get_by_text(loc, exact=True)
+        elif exp == ElementExpEnum.PLACEHOLDER.value:
+            return page.get_by_placeholder(loc)
+        else:
+            raise MangoAutomationError(*ERROR_MSG_0020)

mangoautomation/uidrives/web/async_web/_element.py

@@ -25,15 +25,11 @@ class AsyncWebElement(metaclass=Meta):
     @async_method_callback('web', '元素操作', 0, [MethodModel(f='locating')])
     async def w_click(self, locating: Locator):
         """元素单击"""
-        if await locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         await locating.click()
 
     @async_method_callback('web', '元素操作', 1, [MethodModel(f='locating')])
     async def w_dblclick(self, locating: Locator):
         """元素双击"""
-        if await locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         await locating.dblclick()
 
     @async_method_callback('web', '元素操作', 2, [MethodModel(f='locating')])
@@ -46,8 +42,6 @@ class AsyncWebElement(metaclass=Meta):
         MethodModel(n='输入内容', f='input_value', p='请输入输入内容', d=True)])
     async def w_input(self, locating: Locator, input_value: str):
         """元素输入"""
-        if await locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         await locating.fill(str(input_value))
 
     @async_method_callback('web', '元素操作', 4, [MethodModel(f='locating')])
@@ -61,8 +55,6 @@ class AsyncWebElement(metaclass=Meta):
         MethodModel(n='缓存的key', f='set_cache_key', p='请输入获取元素文本后存储的key', d=True)])
     async def w_get_text(self, locating: Locator, set_cache_key=None):
         """获取元素文本"""
-        if await locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         methods = [
             ("inner_text", locating.inner_text),
             ("text_content", locating.text_content),
@@ -87,8 +79,6 @@ class AsyncWebElement(metaclass=Meta):
         MethodModel(n='输入文本', f='input_value', p='请输入输入内容', d=True)])
     async def w_clear_input(self, locating: Locator, input_value: str):
         """元素清空再输入"""
-        if await locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         await locating.clear()
         await locating.fill(str(input_value))
 

mangoautomation/uidrives/web/sync_web/_element.py

@@ -24,15 +24,11 @@ class SyncWebElement(metaclass=Meta):
     @sync_method_callback('web', '元素操作', 0, [MethodModel(f='locating')])
     def w_click(self, locating: Locator):
         """元素单击"""
-        if locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         locating.click()
 
     @sync_method_callback('web', '元素操作', 1, [MethodModel(f='locating')])
     def w_dblclick(self, locating: Locator):
         """元素双击"""
-        if locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         locating.dblclick()
 
     @sync_method_callback('web', '元素操作', 2, [MethodModel(f='locating')])
@@ -45,8 +41,6 @@ class SyncWebElement(metaclass=Meta):
         MethodModel(n='输入文本', f='input_value', p='请输入输入内容', d=True)])
     def w_input(self, locating: Locator, input_value: str):
         """元素输入"""
-        if locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         locating.fill(str(input_value))
 
     @sync_method_callback('web', '元素操作', 4, [MethodModel(f='locating')])
@@ -60,8 +54,6 @@ class SyncWebElement(metaclass=Meta):
         MethodModel(n='缓存的key', f='set_cache_key', p='请输入获取元素文本后存储的key', d=True)])
     def w_get_text(self, locating: Locator, set_cache_key=None):
         """获取元素文本"""
-        if locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         methods = [
             ("inner_text", lambda: locating.inner_text()),
             ("text_content", lambda: locating.text_content()),
@@ -86,8 +78,6 @@ class SyncWebElement(metaclass=Meta):
         MethodModel(n='输入文本', f='input_value', p='请输入输入内容', d=True)])
     def w_clear_input(self, locating: Locator, input_value: str):
         """元素清空再输入"""
-        if locating.count() < 1:
-            raise TimeoutError('元素个数小于1，直接操作超时！')
         locating.clear()
         locating.fill(str(input_value))
 

{mangoautomation-1.1.22.dist-info → mangoautomation-1.1.23.dist-info}/METADATA

@@ -1,14 +1,12 @@
 Metadata-Version: 2.1
 Name: mangoautomation
-Version: 1.1.22
+Version: 1.1.23
 Summary: 测试工具
 Home-page: https://gitee.com/mao-peng/testkit
 Author: 毛鹏
 Author-email: 729164035@qq.com
 Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
 License-File: LICENSE
 Requires-Dist: setuptools ==78.1.1
 Requires-Dist: pydantic >=2.9.2
@@ -18,6 +16,8 @@ Requires-Dist: uiautomator2 >=3.2.5
 Requires-Dist: mangotools >=1.1.42
 Requires-Dist: adbutils ~=2.8.9
 Requires-Dist: uiautodev >=0.9.0
+Requires-Dist: beautifulsoup4 ==4.14.2
+Requires-Dist: openai ==2.6.1
 
 # testkit
 

mangoautomation-1.1.23.dist-info/RECORD

@@ -0,0 +1,71 @@
+mangoautomation/__init__.py,sha256=JxPNVexFQvkBfh0awPNeHxQGU_gtVgYPwD5rYeuHWEk,125
+mangoautomation/enums/__init__.py,sha256=PnOgNttAXT64elr_Fd1ZthuAddCkm7sh4xwm9Eutm_I,373
+mangoautomation/enums/_base_enum.py,sha256=dpaXsdhWY08BhIzi1B1ksh_p18ZTmhWvTYhbd3n_ONQ,1176
+mangoautomation/enums/_ui_enum.py,sha256=mT_MfQ7TEw-BgHIqH0wXTA_yVWsaWnd3iHSHi9E0pgA,4599
+mangoautomation/exceptions/__init__.py,sha256=2j3ravPuzljR2Mbq8J2JLdXUzII7hzrjdVjmEc_NHJU,211
+mangoautomation/exceptions/_exceptions.py,sha256=BiyST1QPivpfIOuKn1gKhpXOyBv6gfip_KvlQuu8ctw,385
+mangoautomation/exceptions/error_msg.py,sha256=0unQ_tgrRJtVMAOXTzhQJkwO4_7lhMKAMMCfwmzQD58,5517
+mangoautomation/mangos/__init__.py,sha256=gEu8lVHGXm8PDTBZ1XXWNUOAbCtHb37uKQLBRYQR4rs,644
+mangoautomation/mangos/mangos.cp310-win_amd64.pyd,sha256=XkKi2OyGgDxX7lJQnNuJMGiL4b-vntDLwo4wAmOUdrg,1751552
+mangoautomation/mangos/mangos.cp312-win_amd64.pyd,sha256=5ajaOne3Kb1yNEb3Eci1bvjzKIuiJaTPfwBY3f_MK0o,1879040
+mangoautomation/mangos/mangos.cpython-310-darwin.so,sha256=HwZCN0lpGKdAzvygPA4O4Q_xyAd7aDcrZFUzWSqZcWI,3555568
+mangoautomation/mangos/mangos.cpython-310-x86_64-linux-gnu.so,sha256=EaTWB-R4yHJYfkwF0P_YMjlBKY35KTTKrDsAaKCqAZo,10465816
+mangoautomation/mangos/mangos.cpython-312-darwin.so,sha256=iG43WcChIPozrKK98MqBTqDlZFNxKjQSyB0CKu0mIaE,3456880
+mangoautomation/mangos/mangos.cpython-312-x86_64-linux-gnu.so,sha256=P1Hax5huyMm-7eeuWby9mWmTI2HW3JEA8mSvIl65aAk,10670984
+mangoautomation/mangos/__pycache__/__init__.cpython-310.pyc,sha256=kQcwQCOca8l62vfKIC6YKi0lEIUV3Vjnel0ox2bDbOc,608
+mangoautomation/models/__init__.py,sha256=wHACwBCrd9HCvmrAkTFUV3qmK9GnOa-FJFiL0kOQjME,309
+mangoautomation/models/_ui_model.py,sha256=nfxh_4TpOLzKKWC7c-IgMLVZY5wkTPfyS_Oh-SvsYsk,1825
+mangoautomation/tools/__init__.py,sha256=c3IXhoodknOukOyM6db2tXIdDRodgmUZorEAkkruQf0,282
+mangoautomation/tools/_mate.py,sha256=9lk_EJnAX_OyXbe5WacyHkkcEr2ScBgasl6eUnlklWA,405
+mangoautomation/tools/_uiautodev.py,sha256=KYYWOQ_sp91LN_zeKAuIvCqjabb2_1-VhgOg1K7cQrw,1372
+mangoautomation/uidrive/__init__.py,sha256=9N2DTDyLcmgxAHctBjZOb8LwdDeKQnwEOWZ9kNcM4Uw,350
+mangoautomation/uidrives/__init__.py,sha256=JEoWqpoDIch8JiQ7WcnuDh3Bjg0Yqo_xfkbqY5iHSa8,617
+mangoautomation/uidrives/_async_element.py,sha256=87BhadK9BJtcv_VzJUM3kX5sdp8b8Kk-abkU9oE3T7M,16206
+mangoautomation/uidrives/_base_data.py,sha256=gp9730Hd8uarRv8xARIddgMWGhaJ_RKDd_gjewwWr04,4436
+mangoautomation/uidrives/_driver_object.py,sha256=VnnmXO-BODhqEDmqwb2U0G5FwLRdrBKvkrLE2mwC6gs,1951
+mangoautomation/uidrives/_sync_element.py,sha256=TU2E7AKz0Ku4qXLEtxx-whJPJ4KLYLK_EuKgChpcedM,15906
+mangoautomation/uidrives/android/__init__.py,sha256=i6WjXlNq6oUZ4y79Gih_XzI9tV7du-Jrm9IAuROgSzo,5180
+mangoautomation/uidrives/android/_application.py,sha256=G94m53cUzzGuXxPIsvoMuFwJ5RWL_xW5HHFE0TIkrBI,2855
+mangoautomation/uidrives/android/_assertion.py,sha256=FH8sxJC4ZXCbVA9KFBp49Tu8LPWyO26EZYAK7cYaoYE,4350
+mangoautomation/uidrives/android/_customization.py,sha256=7C8j2rv2wDkNu425ooxLSDrGHr_CxJ82sXz1-64id_8,382
+mangoautomation/uidrives/android/_element.py,sha256=ggWEFUbrAGWPQXNvD9JpudwowwvYJrgHdrNXyytAEVw,7213
+mangoautomation/uidrives/android/_equipment.py,sha256=Ul61jpoVr162FIu_2QcJtqWil82ZBC-Kn7_VFQWCUsQ,5216
+mangoautomation/uidrives/android/_new_android.py,sha256=WldFsGm5QnJ9QVPtWxRgwCiWSX3jYObFxAPCQ5xSKyg,1636
+mangoautomation/uidrives/android/_page.py,sha256=IaWiN3kniv2GKfudMLRdvyaqbFLuIfGh7H0HeQrDyVA,4621
+mangoautomation/uidrives/ios/__init__.py,sha256=SdxI8e5AOoAb8JTaDoMkObezIluJ50nm2b6zeDHv1PI,125
+mangoautomation/uidrives/pc/__init__.py,sha256=Jphhspu-3AHCyDX5ZM_-DWeC9ZerWy2Omb3MOck6CI4,2210
+mangoautomation/uidrives/pc/assertion.py,sha256=mH25hZ2i4T8kA1F2loW_nuPJ-Hb0z1pwILTJqkwwpLA,125
+mangoautomation/uidrives/pc/customization.py,sha256=UkpHWFSYsu_t-GmMRDmii1Bs_Y1eKD34oZfJBFHV2Ng,240
+mangoautomation/uidrives/pc/element.py,sha256=OGbJHbgJPBjRjXqM-NIWUnScn7lIHoBxnrakog_s4l8,465
+mangoautomation/uidrives/pc/input_device.py,sha256=k-v7tRW8NkJxzcezqSIfICGRBqEdsssBeRTmtupVTU8,299
+mangoautomation/uidrives/pc/new_windows.py,sha256=lETHZa7cW8Z0iUShzgVTRxqivkYaOo8WIE58q_8KYfQ,2338
+mangoautomation/uidrives/web/__init__.py,sha256=TJyYt6EUoRfWYrV193RHpopKfixipVUp4TrhB5DoEKA,434
+mangoautomation/uidrives/web/_async_web.py,sha256=Uwta7cIYo-0Qe5RbaIvRSQ-IvOxCNtbH_RcIm1JNxEQ,8204
+mangoautomation/uidrives/web/_sync_web.py,sha256=NCF97mYWQBLsOkcEXHKbAGLYGt4rK7NSjks6PubLe_k,8049
+mangoautomation/uidrives/web/async_web/__init__.py,sha256=_jG4DvEbNe4EZTFEBKQhgwRxxIzrdK7APmIicFb0zMk,8379
+mangoautomation/uidrives/web/async_web/_assertion.py,sha256=cG8g_iFMlWcAMDyIfPTJ5lx73BpbGZV5PoAbe1uAy4Q,12821
+mangoautomation/uidrives/web/async_web/_browser.py,sha256=xAaawPbY-rj-HZtcy5LcWUQ_jmdN9Dj4yNUxWxXGCbQ,5421
+mangoautomation/uidrives/web/async_web/_customization.py,sha256=LakibDXpSc-MMxMiwDiktoFccgyiUmWxAVpMOobY9_A,347
+mangoautomation/uidrives/web/async_web/_element.py,sha256=fVJWBUI0C2eFYP8XhLkW2Cez2gdYP1j0PcIBVDzUEk4,10507
+mangoautomation/uidrives/web/async_web/_input_device.py,sha256=wd36vTwsWdq7f52HmEnAF2sNiFVwNxJ2J5bq-lcuuVk,3441
+mangoautomation/uidrives/web/async_web/_new_browser.py,sha256=v-OXD-uYD7XagupxSlG4g_ZBsRq_Jpo88iwE_A_Ilfs,6202
+mangoautomation/uidrives/web/async_web/_page.py,sha256=B1RbnPWaQWa3Sp57ue2sTgcgEr4kTgSEz0DcpA5lOug,2176
+mangoautomation/uidrives/web/sync_web/__init__.py,sha256=gThDpZPtCFdKrxnVYgAsj_6oha9lMed7xoU4XTsLOys,7945
+mangoautomation/uidrives/web/sync_web/_assertion.py,sha256=GEUGvYd3LgisMTgiinFF-Z6nWqxA4EUbvHenw7rPaW0,12595
+mangoautomation/uidrives/web/sync_web/_browser.py,sha256=_Z6NhhP_6N-cxak3X8yUnPzUet7b1tBpI85iAbzwoyQ,5286
+mangoautomation/uidrives/web/sync_web/_customization.py,sha256=Wir5CIGN6YiyVj1-D_zvSWcFuUvQMP8rJ-8boUaU1Bs,356
+mangoautomation/uidrives/web/sync_web/_element.py,sha256=T8Za5wTzSlPleJMz0XkRCbnqSk8OpLHMs3_TalF7xrE,10157
+mangoautomation/uidrives/web/sync_web/_input_device.py,sha256=SwsFi4PiS3sg45MPl_NV88nDrg4SmJ14fV8sODByQ58,3288
+mangoautomation/uidrives/web/sync_web/_new_browser.py,sha256=APGUFhOVAW68x-6y9tQLZJkn2utfR-PrpjgP8WE38SE,6149
+mangoautomation/uidrives/web/sync_web/_page.py,sha256=GbVsK7ZBcoMBnKw2VMPWy3Yt9Jqa84qxrB2Ar2d8aR8,2046
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/demo1.py,sha256=5iK5kqaMkebTk-Z4HoMnfi88CRUIBKwT8PVQXFaqgD4,736
+tests/get_ope.py,sha256=SVhSowMNa87hjzGhFlS0vp-t0RB9yKHOC52FsO93UG4,236
+tests/test_ai_element.py,sha256=k__iifxebvPPfGdRS-yzkEH80fBIlKJcZ-m3N6Xkz6Q,1437
+tests/test_ui_and.py,sha256=A_0pqG6s5Ts02He7P3NwuDSK3g9Q3VP1Sb_P8wq6g0Q,125
+tests/test_ui_web.py,sha256=RBACNYc7XdvENHzg4YElsSQlEmPVa_AyOKWTYTPXM14,2578
+mangoautomation-1.1.23.dist-info/LICENSE,sha256=03WP-mgFmo8ofRYDe1HVDQUEUKQZ2q6P6Q-2A4c-46A,1085
+mangoautomation-1.1.23.dist-info/METADATA,sha256=wbPR7Rb4WurD4bh8nDEoKhLSumkIIMM2008pk-Skvso,783
+mangoautomation-1.1.23.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+mangoautomation-1.1.23.dist-info/top_level.txt,sha256=g-uCmjvEODG8WFbmwbGM0-G0zHntHv8ZsS0PQRaMGtE,22
+mangoautomation-1.1.23.dist-info/RECORD,,

tests/__init__.py

@@ -1,10 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Project: 芒果测试平台
-# @Description: 
-# @Time   : 2024-09-07 23:25
-# @Author : 毛鹏
-import random
-
-data_list = [{'exp': 2, 'loc': 'get_by_role("textbox", name="请输入1密码")'},
-             {'exp': 0, 'loc': '//input[@type="password"]'}]
-print(random.randint(0, len(data_list) - 1), len(data_list))

tests/demo1.py

@@ -0,0 +1,21 @@
+from openai import OpenAI
+
+client = OpenAI(api_key="sk-rruuhjnqawsvduyxlcqckbtgwkprctgkvwcelenooixbhthy",
+                base_url="https://api.siliconflow.cn/v1")
+response = client.chat.completions.create(
+    # model='Pro/deepseek-ai/DeepSeek-R1',
+    model="THUDM/GLM-Z1-9B-0414",
+    messages=[
+        {'role': 'user',
+        'content': "推理模型会给市场带来哪些新的机会"}
+    ],
+    stream=True
+)
+
+for chunk in response:
+    if not chunk.choices:
+        continue
+    if chunk.choices[0].delta.content:
+        print(chunk.choices[0].delta.content, end="", flush=True)
+    if chunk.choices[0].delta.reasoning_content:
+        print(chunk.choices[0].delta.reasoning_content, end="", flush=True)

tests/test_ai_element.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2025-10-28 10:32
+# @Author : 毛鹏
+import unittest
+
+import time
+
+from .demo1 import WebAIFinder
+from mangoautomation.uidrive import BaseData, DriverObject, SyncElement
+from mangoautomation.uidrives import AsyncElement
+from mangotools.data_processor import DataProcessor
+from mangotools.log_collector import set_log
+# api_key = 'sk-'
+# b = 'https://api.siliconflow.cn/v1'
+# model = 'THUDM/GLM-Z1-9B-0414'
+log = set_log('D:\code\mango_automation\logs')
+log.set_debug(True)
+test_data = DataProcessor()
+
+
+class TestUi(unittest.IsolatedAsyncioTestCase):
+    async def test_a(self):
+        driver_object = DriverObject(log, is_async=True)
+        driver_object.set_web(0, r"C:\Program Files\Google\Chrome\Application\chrome.exe")
+        base_data = BaseData(test_data, log)
+        base_data.url = 'https://www.baidu.com/'
+        base_data.context, base_data.page = await driver_object.web.new_web_page()
+        element = AsyncElement(base_data, 0)
+        await element.open_url()
+        loc = await WebAIFinder(log, api_key, b, model).ai_find_element_async(base_data.page, '设置', '设置在右上角', ['//*[text()="设置"]'])
+        await element.w_hover(loc)
+        print('获取元素名称：', await element.w_get_text(loc))
+        time.sleep(5)
+        assert await element.w_get_text(loc) == '设置'

tests/test_ui_web.py

@@ -12,6 +12,7 @@ from mangotools.data_processor import DataProcessor
 from mangotools.log_collector import set_log
 
 log = set_log('D:\code\mango_automation\logs')
+log.set_debug(True)
 test_data = DataProcessor()
 element_model = ElementModel(**{
     "id": 3,
@@ -20,9 +21,10 @@ element_model = ElementModel(**{
     "elements": [
         {
             "exp": 0,
-            "loc": "//span[@name=\"tj_settingicon\"]",
+            "loc": "//span[@name=\"tj_settingicons\"]",
             "sub": None,
-            "is_iframe": 0
+            "is_iframe": 0,
+            "prompt": "设置"
         }
     ],
     "sleep": None,
@@ -47,7 +49,8 @@ class TestUi(unittest.IsolatedAsyncioTestCase):
     async def test_a(self):
         driver_object = DriverObject(log, True)
         driver_object.set_web(0, r"C:\Program Files\Google\Chrome\Application\chrome.exe")
-        base_data = BaseData(test_data, log)
+        base_data = BaseData(test_data, log).set_agent(True, 'sk-rruuhjnqawsvduyxlcqckbtgwkprctgkvwcelenooixbhthy')
+        base_data.screenshot_path = r'D:\code\mango_automation\logs'
         base_data.log = log
         base_data.url = 'https://www.baidu.com/'
 
@@ -56,6 +59,7 @@ class TestUi(unittest.IsolatedAsyncioTestCase):
         await element.open_url()
         await asyncio.sleep(5)
         await element.element_main(element_model, )
+        print(element.element_result_model.model_dump_json())
         assert element.element_result_model.elements[0].element_text == '设置'