def plug(self):
"""Reset the power status of the device"""
if self.get_api_level() < 23:
# API level < 23, 4.4.3+ tested, WARNING: hardcoding
# reset only restarts auto-update
Adb.shell(self.id, 'dumpsys battery set usb 1')
# API level 23+ (Android 6.0+)
Adb.shell(self.id, 'dumpsys battery reset')
def __init__(self, names, adb_path='adb'):
Adb.setup(adb_path)
mapping_file = load_json(op.join(ROOT_DIR, 'devices.json'))
self._device_map = {n: mapping_file.get(n, None) for n in names}
for name, device_id in self._device_map.items():
if not device_id:
raise ConfigError(name)
self.devices = [Device(name, device_id) for name, device_id in self._device_map.items()]
def __init__(self, name, device_id, settings):
self.logger = logging.getLogger(self.__class__.__name__)
self.name = name
self.id = device_id
self.root_unplug = settings.get('root_disable_charging', False)
self.root_unplug_value = settings.get('charging_disabled_value', None)
self.root_unplug_file = settings.get('usb_charging_disabled_file',
None)
self.root_plug_value = None
Adb.connect(device_id)
def unplug(self):
"""Fakes the device to think it is unplugged, so the Doze mode can be activated"""
if self.get_api_level() < 23:
# API level < 23, 4.4.3+ tested, WARNING: hardcoding
Adb.shell(self.id, 'dumpsys battery set usb 0')
# Adb.shell(self.id, 'dumpsys battery set ac 0')
# Adb.shell(self.id, 'dumpsys battery set wireless 0')
else:
# API level 23+ (Android 6.0+)
Adb.shell(self.id, 'dumpsys battery unplug')
def __init__(self, devices, adb_path='adb', devices_spec=None):
if devices_spec is None:
devices_spec = op.join(ROOT_DIR, 'devices.json')
Adb.setup(adb_path)
mapping_file = load_json(devices_spec)
self._device_map = {n: mapping_file.get(n, None) for n in devices}
for name, device_id in self._device_map.items():
if not device_id:
raise ConfigError(name)
self.devices = [
Device(name, device_id, devices[name])
for name, device_id in self._device_map.items()
]
def su_unplug(self, restart):
"""Root unplugs the device"""
self.root_plug_value = Adb.shell_su(self.id,
'cat %s' % self.root_unplug_file)
if 'su: not found' in self.root_plug_value:
raise AdbError("%s %s: is not rooted" % (self.id, self.name))
if 'No such file or directory' in self.root_plug_value:
raise ConfigError(
'%s %s: the root unplug file seems to be invalid' %
(self.id, self.name))
if restart:
self.check_plug_value()
Adb.shell_su(
self.id,
'echo %s > %s' % (self.root_unplug_value, self.root_unplug_file))
def unplug(self, restart):
"""Makes the device to think it is unplugged, so the Doze mode can be activated"""
if self.root_unplug:
self.su_unplug(restart)
self.logger.info('Root unpluged')
else:
self.logger.info('Default unplug')
if self.get_api_level() < 23:
# API level < 23, 4.4.3+ tested, WARNING: hardcoding
Adb.shell(self.id, 'dumpsys battery set usb 0')
# Adb.shell(self.id, 'dumpsys battery set ac 0')
# Adb.shell(self.id, 'dumpsys battery set wireless 0')
else:
# API level 23+ (Android 6.0+)
Adb.shell(self.id, 'dumpsys battery unplug')
def current_activity(self):
"""Returns the current focused activity on the system"""
# https://github.com/aldonin/appium-adb/blob/7b4ed3e7e2b384333bb85f8a2952a3083873a90e/lib/adb.js#L1278
windows = Adb.shell(self.id, 'dumpsys window windows')
null_re = r'mFocusedApp=null'
# https://regex101.com/r/xZ8vF7/1
current_focus_re = r'mCurrentFocus.+\s([^\s\/\}]+)\/[^\s\/\}]+(\.[^\s\/\}]+)}'
focused_app_re = r'mFocusedApp.+Record\{.*\s([^\s\/\}]+)\/([^\s\/\}\,]+)(\s[^\s\/\}]+)*\}'
match = None
found_null = False
for line in windows.split('\n'):
current_focus = re.search(current_focus_re, line)
focused_app = re.search(focused_app_re, line)
if current_focus:
match = current_focus
elif focused_app and match is None:
match = focused_app
elif re.search(null_re, line):
found_null = True
if match:
result = match.group(1).strip()
self.logger.debug('Current activity: %s' % result)
return result
elif found_null:
self.logger.debug('Current activity: null')
return None
else:
self.logger.error('Results from dumpsys window windows: \n%s' %
windows)
raise AdbError('Could not parse activity from dumpsys')
def logcat_to_file(self, path):
"""Dumps the last x lines of logcat into a file specified by path"""
makedirs(path)
with open(
op.join(
path,
'%s_%s.txt' % (self.id, time.strftime('%Y.%m.%d_%H%M%S'))),
'w+') as f:
f.write(Adb.logcat(self.id))
def launch_activity(self, package, activity, action='', data_uri='', from_scratch=False, force_stop=False):
"""Launches an activity using 'am start', returns instantly"""
# https://developer.android.com/studio/command-line/adb.html#am
# https://developer.android.com/studio/command-line/adb.html#IntentSpec
# https://stackoverflow.com/a/3229077
cmd = 'am start'
if force_stop:
cmd += ' -S'
if action:
cmd += ' -a %s' % action
cmd += ' -n %s/%s' % (package, activity)
if data_uri:
cmd += ' -d %s' % data_uri
# https://android.stackexchange.com/a/113919
if from_scratch:
cmd += ' --activity-clear-task'
return Adb.shell(self.id, cmd)
def shell(self, cmd):
"""Runs the device shell with command specified by cmd"""
return Adb.shell(self.id, cmd)
def clear_app_data(self, name):
"""Clears the data of an app by package name"""
Adb.clear_app_data(self.id, name)
def uninstall(self, name):
"""Uninstalls the package on the device"""
Adb.uninstall(self.id, name)
def install(self, apk):
"""Check if the file exists, and then install the package"""
if not op.isfile(apk):
raise AdbError("%s is not found" % apk)
Adb.install(self.id, apk)
def get_app_list(self):
"""Returns a list of installed packages on the system"""
return Adb.list_apps(self.id)
def get_api_level(self):
"""Returns the Android API level as a number"""
return Adb.shell(self.id, 'getprop ro.build.version.sdk')
def __init__(self, name, device_id):
self.logger = logging.getLogger(self.__class__.__name__)
self.name = name
self.id = device_id
self.iterator = 0
Adb.connect(device_id)
def push(self, local, remote):
"""Pushes a file from the computer to the device"""
return Adb.push(self.id, local, remote)
def logcat_regex(self, regex):
return Adb.logcat(self.id, regex=regex)
def clear_app_data(self, name):
"""Clears the data of an app by package name"""
self.iterator = self.iterator + 1
if (self.iterator % 2 == 0):
Adb.clear_app_data(self.id, name)
def force_stop(self, name):
"""Force stop an app by package name"""
Adb.shell(self.id, 'am force-stop %s' % name)
def launch_package(self, package):
"""Launches a package by name without activity, returns instantly"""
# https://stackoverflow.com/a/25398877
result = Adb.shell(self.id, 'monkey -p {} 1'.format(package))
if 'monkey aborted' in result:
raise AdbError('Could not launch "{}"'.format(package))
def get_version(self):
"""Returns the Android version"""
return Adb.shell(self.id, 'getprop ro.build.version.release')
def pull(self, remote, local):
"""Pulls a file from the device to the computer"""
return Adb.pull(self.id, remote, local)
def su_plug(self):
"""Reset the power status of the device if root unpluged"""
self.logger.info('Root pluged, please check if device is charging')
Adb.shell_su(
self.id,
'echo %s > %s' % (self.root_plug_value, self.root_unplug_file))