uiautodev 0.3.7__tar.gz → 0.4.0__tar.gz

{uiautodev-0.3.7 → uiautodev-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: uiautodev
-Version: 0.3.7
+Version: 0.4.0
 Summary: Mobile UI Automation, include UI hierarchy inspector, script recorder
 Home-page: https://uiauto.dev
 License: MIT
@@ -15,7 +15,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: appium
-Requires-Dist: adbutils (>=2.6.0,<3.0.0)
+Requires-Dist: adbutils (>=2.7.0,<3.0.0)
 Requires-Dist: appium-python-client (>=4.0.0,<5.0.0) ; extra == "appium"
 Requires-Dist: click (>=8.1.7,<9.0.0)
 Requires-Dist: construct
@@ -29,6 +29,7 @@ Requires-Dist: pydantic (>=2.6,<3.0)
 Requires-Dist: pygments (>=2)
 Requires-Dist: uiautomator2 (>=2)
 Requires-Dist: uvicorn[standard]
+Requires-Dist: wdapy (>=0.2.2,<0.3.0)
 Description-Content-Type: text/markdown
 
 # uiautodev

{uiautodev-0.3.7 → uiautodev-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "uiautodev"
-version = "0.3.7"
+version = "0.4.0"
 description = "Mobile UI Automation, include UI hierarchy inspector, script recorder"
 homepage = "https://uiauto.dev"
 authors = ["codeskyblue <codeskyblue@gmail.com>"]
@@ -10,7 +10,7 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.8"
 pillow = "*"
-adbutils = "^2.6.0"
+adbutils = "^2.7.0"
 construct = "*"
 lxml = "*"
 click = "^8.1.7"
@@ -23,6 +23,7 @@ fastapi = "^0.111.0"
 uvicorn = {version = "*", extras = ["standard"]}
 poetry = "^1.8.2"
 pydantic = "^2.6"
+wdapy = "^0.2.2"
 
 [tool.poetry.extras]
 appium = ["appium-python-client", "httppretty"]

uiautodev-0.4.0/uiautodev/__init__.py

@@ -0,0 +1,8 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+"""Created on Mon Mar 04 2024 14:28:53 by codeskyblue
+"""
+
+# version is auto managed by poetry
+__version__ = "0.4.0"

{uiautodev-0.3.7 → uiautodev-0.4.0}/uiautodev/command_proxy.py

@@ -105,6 +105,31 @@ def home(driver: BaseDriver):
     driver.home()
 
 
+@register(Command.BACK)
+def back(driver: BaseDriver):
+    driver.back()
+
+
+@register(Command.APP_SWITCH)
+def app_switch(driver: BaseDriver):
+    driver.app_switch()
+
+
+@register(Command.VOLUME_UP)
+def volume_up(driver: BaseDriver):
+    driver.volume_up()
+
+
+@register(Command.VOLUME_DOWN)
+def volume_down(driver: BaseDriver):
+    driver.volume_down()
+
+
+@register(Command.VOLUME_MUTE)
+def volume_mute(driver: BaseDriver):
+    driver.volume_mute()
+
+
 @register(Command.DUMP)
 def dump(driver: BaseDriver) -> DumpResponse:
     source, _ = driver.dump_hierarchy()

{uiautodev-0.3.7 → uiautodev-0.4.0}/uiautodev/command_types.py

@@ -32,6 +32,13 @@ class Command(str, enum.Enum):
 
     LIST = "list"
 
+    # 0.4.0
+    BACK = "back"
+    APP_SWITCH = "appSwitch"
+    VOLUME_UP = "volumeUp"
+    VOLUME_DOWN = "volumeDown"
+    VOLUME_MUTE = "volumeMute"
+
 
 class TapRequest(BaseModel):
     x: Union[int, float]

{uiautodev-0.3.7 → uiautodev-0.4.0}/uiautodev/driver/android.py

@@ -144,6 +144,21 @@ class AndroidDriver(BaseDriver):
     
     def wake_up(self):
         self.adb_device.keyevent("WAKEUP")
+    
+    def back(self):
+        self.adb_device.keyevent("BACK")
+    
+    def app_switch(self):
+        self.adb_device.keyevent("APP_SWITCH")
+    
+    def volume_up(self):
+        self.adb_device.keyevent("VOLUME_UP")
+    
+    def volume_down(self):
+        self.adb_device.keyevent("VOLUME_DOWN")
+    
+    def volume_mute(self):
+        self.adb_device.keyevent("VOLUME_MUTE")
 
 
 def parse_xml(xml_data: str, wsize: WindowSize, display_id: Optional[int] = None) -> Node:

{uiautodev-0.3.7 → uiautodev-0.4.0}/uiautodev/driver/base_driver.py

@@ -70,6 +70,26 @@ class BaseDriver(abc.ABC):
     def home(self):
         """ press home button """
         raise NotImplementedError()
+    
+    def back(self):
+        """ press back button """
+        raise NotImplementedError()
+
+    def app_switch(self):
+        """ switch app """
+        raise NotImplementedError()
+    
+    def volume_up(self):
+        """ volume up """
+        raise NotImplementedError()
+    
+    def volume_down(self):
+        """ volume down """
+        raise NotImplementedError()
+    
+    def volume_mute(self):
+        """ volume mute """
+        raise NotImplementedError()
 
     def wake_up(self):
         """ wake up the device """

{uiautodev-0.3.7 → uiautodev-0.4.0}/uiautodev/driver/ios.py

@@ -13,13 +13,14 @@ from functools import partial
 from typing import List, Optional, Tuple
 from xml.etree import ElementTree
 
+import wdapy
 from PIL import Image
 
 from uiautodev.command_types import CurrentAppResponse
 from uiautodev.driver.base_driver import BaseDriver
 from uiautodev.exceptions import IOSDriverException
 from uiautodev.model import Node, WindowSize
-from uiautodev.utils.usbmux import MuxDevice, select_device
+from uiautodev.utils.usbmux import select_device
 
 
 class IOSDriver(BaseDriver):
@@ -27,6 +28,7 @@ class IOSDriver(BaseDriver):
         """ serial is the udid of the ios device """
         super().__init__(serial)
         self.device = select_device(serial)
+        self.wda = wdapy.AppiumUSBClient(self.device.serial)
     
     def _request(self, method: str, path: str, payload: Optional[dict] = None) -> bytes:
         conn = self.device.make_http_connection(port=8100)
@@ -56,29 +58,36 @@ class IOSDriver(BaseDriver):
         return self._request_json("GET", "/status")
     
     def screenshot(self, id: int = 0) -> Image.Image:
-        png_base64 = self._request_json_value("GET", "/screenshot")
-        png_data = base64.b64decode(png_base64)
-        return Image.open(io.BytesIO(png_data))
+        return self.wda.screenshot()
     
     def window_size(self):
-        return self._request_json_value("GET", "/window/size")
+        return self.wda.window_size()
     
     def dump_hierarchy(self) -> Tuple[str, Node]:
         """returns xml string and hierarchy object"""
-        xml_data = self._request_json_value("GET", "/source")
+        t = self.wda.sourcetree()
+        xml_data = t.value
         root = ElementTree.fromstring(xml_data)
         return xml_data, parse_xml_element(root, WindowSize(width=1, height=1))
     
     def tap(self, x: int, y: int):
-        self._request("POST", f"/wda/tap/0", {"x": x, "y": y})
+        self.wda.tap(x, y)
     
     def app_current(self) -> CurrentAppResponse:
-        # {'processArguments': {'env': {}, 'args': []}, 'name': '', 'pid': 32, 'bundleId': 'com.apple.springboard'}
-        value = self._request_json_value("GET", "/wda/activeAppInfo")
-        return CurrentAppResponse(package=value["bundleId"], pid=value["pid"])
+        info = self.wda.app_current()
+        return CurrentAppResponse(package=info.bundle_id, pid=info.pid)
 
     def home(self):
-        self._request("POST", "/wda/homescreen")
+        self.wda.homescreen()
+    
+    def app_switch(self):
+        raise NotImplementedError()
+
+    def volume_up(self):
+        self.wda.volume_up()
+    
+    def volume_down(self):
+        self.wda.volume_down()
         
 
 def parse_xml_element(element, wsize: WindowSize, indexes: List[int]=[0]) -> Node:

uiautodev-0.3.7/uiautodev/__init__.py

@@ -1,12 +0,0 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-
-"""Created on Mon Mar 04 2024 14:28:53 by codeskyblue
-"""
-
-from importlib.metadata import PackageNotFoundError, version
-
-try:
-    __version__ = version("uiautodev")
-except PackageNotFoundError:
-    __version__ = "0.0.0"