uiautodev 0.4.0__tar.gz → 0.5.0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: uiautodev
-Version: 0.4.0
+Version: 0.5.0
 Summary: Mobile UI Automation, include UI hierarchy inspector, script recorder
 Home-page: https://uiauto.dev
 License: MIT
@@ -14,13 +14,10 @@ Classifier: Programming Language :: Python :: 3.9
 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.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
 Requires-Dist: fastapi (>=0.111.0,<0.112.0)
-Requires-Dist: httpretty (>=1.1.4,<2.0.0)
 Requires-Dist: httpx
 Requires-Dist: lxml
 Requires-Dist: pillow
@@ -38,6 +35,8 @@ Description-Content-Type: text/markdown
 
 https://uiauto.dev
 
+> backup site: https://uiauto.devsleep.com
+
 UI Inspector for Android and iOS, help inspector element properties, and auto generate XPath, script.
 
 # Install

{uiautodev-0.4.0 → uiautodev-0.5.0}/README.md

@@ -4,6 +4,8 @@
 
 https://uiauto.dev
 
+> backup site: https://uiauto.devsleep.com
+
 UI Inspector for Android and iOS, help inspector element properties, and auto generate XPath, script.
 
 # Install

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "uiautodev"
-version = "0.4.0"
+version = "0.5.0"
 description = "Mobile UI Automation, include UI hierarchy inspector, script recorder"
 homepage = "https://uiauto.dev"
 authors = ["codeskyblue <codeskyblue@gmail.com>"]
@@ -15,8 +15,6 @@ construct = "*"
 lxml = "*"
 click = "^8.1.7"
 pygments = ">=2"
-httpretty = {version = "^1.1.4", optional = true}
-appium-python-client = {version = "^4.0.0", optional = true}
 uiautomator2 = ">=2"
 httpx = "*"
 fastapi = "^0.111.0"
@@ -25,8 +23,8 @@ poetry = "^1.8.2"
 pydantic = "^2.6"
 wdapy = "^0.2.2"
 
-[tool.poetry.extras]
-appium = ["appium-python-client", "httppretty"]
+#[tool.poetry.extras]
+#appium = ["appium-python-client", "httppretty"]
 
 [tool.poetry.scripts]
 "uiauto.dev" = "uiautodev.__main__:main"

{uiautodev-0.4.0 → uiautodev-0.5.0}/uiautodev/__init__.py

@@ -5,4 +5,4 @@
 """
 
 # version is auto managed by poetry
-__version__ = "0.4.0"
+__version__ = "0.5.0"

{uiautodev-0.4.0 → uiautodev-0.5.0}/uiautodev/cli.py

@@ -140,7 +140,7 @@ def self_update():
 @click.option("-f", "--force", is_flag=True, default=False, help="shutdown alrealy runningserver")
 @click.option("-s", "--no-browser", is_flag=True, default=False, help="silent mode, do not open browser")
 def server(port: int, host: str, reload: bool, force: bool, no_browser: bool):
-    logger.info("version: %s", __version__)
+    print("uiautodev version:", __version__)
     if force:
         try:
             httpx.get(f"http://{host}:{port}/shutdown", timeout=3)

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

@@ -8,7 +8,7 @@ from __future__ import annotations
 
 import time
 import typing
-from typing import Callable, Dict, Optional
+from typing import Callable, Dict, List, Optional, Union
 
 from pydantic import BaseModel
 
@@ -17,7 +17,7 @@ from uiautodev.command_types import AppLaunchRequest, AppTerminateRequest, By, C
     WindowSizeResponse
 from uiautodev.driver.base_driver import BaseDriver
 from uiautodev.exceptions import ElementNotFoundError
-from uiautodev.model import Node
+from uiautodev.model import Node, AppInfo
 from uiautodev.utils.common import node_travel
 
 COMMANDS: Dict[Command, Callable] = {}
@@ -39,17 +39,21 @@ def get_command_params_type(command: Command) -> Optional[BaseModel]:
     return type_hints.get("params")
 
 
-def send_command(driver: BaseDriver, command: Command, params=None):
+def send_command(driver: BaseDriver, command: Union[str, Command], params=None):
     if command not in COMMANDS:
         raise NotImplementedError(f"command {command} not implemented")
     func = COMMANDS[command]
-    type_hints = typing.get_type_hints(func)
-    if type_hints.get("params"):
+    params_model = get_command_params_type(command)
+    if params_model:
         if params is None:
             raise ValueError(f"params is required for {command}")
-        if not isinstance(params, type_hints["params"]):
-            raise TypeError(f"params should be {type_hints['params']}")
-    if params is None:
+        if isinstance(params, dict):
+            params = params_model.model_validate(params)
+        elif isinstance(params, params_model):
+            pass
+        else:
+            raise TypeError(f"params should be {params_model}", params)
+    if not params:
         return func(driver)
     return func(driver, params)
 
@@ -177,3 +181,10 @@ def click_element(driver: BaseDriver, params: FindElementRequest):
     center_x = (node.bounds[0] + node.bounds[2]) / 2
     center_y = (node.bounds[1] + node.bounds[3]) / 2
     tap(driver, TapRequest(x=center_x, y=center_y, isPercent=True))
+
+
+@register(Command.APP_LIST)
+def app_list(driver: BaseDriver) -> List[AppInfo]:
+    # added in v0.5.0
+    return driver.app_list()
+

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

@@ -22,6 +22,7 @@ class Command(str, enum.Enum):
     APP_CURRENT = "currentApp"
     APP_LAUNCH = "appLaunch"
     APP_TERMINATE = "appTerminate"
+    APP_LIST = "appList"
 
     GET_WINDOW_SIZE = "getWindowSize"
     HOME = "home"

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

@@ -4,12 +4,11 @@
 """Created on Fri Mar 01 2024 14:19:29 by codeskyblue
 """
 
-import json
 import logging
 import re
 import time
 from functools import cached_property, partial
-from typing import List, Optional, Tuple
+from typing import Iterator, List, Optional, Tuple
 from xml.etree import ElementTree
 
 import adbutils
@@ -18,9 +17,8 @@ from PIL import Image
 
 from uiautodev.command_types import CurrentAppResponse
 from uiautodev.driver.base_driver import BaseDriver
-from uiautodev.driver.udt.udt import UDT, UDTError
 from uiautodev.exceptions import AndroidDriverException, RequestError
-from uiautodev.model import Node, Rect, ShellResponse, WindowSize
+from uiautodev.model import Node, AppInfo, Rect, ShellResponse, WindowSize
 from uiautodev.utils.common import fetch_through_socket
 
 logger = logging.getLogger(__name__)
@@ -29,28 +27,15 @@ class AndroidDriver(BaseDriver):
     def __init__(self, serial: str):
         super().__init__(serial)
         self.adb_device = adbutils.device(serial)
-        self._try_dump_list = [
-            self._get_u2_hierarchy,
-            self._get_udt_dump_hierarchy,
-            # self._get_appium_hierarchy,
-        ]
-    
-    @cached_property
-    def udt(self) -> UDT:    
-        return UDT(self.adb_device)
 
     @cached_property
     def ud(self) -> u2.Device:
         return u2.connect_usb(self.serial)
     
     def screenshot(self, id: int) -> Image.Image:
-        try:
-            return self.adb_device.screenshot() # display_id is not OK now
-        except adbutils.AdbError as e:
-            logger.warning("screenshot error: %s", str(e))
-            if id > 0:
-                raise AndroidDriverException("multi-display is not supported yet for uiautomator2")
-            return self.ud.screenshot()
+        if id > 0:
+            raise AndroidDriverException("multi-display is not supported yet for uiautomator2")
+        return self.ud.screenshot()
 
     def shell(self, command: str) -> ShellResponse:
         try:
@@ -82,38 +67,12 @@ class AndroidDriver(BaseDriver):
 
         uiautomator dump errors:
         - ERROR: could not get idle state.
-
         """
-        for dump_func in self._try_dump_list[:]:
-            try:
-                logger.debug(f"try to dump with %s", dump_func.__name__)
-                result = dump_func()
-                logger.debug("dump success")
-                self._try_dump_list.remove(dump_func)
-                self._try_dump_list.insert(0, dump_func)
-                return result
-            except Exception as e:
-                logger.exception("unexpected dump error: %s", e)
-        raise AndroidDriverException("Failed to dump hierarchy")
-    
-    def _get_u2_hierarchy(self) -> str:
-        d = u2.connect_usb(self.serial)
-        return d.dump_hierarchy()
-
-    def _get_appium_hierarchy(self) -> str:
-        c = self.adb_device.create_connection(adbutils.Network.TCP, 6790)
         try:
-            content = fetch_through_socket(c, "/wd/hub/session/0/source", timeout=10)
-            return json.loads(content)["value"]
-        except (adbutils.AdbError, RequestError) as e:
-            raise AndroidDriverException(
-                f"Failed to get hierarchy from appium server: {str(e)}"
-            )
-        finally:
-            c.close()
-
-    def _get_udt_dump_hierarchy(self) -> str:
-        return self.udt.dump_hierarchy()
+            return self.ud.dump_hierarchy()
+        except Exception as e:
+            logger.exception("unexpected dump error: %s", e)
+            raise AndroidDriverException("Failed to dump hierarchy")
     
     def tap(self, x: int, y: int):
         self.adb_device.click(x, y)
@@ -159,6 +118,22 @@ class AndroidDriver(BaseDriver):
     
     def volume_mute(self):
         self.adb_device.keyevent("VOLUME_MUTE")
+    
+    def app_list(self) -> List[AppInfo]:
+        results = []
+        output = self.adb_device.shell(["pm", "list", "packages", '-3'])
+        for m in re.finditer(r"^package:([^\s]+)\r?$", output, re.M):
+            packageName = m.group(1)
+            results.append(AppInfo(packageName=packageName))
+        return results
+
+    def open_app_file(self, package: str) -> Iterator[bytes]:
+        line = self.adb_device.shell(f"pm path {package}")
+        if not line.startswith("package:"):
+            raise AndroidDriverException(f"Failed to get package path: {line}")
+        remote_path = line.split(':', 1)[1]
+        yield from self.adb_device.sync.iter_content(remote_path)
+
 
 
 def parse_xml(xml_data: str, wsize: WindowSize, display_id: Optional[int] = None) -> Node:

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

@@ -4,14 +4,14 @@
 """Created on Fri Mar 01 2024 14:18:30 by codeskyblue
 """
 import abc
-import enum
-from typing import Tuple
+from io import FileIO
+from typing import Iterator, List, Tuple
 
 from PIL import Image
 from pydantic import BaseModel
 
 from uiautodev.command_types import CurrentAppResponse
-from uiautodev.model import Node, ShellResponse, WindowSize
+from uiautodev.model import Node, AppInfo, ShellResponse, WindowSize
 
 
 class BaseDriver(abc.ABC):
@@ -93,4 +93,13 @@ class BaseDriver(abc.ABC):
 
     def wake_up(self):
         """ wake up the device """
-        raise NotImplementedError()
+        raise NotImplementedError()
+    
+    def app_list(self) -> List[AppInfo]:
+        """ list installed packages """
+        raise NotImplementedError()
+    
+    def open_app_file(self, package: str) -> Iterator[bytes]:
+        """ open app file """
+        raise NotImplementedError()
+        

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

@@ -5,10 +5,7 @@
 """
 
 
-import base64
-import io
 import json
-import re
 from functools import partial
 from typing import List, Optional, Tuple
 from xml.etree import ElementTree

{uiautodev-0.4.0 → uiautodev-0.5.0}/uiautodev/model.py

@@ -42,4 +42,8 @@ class Node(BaseModel):
 
 class WindowSize(typing.NamedTuple):
     width: int
-    height: int
+    height: int
+
+
+class AppInfo(BaseModel):
+    packageName: str

{uiautodev-0.4.0 → uiautodev-0.5.0}/uiautodev/router/device.py

@@ -6,9 +6,10 @@
 
 import io
 import logging
-from typing import Any, List
+from typing import Any, Dict, List
 
 from fastapi import APIRouter, Response
+from fastapi.responses import StreamingResponse
 from pydantic import BaseModel
 
 from uiautodev import command_proxy
@@ -102,14 +103,25 @@ def make_router(provider: BaseProvider) -> APIRouter:
         return command_proxy.app_current(driver)
 
     @router.post('/{serial}/command/{command}')
-    def _command_proxy_other(serial: str, command: Command, params: Any = None):
+    def _command_proxy_other(serial: str, command: Command, params: Dict[str, Any] = None):
         """Run a command on the device"""
         driver = provider.get_device_driver(serial)
-        func = command_proxy.COMMANDS[command]
-        if params is None:
-            response = func(driver)
-        else:
-            response = func(driver, params)
+        response = command_proxy.send_command(driver, command, params)
         return response
+    
+    @router.get('/{serial}/backupApp')
+    def _backup_app(serial: str, packageName: str):
+        """Backup app
+        
+        Added in 0.5.0
+        """
+        driver = provider.get_device_driver(serial)
+        file_name = f"{packageName}.apk"
+        headers = {
+            'Content-Disposition': f'attachment; filename="{file_name}"'
+        }
+        return StreamingResponse(driver.open_app_file(packageName), headers=headers)
+        
+
 
     return router