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

tests/demo1.py

@@ -0,0 +1,94 @@
+from bs4 import BeautifulSoup
+
+
+def get_optimized_page_html(html_content) -> str:
+    soup = BeautifulSoup(html_content, 'html.parser')
+
+    # 配置化的清理规则
+    cleanup_rules = {
+        # 直接删除的标签
+        'remove_tags': [
+            'script', 'style', 'link', 'noscript', 'path',
+            'svg', 'head', 'symbol', 'doubao-ai-csui', 'template'
+        ],
+
+        # 按class删除的元素
+        'remove_by_class': [
+            'watermark-box'
+        ],
+
+        # 按属性删除的元素
+        'remove_by_attrs': {
+            # 'tag_name': {'attr': 'value'}
+        },
+
+        # 清空内容但保留标签的条件
+        'empty_content_conditions': [
+            {
+                'tag': 'textarea',
+                'condition': lambda el: el.get('style') and 'display:none' in el.get('style')
+            }
+        ],
+
+        # 要删除的属性
+        'remove_attributes': [
+            'style',
+            'ssr'
+        ],
+
+        # 要删除的属性前缀（包含该前缀的属性都会被删除）
+        'remove_attribute_prefixes': [
+            'data-v-',
+            'data-testid'
+        ]
+    }
+
+    # 执行清理规则
+    # 1. 删除指定标签
+    for tag_name in cleanup_rules['remove_tags']:
+        for element in soup.find_all(tag_name):
+            element.decompose()
+
+    # 2. 按class删除元素
+    for class_name in cleanup_rules['remove_by_class']:
+        for element in soup.find_all(class_=class_name):
+            element.decompose()
+
+    # 3. 按属性删除元素
+    for tag_name, attrs in cleanup_rules['remove_by_attrs'].items():
+        for element in soup.find_all(tag_name, attrs=attrs):
+            element.decompose()
+
+    # 4. 清空特定元素内容
+    for condition_config in cleanup_rules['empty_content_conditions']:
+        tag = condition_config['tag']
+        condition_func = condition_config['condition']
+        for element in soup.find_all(tag):
+            if condition_func(element):
+                element.string = ''
+
+    # 5. 删除属性和属性前缀
+    for element in soup.find_all():
+        # 删除指定属性
+        for attr in cleanup_rules['remove_attributes']:
+            if element.has_attr(attr):
+                del element[attr]
+
+        # 删除包含特定前缀的属性
+        for attr_prefix in cleanup_rules['remove_attribute_prefixes']:
+            for attr in list(element.attrs.keys()):
+                if attr.startswith(attr_prefix):
+                    del element[attr]
+
+    # 将BeautifulSoup对象转换为字符串
+    html_content_str = str(soup)
+    return html_content_str
+
+
+with open('element.txt', 'r', encoding='utf-8') as f:
+    file_content = f.read()
+    optimize = get_optimized_page_html(file_content)
+    print(f'原始的html大小：{len(file_content)}')
+    print(f'优化之后的html大小：{len(optimize)}')
+    with open('optimized.html', 'w', encoding='utf-8') as f:
+        f.write(optimize)

tests/get_ope.py

@@ -3,6 +3,15 @@
 # @Description: 
 # @Time   : 2025-08-31 17:41
 # @Author : 毛鹏
-loc = 'await page.cdeadapage.'
-cleaned_loc = loc.replace('page.', '').replace('await', '')
-print(cleaned_loc)
+from mangoautomation.uidrive import DriverObject
+
+d = DriverObject(1, )
+d.set_web(web_type=2,  # type: ignore
+          web_path=3,  # type: ignore
+          web_max=1,
+          web_headers=4,
+          web_recording=5,
+          web_h5=1,
+          web_is_default=1,  # type: ignore
+          videos_path=2,
+          user_cache_path=3)

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 == '设置'
 
 

mangoautomation/uidrives/android/_application.py

@@ -1,70 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Project: 芒果测试平台
-# @Description:
-# @Time   : 2023-09-09 23:17
-# @Author : 毛鹏
-import time
-
-from mangotools.decorator import sync_method_callback
-from mangotools.models import MethodModel
-from ...exceptions import MangoAutomationError
-from ...exceptions.error_msg import ERROR_MSG_0046
-from ...tools import Meta
-from ...uidrives._base_data import BaseData
-
-
-class AndroidApplication(metaclass=Meta):
-    """应用操作"""
-
-    def __init__(self, base_data: BaseData):
-        self.base_data = base_data
-
-    def is_app_installed(self, package_name: str) -> bool:
-        return any(package_name in str(i) for i in self.base_data.android.shell("pm list packages"))
-
-    @sync_method_callback('android', '应用操作', 0,
-                          [MethodModel(n='包名', f='package_name', p='请输入应用名称', d=True)])
-    def a_start_app(self, package_name: str):
-        """启动应用"""
-        if not package_name:
-            raise MangoAutomationError(*ERROR_MSG_0046)
-        if not self.is_app_installed(package_name):
-            raise MangoAutomationError(*ERROR_MSG_0046)
-        self.base_data.android.app_start(package_name)
-        time.sleep(4)
-
-    @sync_method_callback('android', '应用操作', 1, [
-        MethodModel(n='包名', f='package_name', p='请输入应用名称', d=True)])
-    def a_close_app(self, package_name: str):
-        """关闭应用"""
-        if not package_name:
-            raise MangoAutomationError(*ERROR_MSG_0046)
-        if not self.is_app_installed(package_name):
-            raise MangoAutomationError(*ERROR_MSG_0046)
-
-        self.base_data.android.app_stop(package_name)
-
-    @sync_method_callback('android', '应用操作', 2, [
-        MethodModel(n='包名', f='package_name', p='请输入应用名称', d=True)])
-    def a_clear_app(self, package_name: str):
-        """清除app数据"""
-        if not package_name:
-            raise MangoAutomationError(*ERROR_MSG_0046)
-        if not self.is_app_installed(package_name):
-            raise MangoAutomationError(*ERROR_MSG_0046)
-
-        self.base_data.android.app_clear(package_name)
-
-    @sync_method_callback('android', '应用操作', 3)
-    def a_app_stop_all(self):
-        """停止所有app"""
-        self.base_data.android.app_stop_all()
-
-    @sync_method_callback('android', '应用操作', 4, [
-        MethodModel(n='包名List', f='package_name', p='请输入应用名称列表', d=True)])
-    def a_app_stop_appoint(self, package_name_list: list):
-        """停止除指定app外所有app"""
-        for i in package_name_list:
-            if not self.is_app_installed(i):
-                raise MangoAutomationError(*ERROR_MSG_0046)
-        self.base_data.android.app_stop_all(excludes=package_name_list)

mangoautomation/uidrives/android/_assertion.py

@@ -1,90 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Project: 芒果测试平台
-# @Description:
-# @Time   : 2023-09-09 23:17
-# @Author : 毛鹏
-from uiautomator2 import UiObject
-
-from mangoautomation.tools import Meta
-from mangoautomation.uidrives._base_data import BaseData
-from mangotools.decorator import sync_method_callback
-from mangotools.models import MethodModel
-
-
-class AndroidAssertion(metaclass=Meta):
-    """元素断言"""
-
-    def __init__(self, base_data: BaseData):
-        self.base_data = base_data
-
-    @sync_method_callback('ass_android', '元素断言', 1, [
-        MethodModel(f='actual')])
-    def a_assert_ele_exists(self, actual: UiObject):
-        """元素存在"""
-        assert actual.count, f'实际={actual.count}, 预期=元素存在'
-        return f'实际={actual.count}, 预期=元素存在'
-
-    @sync_method_callback('ass_android', '元素断言', 2, [
-        MethodModel(f='actual'),
-        MethodModel(n='预期值', f='expect', p='请输入预期内容', d=True)])
-    def a_assert_ele_count(self, actual: UiObject, expect):
-        """元素计数"""
-        assert int(actual.count) == int(expect), f'实际={actual.count}, 预期={expect}'
-        return f'实际={actual.count}, 预期={expect}'
-
-    @sync_method_callback('ass_android', '元素断言', 3, [
-        MethodModel(f='actual'),
-        MethodModel(n='预期值', f='expect', p='请输入预期文本', d=True)])
-    def a_assert_ele_text(self, actual: UiObject, expect: str):
-        """元素文本内容"""
-        assert actual.get_text() == expect, f"实际='{actual.get_text()}', 预期='{expect}'"
-        return f'实际={actual.get_text()}, 预期={expect}'
-
-    @sync_method_callback('ass_android', '元素断言', 4, [
-        MethodModel(f='actual')])
-    def a_assert_ele_clickable_true(self, actual: UiObject):
-        """元素可点击"""
-        assert actual.info['clickable'], f"实际={actual.info['clickable']}, 预期=可点击"
-        return f"实际={actual.info['clickable']}, 预期=可点击"
-
-    @sync_method_callback('ass_android', '元素断言', 5, [
-        MethodModel(f='actual')])
-    def a_assert_ele_clickable_false(self, actual: UiObject):
-        """元素不可点击"""
-        assert not actual.info['clickable'], f"实际={actual.info['clickable']}, 预期=元素不可点击"
-        return f"实际={actual.info['clickable']}, 预期=元素不可点击"
-
-    @sync_method_callback('ass_android', '元素断言', 6, [
-        MethodModel(f='actual')])
-    def a_assert_ele_visible_true(self, actual: UiObject):
-        """元素可见"""
-        assert actual.exists and actual.info['visible'], f"实际={actual.info['visible']}, 预期=元素可见"
-
-        return f"实际={actual.info['visible']}, 预期=元素可见"
-
-    @sync_method_callback('ass_android', '元素断言', 7, [
-        MethodModel(f='actual')])
-    def a_assert_ele_visible_false(self, actual: UiObject):
-        """元素不可见"""
-        assert actual.exists and not actual.info['visible'], f"实际={actual.info['visible']}, 预期=元素不可见"
-        return f"实际={actual.info['visible']}, 预期=元素不可见"
-
-    @sync_method_callback('ass_android', '元素断言', 8, [
-        MethodModel(n='预期值', f='expect', p='请输入弹窗标题文本', d=False)])
-    def a_assert_dialog_exists(self, expect: str):
-        """弹窗存在"""
-        dialog = self.base_data.android(text=expect) if expect else self.base_data.android(
-            className="android.app.AlertDialog")
-        assert dialog.exists, "未找到预期弹窗"
-        return f"实际={dialog.exists}, 预期=弹窗存在"
-
-    @sync_method_callback('ass_android', '元素断言', 9, [
-        MethodModel(f='actual'),
-        MethodModel(n='预期值', f='expect', p='请输入断言目标文本', d=True)])
-    def a_assert_ele_in_list(self, actual: UiObject, expect: str):
-        """列表滑动后目标元素存在"""
-        if not actual.exists:
-            raise AssertionError("传入的元素不是可滑动的列表")
-        actual.scroll.vert.to(text=expect)
-        assert self.base_data.android(text=expect).exists, f"列表中未找到文本: {expect}"
-        return f"实际={self.base_data.android(text=expect).exists}, 预期={expect}"

mangoautomation/uidrives/android/_customization.py

@@ -1,15 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Project: 芒果测试平台
-# @Description: # @Time   : 2023-04-26 22:22
-# @Author : 毛鹏
-
-
-from mangoautomation.tools import Meta
-from mangoautomation.uidrives._base_data import BaseData
-
-
-class AndroidCustomization(metaclass=Meta):
-    """定制开发"""
-
-    def __init__(self, base_data: BaseData):
-        self.base_data = base_data

mangoautomation/uidrives/android/_element.py

@@ -1,169 +0,0 @@
-# -*- coding: utf-8 -*-
-# @Project: 芒果测试平台
-# @Description:
-# @Time   : 2023-09-09 23:17
-# @Author : 毛鹏
-import os.path
-
-import time
-from uiautomator2 import UiObject
-from uiautomator2.xpath import XPathSelector
-
-from mangotools.decorator import sync_method_callback
-from mangotools.models import MethodModel
-from ...exceptions import MangoAutomationError
-from ...exceptions.error_msg import ERROR_MSG_0043, ERROR_MSG_0044
-from ...tools import Meta
-from ...uidrives._base_data import BaseData
-
-
-class AndroidElement(metaclass=Meta):
-    """元素操作"""
-
-    def __init__(self, base_data: BaseData):
-        self.base_data = base_data
-
-    @sync_method_callback('android', '元素操作', 0, [
-        MethodModel(f='locating')])
-    def a_click(self, locating: UiObject | XPathSelector):
-        """元素单击"""
-        locating.click()
-
-    @sync_method_callback('android', '元素操作', 1, [
-        MethodModel(f='locating')])
-    def a_double_click(self, locating: UiObject):
-        """元素双击"""
-        locating.click()
-
-    @sync_method_callback('android', '元素操作', 2, [
-        MethodModel(n='输入文本', f='locating'), MethodModel(f='text', p='请输入内容', d=True)])
-    def a_input(self, locating: UiObject, text):
-        """单击输入"""
-        locating.click()
-        self.base_data.android.set_fastinput_ime(True)
-        time.sleep(1)
-        self.base_data.android.send_keys(text)
-
-    @sync_method_callback('android', '元素操作', 3, [
-        MethodModel(f='locating'), MethodModel(n='设置文本', f='text', p='请输入内容', d=True)])
-    def a_set_text(self, locating: UiObject, text):
-        """设置文本"""
-        locating.set_text(text)
-
-    @sync_method_callback('android', '元素操作', 4, [
-        MethodModel(f='locating'),
-        MethodModel(n='x坐标', f='x', p='请输入x坐标', d=True),
-        MethodModel(n='y坐标', f='y', p='请输入y坐标', d=True)])
-    def a_click_coord(self, x, y):
-        """坐标单击"""
-        self.base_data.android.click(x, y)
-
-    @sync_method_callback('android', '元素操作', 5, [
-        MethodModel(n='x坐标', f='x', p='请输入x坐标', d=True),
-        MethodModel(n='y坐标', f='y', p='请输入y坐标', d=True)])
-    def a_double_click_coord(self, x, y):
-        """坐标双击"""
-        self.base_data.android.double_click(x, y)
-
-    @sync_method_callback('android', '元素操作', 6, [
-        MethodModel(f='locating'),
-        MethodModel(n='长按时间', f='time_', p='请输入长按时间', d=True)])
-    def a_long_click(self, locating: UiObject, time_):
-        """长按元素"""
-        locating.long_click(duration=float(time_))
-
-    @sync_method_callback('android', '元素操作', 7, [MethodModel(f='locating')])
-    def a_clear_text(self, locating: UiObject):
-        """清空输入框"""
-        locating.clear_text()
-
-    @sync_method_callback('android', '元素操作', 8, [
-        MethodModel(f='locating'), MethodModel(n='缓存的key', f='set_cache_key', p='请输入元素文本存储的key', d=True)])
-    def a_get_text(self, locating: UiObject, set_cache_key=None):
-        """获取元素文本"""
-        value = locating.get_text()
-        if set_cache_key:
-            self.base_data.test_data.set_cache(key=set_cache_key, value=value)
-        return value
-
-    @sync_method_callback('android', '元素操作', 9, [
-        MethodModel(f='locating'),
-        MethodModel(n='截图名称', f='file_name', p='请输入元素截图存储的名称，后续可以通过名称获取', d=True)])
-    def a_element_screenshot(self, locating: UiObject, file_name: str):
-        """元素截图"""
-        im = locating.screenshot()
-        file_path = os.path.join(self.base_data.screenshot_path, file_name)
-        self.base_data.test_data.set_cache(file_name, file_path)
-        im.save(file_path)
-
-    @sync_method_callback('android', '元素操作', 10, [
-        MethodModel(f='locating')])
-    def a_pinch_in(self, locating: UiObject):
-        """元素缩小"""
-        locating.pinch_in()
-
-    @sync_method_callback('android', '元素操作', 11, [
-        MethodModel(f='locating')])
-    def a_pinch_out(self, locating: UiObject):
-        """元素放大"""
-        locating.pinch_out()
-
-    @sync_method_callback('android', '元素操作', 12, [
-        MethodModel(f='locating'), MethodModel(n='等待时间', f='time_', p='请输入等待元素出现的时间', d=True)])
-    def a_wait(self, locating: UiObject, time_):
-        """等待元素出现"""
-        if not locating.wait(timeout=float(time_)):
-            raise MangoAutomationError(*ERROR_MSG_0043)
-
-    @sync_method_callback('android', '元素操作', 13, [
-        MethodModel(f='locating'), MethodModel(n='等待时间', f='time_', p='请输入等待元素消失的时间', d=True)])
-    def a_wait_gone(self, locating: UiObject, time_: str):
-        """等待元素消失"""
-        if not locating.wait_gone(timeout=float(time_)):
-            raise MangoAutomationError(*ERROR_MSG_0044)
-
-    @sync_method_callback('android', '元素操作', 14, [
-        MethodModel(f='locating'), MethodModel(f='locating2')])
-    def a_drag_to_ele(self, locating: UiObject, locating2: UiObject):
-        """拖动A元素到达B元素上"""
-        locating.drag_to(locating2)
-
-    @sync_method_callback('android', '元素操作', 15, [
-        MethodModel(f='locating'),
-        MethodModel(n='x坐标', f='x', p='请输入x坐标', d=True),
-        MethodModel(n='y坐标', f='y', p='请输入y坐标', d=True)])
-    def a_drag_to_coord(self, locating: UiObject, x, y):
-        """拖动元素到坐标上"""
-        locating.drag_to(x, y)
-
-    @sync_method_callback('android', '元素操作', 16, [MethodModel(f='locating')])
-    def a_swipe_right(self, locating: UiObject):
-        """元素内向右滑动"""
-        locating.swipe('right')
-
-    @sync_method_callback('android', '元素操作', 17, [MethodModel(f='locating')])
-    def a_swipe_left(self, locating: UiObject):
-        """元素内向左滑动"""
-        locating.swipe('left')
-
-    @sync_method_callback('android', '元素操作', 18, [MethodModel(f='locating')])
-    def a_swipe_up(self, locating: UiObject):
-        """元素内向上滑动"""
-        locating.swipe('up')
-
-    @sync_method_callback('android', '元素操作', 19, [MethodModel(f='locating')])
-    def a_swipe_ele(self, locating: UiObject):
-        """元素内向下滑动"""
-        locating.swipe('down')
-
-    @sync_method_callback('android', '元素操作', 20, [
-        MethodModel(f='locating'),
-        MethodModel(n='x坐标', f='x_key', p='请输入x坐标', d=True),
-        MethodModel(n='y坐标', f='y_key', p='请输入y坐标', d=True)])
-    def a_get_center(self, locating: UiObject, x_key, y_key):
-        """提取元素坐标"""
-        x, y = locating.center()
-        if x_key and y_key:
-            self.base_data.test_data.set_cache(key=x_key, value=x)
-            self.base_data.test_data.set_cache(key=y_key, value=y)
-        return x, y