129 lines
4.4 KiB
Python
129 lines
4.4 KiB
Python
import logging
|
||
from typing import cast
|
||
from typing_extensions import override
|
||
|
||
import cv2
|
||
import numpy as np
|
||
from cv2.typing import MatLike
|
||
from adbutils._device import AdbDevice as AdbUtilsDevice
|
||
|
||
from ..device import AndroidDevice
|
||
from ..protocol import AndroidCommandable, Touchable, Screenshotable
|
||
from ..registration import register_impl, ImplConfig
|
||
from dataclasses import dataclass
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 定义配置模型
|
||
@dataclass
|
||
class AdbImplConfig(ImplConfig):
|
||
addr: str
|
||
connect: bool = True
|
||
disconnect: bool = True
|
||
device_serial: str | None = None
|
||
timeout: float = 180
|
||
|
||
class AdbImpl(AndroidCommandable, Touchable, Screenshotable):
|
||
def __init__(self, adb_connection: AdbUtilsDevice):
|
||
self.adb = adb_connection
|
||
|
||
@override
|
||
def launch_app(self, package_name: str) -> None:
|
||
self.adb.shell(f"monkey -p {package_name} 1")
|
||
|
||
@override
|
||
def current_package(self) -> str | None:
|
||
# https://blog.csdn.net/guangdeshishe/article/details/117154406
|
||
result_text = self.adb.shell('dumpsys activity top | grep ACTIVITY | tail -n 1')
|
||
logger.debug(f"adb returned: {result_text}")
|
||
if not isinstance(result_text, str):
|
||
logger.error(f"Invalid result_text: {result_text}")
|
||
return None
|
||
result_text = result_text.strip()
|
||
if result_text == '':
|
||
logger.error("No current package found")
|
||
return None
|
||
_, activity, *_ = result_text.split(' ')
|
||
package = activity.split('/')[0]
|
||
return package
|
||
|
||
def adb_shell(self, cmd: str) -> str:
|
||
"""执行 ADB shell 命令"""
|
||
return cast(str, self.adb.shell(cmd))
|
||
|
||
@override
|
||
def detect_orientation(self):
|
||
# 判断方向：https://stackoverflow.com/questions/10040624/check-if-device-is-landscape-via-adb
|
||
# 但是上面这种方法不准确
|
||
# 因此这里直接通过截图判断方向
|
||
img = self.screenshot()
|
||
if img.shape[0] > img.shape[1]:
|
||
return 'portrait'
|
||
return 'landscape'
|
||
|
||
@property
|
||
def screen_size(self) -> tuple[int, int]:
|
||
ret = cast(str, self.adb.shell("wm size")).strip('Physical size: ')
|
||
spiltted = tuple(map(int, ret.split("x")))
|
||
# 检测当前方向
|
||
orientation = self.detect_orientation()
|
||
landscape = orientation == 'landscape'
|
||
spiltted = tuple(sorted(spiltted, reverse=landscape))
|
||
if len(spiltted) != 2:
|
||
raise ValueError(f"Invalid screen size: {ret}")
|
||
return spiltted
|
||
|
||
def screenshot(self) -> MatLike:
|
||
return cv2.cvtColor(np.array(self.adb.screenshot()), cv2.COLOR_RGB2BGR)
|
||
|
||
def click(self, x: int, y: int) -> None:
|
||
self.adb.shell(f"input tap {x} {y}")
|
||
|
||
def swipe(self, x1: int, y1: int, x2: int, y2: int, duration: float | None = None) -> None:
|
||
if duration is not None:
|
||
logger.warning("Swipe duration is not supported with AdbDevice. Ignoring duration.")
|
||
self.adb.shell(f"input touchscreen swipe {x1} {y1} {x2} {y2}")
|
||
|
||
|
||
def _create_adb_device_base(config: AdbImplConfig, impl_class: type) -> AndroidDevice:
|
||
"""
|
||
通用的 ADB 设备创建工厂函数。
|
||
其他任意基于 ADB 的 Impl 可以直接复用这个函数。
|
||
|
||
:param config: ADB 实现配置
|
||
:param impl_class: 实现类或工厂函数。构造函数接收 adb_connection 参数。
|
||
"""
|
||
from adbutils import adb
|
||
|
||
if config.disconnect:
|
||
logger.debug('adb disconnect %s', config.addr)
|
||
adb.disconnect(config.addr)
|
||
if config.connect:
|
||
logger.debug('adb connect %s', config.addr)
|
||
result = adb.connect(config.addr)
|
||
if 'cannot connect to' in result:
|
||
raise ValueError(result)
|
||
serial = config.device_serial or config.addr
|
||
logger.debug('adb wait for %s', serial)
|
||
adb.wait_for(serial, timeout=config.timeout)
|
||
devices = adb.device_list()
|
||
logger.debug('adb device_list: %s', devices)
|
||
d = [d for d in devices if d.serial == serial]
|
||
if len(d) == 0:
|
||
raise ValueError(f"Device {config.addr} not found")
|
||
d = d[0]
|
||
|
||
device = AndroidDevice(d)
|
||
impl = impl_class(d)
|
||
device._touch = impl
|
||
device._screenshot = impl
|
||
device.commands = impl
|
||
|
||
return device
|
||
|
||
|
||
@register_impl('adb', config_model=AdbImplConfig)
|
||
def create_adb_device(config: AdbImplConfig) -> AndroidDevice:
|
||
"""AdbImpl 工厂函数"""
|
||
return _create_adb_device_base(config, AdbImpl)
|