mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-25 10:01:10 +00:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
from lxml import etree
|
|
|
|
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.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 = []
|
|
|
|
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 dump_hierarchy(self) -> etree._Element:
|
|
"""
|
|
Returns:
|
|
etree._Element: Select elements with `self.hierarchy.xpath('//*[@text="Hermit"]')` for example.
|
|
"""
|
|
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()
|
|
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)
|