StarRailCopilot/module/device/app_control.py

93 lines
3.1 KiB
Python

from lxml import etree
from module.base.timer import Timer
from module.device.method.adb import Adb
from module.device.method.uiautomator_2 import Uiautomator2
from module.device.method.utils import HierarchyButton
from module.device.method.wsa import WSA
from module.exception import ScriptError
from module.logger import logger
class AppControl(Adb, WSA, Uiautomator2):
hierarchy: etree._Element
# Use ADB for all
# See https://github.com/openatx/uiautomator2/issues/565
_app_u2_family = []
_hierarchy_interval = Timer(0.1)
def app_is_running(self) -> bool:
method = self.config.Emulator_ControlMethod
if self.is_wsa:
package = self.app_current_wsa()
elif method in AppControl._app_u2_family:
package = self.app_current_uiautomator2()
else:
package = self.app_current_adb()
package = package.strip(' \t\r\n')
logger.attr('Package_name', package)
return package == self.package
def app_start(self):
method = self.config.Emulator_ControlMethod
logger.info(f'App start: {self.package}')
if self.config.Emulator_Serial == 'wsa-0':
self.app_start_wsa(display=0)
elif method in AppControl._app_u2_family:
self.app_start_uiautomator2()
else:
self.app_start_adb()
def app_stop(self):
method = self.config.Emulator_ControlMethod
logger.info(f'App stop: {self.package}')
if method in AppControl._app_u2_family:
self.app_stop_uiautomator2()
else:
self.app_stop_adb()
def hierarchy_timer_set(self, interval=None):
if interval is None:
interval = 0.1
elif isinstance(interval, (int, float)):
# No limitation for manual set in code
pass
else:
logger.warning(f'Unknown hierarchy interval: {interval}')
raise ScriptError(f'Unknown hierarchy interval: {interval}')
if interval != self._hierarchy_interval.limit:
logger.info(f'Hierarchy interval set to {interval}s')
self._hierarchy_interval.limit = interval
def dump_hierarchy(self) -> etree._Element:
"""
Returns:
etree._Element: Select elements with `self.hierarchy.xpath('//*[@text="Hermit"]')` for example.
"""
self._hierarchy_interval.wait()
self._hierarchy_interval.reset()
# method = self.config.Emulator_ControlMethod
# if method in AppControl._app_u2_family:
# self.hierarchy = self.dump_hierarchy_uiautomator2()
# else:
# self.hierarchy = self.dump_hierarchy_adb()
# Using uiautomator2
self.hierarchy = self.dump_hierarchy_uiautomator2()
return self.hierarchy
def xpath_to_button(self, xpath: str) -> HierarchyButton:
"""
Args:
xpath (str):
Returns:
HierarchyButton:
An object with methods and properties similar to Button.
If element not found or multiple elements were found, return None.
"""
return HierarchyButton(self.hierarchy, xpath)