mangoautomation 1.0.0__py3-none-any.whl

mangoautomation/uidrive/android/_element.py

@@ -0,0 +1,168 @@
+# -*- 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 ...exceptions import MangoAutomationError
+from ...exceptions._error_msg import ERROR_MSG_0043, ERROR_MSG_0044
+from ...tools import Meta
+from ...uidrive._base_data import BaseData
+from mangotools.decorator import sync_method_callback
+from mangotools.models import MethodModel
+
+
+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(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(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(f='x', p='请输入x坐标', d=True),
+        MethodModel(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(f='x', p='请输入x坐标', d=True),
+        MethodModel(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(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(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(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(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(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(f='x', p='请输入x坐标', d=True),
+        MethodModel(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(f='x_key', p='请输入x坐标', d=True),
+        MethodModel(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

mangoautomation/uidrive/android/_equipment.py

@@ -0,0 +1,150 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台# @Description:
+# @Time   : 03-09-09 3:17
+# @Author : 毛鹏
+from time import sleep
+
+from ...tools import Meta
+from ...uidrive._base_data import BaseData
+from mangotools.decorator import sync_method_callback
+from mangotools.models import MethodModel
+
+
+class AndroidEquipment(metaclass=Meta):
+    """设备操作"""
+
+    def __init__(self, base_data: BaseData):
+        self.base_data = base_data
+
+    @sync_method_callback('android', '设备操作', 1, [
+        MethodModel(f='_time', p='请输入等待时间', d=True)])
+    def a_sleep(self, time_: int):
+        """强制等待"""
+        sleep(time_)
+
+    @sync_method_callback('android', '设备操作', )
+    def a_screen_on(self):
+        """打开屏幕"""
+        self.base_data.android.screen_on()
+        self.a_sleep(1)
+
+    @sync_method_callback('android', '设备操作', )
+    def a_screen_off(self):
+        """关闭屏幕"""
+        self.base_data.android.screen_off()
+        self.a_sleep(1)
+
+    @sync_method_callback('android', '设备操作', )
+    def a_swipe_left(self):
+        """获取屏幕开关状态"""
+        self.base_data.android.info.get('screenOn')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_get_window_size(self):
+        """提取屏幕尺寸"""
+        w, h = self.base_data.android.window_size()
+        return w, h
+
+    @sync_method_callback('android', '设备操作', [
+        MethodModel(f='feli_path', p='请输入计算机文件路径', d=True),
+        MethodModel(f='catalogue', p='请输入设备目录', d=True)])
+    def a_push(self, feli_path, catalogue):
+        """推送一个文件到设备"""
+        self.base_data.android.push(feli_path, catalogue)
+
+    @sync_method_callback('android', '设备操作', [
+        MethodModel(f='feli_path', p='请输入设备文件路径', d=True),
+        MethodModel(f='catalogue', p='请输入计算机目录', d=True)])
+    def a_pull(self, feli_path, catalogue):
+        """提取文件"""
+        self.base_data.android.pull(feli_path, catalogue)
+
+    @sync_method_callback('android', '设备操作', )
+    def a_unlock(self):
+        """解锁屏幕"""
+        self.base_data.android.unlock()
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_home(self):
+        """按home键"""
+        self.base_data.android.press('home')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_back(self):
+        """按back键"""
+        self.base_data.android.press('back')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_left(self):
+        """按left键"""
+        self.base_data.android.press('left')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_right(self):
+        """按right键"""
+        self.base_data.android.press('right')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_up(self):
+        """按up键"""
+        self.base_data.android.press('up')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_down(self):
+        """按down键"""
+        self.base_data.android.press('down')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_center(self):
+        """按center键"""
+        self.base_data.android.press('center')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_menu(self):
+        """按menu键"""
+        self.base_data.android.press('menu')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_search(self):
+        """按search键"""
+        self.base_data.android.press('search')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_enter(self):
+        """按enter键"""
+        self.base_data.android.press('enter')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_delete(self):
+        """按delete键"""
+        self.base_data.android.press('delete')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_recent(self):
+        """按recent键"""
+        self.base_data.android.press('recent')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_volume_up(self):
+        """按volume_up键"""
+        self.base_data.android.press('volume_up')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_volume_down(self):
+        """按volume_down键"""
+        self.base_data.android.press('volume_down')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_volume_mute(self):
+        """按volume_mute键"""
+        self.base_data.android.press('volume_mute')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_camera(self):
+        """按camera键"""
+        self.base_data.android.press('camera')
+
+    @sync_method_callback('android', '设备操作', )
+    def a_press_power(self):
+        """按power键"""
+        self.base_data.android.press('power')

mangoautomation/uidrive/android/_new_android.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2024-05-23 15:05
+# @Author : 毛鹏
+
+from typing import Optional
+
+import uiautomator2 as u2
+from adbutils import AdbTimeout
+from uiautomator2 import ConnectError
+
+from ...exceptions import MangoAutomationError
+from ...exceptions._error_msg import ERROR_MSG_0042, ERROR_MSG_0045, ERROR_MSG_0040
+
+"""
+python -m uiautomator2 init
+python -m weditor
+
+"""
+
+
+class NewAndroid:
+
+    def __init__(self, and_equipment):
+        self.and_equipment = and_equipment
+        self.info: Optional[dict | None] = None
+        self.example_dict = []
+
+    def new_android(self):
+        if self.and_equipment is None:
+            raise MangoAutomationError(*ERROR_MSG_0042)
+        try:
+
+            android = u2.connect(self.and_equipment)
+            self.info = android.info
+            # msg = f"设备启动成功！产品名称：{self.info.get('productName')}"
+            self.example_dict.append({
+                'config': self.and_equipment,
+                'info': self.info,
+                'android': android
+            })
+        except ConnectError:
+            raise MangoAutomationError(*ERROR_MSG_0040, value=(self.and_equipment,))
+        except RuntimeError:
+            raise MangoAutomationError(*ERROR_MSG_0045, value=(self.and_equipment,))
+        except (AdbTimeout, TimeoutError):
+            raise MangoAutomationError(*ERROR_MSG_0040, value=(self.and_equipment,))
+        else:
+            android.implicitly_wait(10)
+            return android
+
+    def close_android(self):
+        pass

mangoautomation/uidrive/android/_page.py

@@ -0,0 +1,116 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: # @Time   : 2023/4/6 13:31
+# @Author : 毛鹏
+import os.path
+
+from uiautomator2 import Direction
+
+from ...tools import Meta
+from ...uidrive._base_data import BaseData
+from mangotools.decorator import sync_method_callback
+from mangotools.models import MethodModel
+
+
+class AndroidPage(metaclass=Meta):
+    """页面操作"""
+
+    def __init__(self, base_data: BaseData):
+        self.base_data = base_data
+
+    @sync_method_callback('android', '页面操作')
+    def a_swipe_right(self):
+        """右滑"""
+        self.base_data.android.swipe_ext(Direction.HORIZ_FORWARD)
+
+    @sync_method_callback('android', '页面操作')
+    def a_swipe_left(self):
+        """左滑"""
+        self.base_data.android.swipe_ext(Direction.HORIZ_BACKWARD)
+
+    @sync_method_callback('android', '页面操作')
+    def a_swipe_up(self):
+        """上滑"""
+        self.base_data.android.swipe_ext(Direction.FORWARD)
+
+    @sync_method_callback('android', '页面操作')
+    def a_swipe_down(self):
+        """下滑"""
+        self.base_data.android.swipe_ext(Direction.BACKWARD)
+
+    @sync_method_callback('android', '页面操作', [
+        MethodModel(f='sx', p='请输入sx坐标', d=True),
+        MethodModel(f='sy', p='请输入sy坐标', d=True),
+        MethodModel(f='ex', p='请输入ex坐标', d=True),
+        MethodModel(f='ey', p='请输入ey坐标', d=True)])
+    def a_swipe(self, sx, sy, ex, ey):
+        """坐标滑动"""
+        self.base_data.android.swipe(sx, sy, ex, ey, 0.5)
+
+    @sync_method_callback('android', '页面操作', [
+        MethodModel(f='sx', p='请输入sx坐标', d=True),
+        MethodModel(f='sy', p='请输入sy坐标', d=True),
+        MethodModel(f='ex', p='请输入ex坐标', d=True),
+        MethodModel(f='ey', p='请输入ey坐标', d=True)])
+    def a_drag(self, sx, sy, ex, ey):
+        """坐标拖动"""
+        self.base_data.android.drag(sx, sy, ex, ey, 0.5)
+
+    @sync_method_callback('android', '页面操作')
+    def a_open_quick_settings(self):
+        """打开快速通知"""
+        self.base_data.android.open_quick_settings()
+
+    @sync_method_callback('android', '页面操作', [
+        MethodModel(f='file_name', p='请输入截图文件名称', d=True)])
+    def a_screenshot(self, file_name: str):
+        """屏幕截图"""
+        self.base_data.android.screenshot(filename=os.path.join(self.base_data.screenshot_path, file_name))
+
+    @sync_method_callback('android', '页面操作', [
+        MethodModel(f='x', p='请输入按下的x坐标', d=True),
+        MethodModel(f='y', p='请输入按下的x坐标', d=True),
+        MethodModel(f='time_', p='请输入长按时间', d=True)])
+    def a_long_click(self, x, y, time_):
+        """长按屏幕N秒"""
+        self.base_data.android.long_click(x, y, time_)
+
+    @sync_method_callback('android', '页面操作')
+    def a_set_orientation_natural(self):
+        """设置为natural"""
+        self.base_data.android.set_orientation("natural")
+
+    @sync_method_callback('android', '页面操作')
+    def a_set_orientation_left(self):
+        """设置为natural"""
+        self.base_data.android.set_orientation("left")
+
+    @sync_method_callback('android', '页面操作')
+    def a_set_orientation_right(self):
+        """设置为right"""
+        self.base_data.android.set_orientation("right")
+
+    @sync_method_callback('android', '页面操作')
+    def a_set_orientation_upsidedown(self):
+        """设置为upsidedown"""
+        self.base_data.android.set_orientation("upsidedown")
+
+    @sync_method_callback('android', '页面操作')
+    def a_freeze_rotation(self):
+        """冻结旋转"""
+        self.base_data.android.freeze_rotation()
+
+    @sync_method_callback('android', '页面操作')
+    def a_freeze_rotation_false(self):
+        """取消冻结旋转"""
+        self.base_data.android.freeze_rotation(False)
+
+    @sync_method_callback('android', '页面操作')
+    def a_dump_hierarchy(self):
+        """获取转储的内容"""
+        return self.base_data.android.dump_hierarchy()
+
+    @sync_method_callback('android', '页面操作')
+    def a_open_notification(self):
+        """打开通知"""
+        return self.base_data.android.dump_hierarchy()

mangoautomation/uidrive/pc/__init__.py

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: # @Time   : 2023-07-15 12:02
+# @Author : 毛鹏
+from typing import Optional
+
+import sys
+
+if not sys.platform.startswith('win32'):
+    print("警告: uiautomation 仅支持 Windows，当前环境已自动跳过")
+
+
+    class Auto:
+        pass
+
+
+    auto = Auto()
+
+
+    class Control:
+        pass
+else:
+
+    from uiautomation import Control
+    import uiautomation
+
+from mangoautomation.uidrive._base_data import BaseData
+from mangoautomation.uidrive.pc.element import WinElement
+from mangoautomation.uidrive.pc.input_device import WinDeviceInput
+
+
+class WinDriver(WinElement, WinDeviceInput):
+    def __init__(self, base_data: BaseData):
+        super().__init__(base_data)
+
+    def find_element(
+            self,
+            *,
+            control_type: str = "Control",
+            name: Optional[str] = None,
+            automation_id: Optional[str] = None,
+            depth: int = 2,
+            parent: Optional[Control] = None,
+            **extra_attrs
+    ) -> Control:
+
+        if not any([name, automation_id, extra_attrs]):
+            raise ValueError("必须提供至少一个定位参数(name/automation_id/其他属性)")
+
+        # 设置默认父控件
+        parent = parent or self.base_data.window.GetParentControl()
+
+        # 构造搜索参数
+        search_params = {
+            "searchDepth": depth,
+            "Timeout": 10 * 1000,
+            **{
+                key.capitalize() if key != "className" else "ClassName": value
+                for key, value in {
+                    "name": name,
+                    "automationId": automation_id,
+                    **extra_attrs
+                }.items()
+                if value is not None
+            }
+        }
+
+        # 获取控件类并查找
+        control_class = getattr(uiautomation, f"{control_type}Control", Control)
+        control = control_class(parent=parent, **search_params)
+
+        if not control.Exists():
+            raise ElementNotFoundError(
+                f"未找到控件: type={control_type}, params={search_params}"
+            )
+        return control
+
+
+class ElementNotFoundError(Exception):
+    pass

mangoautomation/uidrive/pc/assertion.py

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2024-11-30 23:44
+# @Author : 毛鹏

mangoautomation/uidrive/pc/customization.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2024-08-13 14:07
+# @Author : 毛鹏
+from ...uidrive._base_data import BaseData
+
+
+class PlaywrightCustomization(BaseData):
+    """定制开发"""

mangoautomation/uidrive/pc/element.py

@@ -0,0 +1,21 @@
+from ...tools import Meta
+from ...uidrive._base_data import BaseData
+
+
+class WinElement(metaclass=Meta):
+    """元素操作"""
+
+    def __init__(self, base_data: BaseData):
+        self.base_data = base_data
+
+    def click(self, control):
+        """点击控件"""
+        control.Click()
+
+    def input_text(self, control, text: str):
+        """输入文本"""
+        control.SendKeys(text)
+
+    def get_text(self, control) -> str:
+        """获取控件文本"""
+        return control.Name

mangoautomation/uidrive/pc/input_device.py

@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2024-11-30 23:44
+# @Author : 毛鹏
+from ...tools import Meta
+from ...uidrive._base_data import BaseData
+
+
+class WinDeviceInput(metaclass=Meta):
+    """输入设备操作"""
+
+    def __init__(self, base_data: BaseData):
+        self.base_data = base_data