import os
import re
import subprocess
import sys
import uiautomator2 as u2
import time
from pandas.tests.io.excel.test_xlrd import xlrd, xlwt
from xlutils.copy import copy
def set_up():
d = u2.connect(device)
time.sleep(1)
d.screen_on()
d.shell('input keyevent 82')
d.shell('input keyevent 3')
def exists(path):
is_exists = os.path.exists(path)
if is_exists:
return True
else:
os.mkdir(path)
# 创建结果文件夹
def create_xls(xls_name):
# 结果文件夹
exists(os.path.join(os.getcwd(), 'Result'))
result_dir = os.path.join(os.getcwd(), 'Result')
# print(result_dir)
# 判断目标结果excel文件是否存在，若无创建新文件
result_path = os.path.join(result_dir, xls_name)
if not os.path.exists(result_path):
# 创建xls表格
workbook = xlwt.Workbook(encoding='utf-8')
# 创建一个worksheet
worksheet = workbook.add_sheet('result')
worksheet.col(0).width = 40 * 300 # Set the column width
worksheet.col(1).width = 40 * 200 # Set the column width
worksheet.col(2).width = 40 * 200 # Set the column width
worksheet.col(3).width = 40 * 200 # Set the column width
# 写入excel
# 参数对应 行, 列, 值
worksheet.write(0, 0, 'country')
worksheet.write(0, 1, 'camera_info')
worksheet.write(0, 2, 'storage_used')
worksheet.write(0, 3, 'result')
# worksheet.write(0, 3, 'rate')
workbook.save(result_path)
else:
pass
return result_path
def write_excel_xls_append(path, country=[], camera_info=[], storage_used=[], result=[]):
global worksheet
workbook = xlrd.open_workbook(path, formatting_info=True)
try:
worksheet = workbook.sheet_by_name('result')
except:
print('xls文件sheet名称错误，默认名称是result,请检查')
# 在写入之前需要获取已经写入的行数
old_nrows = worksheet.nrows
# print(old_nrows)
new_workbook = copy(workbook) # 将xlrd对象拷贝转化为xlwt对象
new_worksheet = new_workbook.get_sheet(0) # 获取转化后工作簿中的第一个表格
# 开始进行写操作
new_worksheet.write(old_nrows + 1, 0, country)
new_worksheet.write(old_nrows + 1, 1, camera_info)
new_worksheet.write(old_nrows + 1, 2, storage_used)
new_worksheet.write(old_nrows + 1, 3, result)
new_workbook.save(path)
# 到设置里面的平板信息
def open_to_the_camera():
for i in range(5):
d.app_start('com.android.settings')
time.sleep(1)
if d(resourceId='android:id/button2').wait(timeout=3):
d(resourceId='android:id/button2').click()
time.sleep(4)
d(scrollable=True).fling.toEnd()
time.sleep(1)
d.dump_hierarchy()
try:
d.xpath(
'//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'9]/android.widget.ImageView[1]').click()
time.sleep(1)
d.xpath(
'//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'1]/android.widget.LinearLayout[1]/android.widget.ImageView[1]').click()
time.sleep(4)
d(scrollable=True).fling.toEnd()
time.sleep(1)
break
except:
print('未能找到设置里面System的xpath,再次执行该操作')
continue
# 到设置里面的存储界面
def open_to_the_storage():
os.system('adb -s %s shell input keyevent 82' % device)
os.system('adb -s %s shell input keyevent 82' % device)
for i in range(5):
try:
d.app_start('com.android.settings')
time.sleep(4)
d(scrollable=True).fling.toEnd()
time.sleep(1)
d.dump_hierarchy()
d.xpath(
'//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'7]/android.widget.ImageView[1]').click()
time.sleep(2)
break
except:
print('未能找到设置里面Storage的xpath,再次执行该操作')
continue
# 获取camera里面的信息并判断
def camera_text():
d.dump_hierarchy()
for i in range(5):
for j in range(5):
if not d.xpath('//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'9]/android.widget.RelativeLayout[1]/android.widget.TextView[2]').exists:
time.sleep(2)
else:
break
try:
for i in d.xpath(
'//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'9]/android.widget.RelativeLayout[1]/android.widget.TextView[2]').all():
text = i.text
print(text)
# 正则表达式 排除里面没有特殊字符+- 只有8和5
result_1 = re.findall('[\+,\-,=,\*,#,@,!,\$,%,\^,~,\(,\),`]', text)
if len(result_1) != 0:
print('有特殊字符')
result = 'False'
return text, result
else:
result = re.findall(
'([\+,\-,=,\*,#,@,!,\$,%,\^,~,\(,\),\.`,\/]*\d+[\+,\-,=,\*,#,@,!,\$,%,\^,~,\(,\),\.`,\/]*)+',
text)
print('Camera_text', result)
if len(result) != 2:
print("数字不是默认的2个数字")
result = 'False'
return text, result
for i in range(len(result)):
if '8' != result[i] and '13' != result[i]:
# print('Camera_text', result)
print("数字不是默认值")
result = 'False'
return text, result
else:
result = "True"
return text, result
except:
print('读取Camera信息失败,再次操作')
continue
# 获取stroage里面的信息并判断
def storage_text():
for i in range(5):
try:
d.dump_hierarchy()
for i in range(10):
if d.xpath('//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'1]/android.widget.LinearLayout[1]/android.widget.TextView[1]').exists:
break
else:
time.sleep(2)
for i in d.xpath(
'//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'1]/android.widget.LinearLayout[1]/android.widget.TextView[1]').all():
text_use = i.text
print('已经使用', text_use)
for i in d.xpath(
'//*[@resource-id="com.android.settings:id/recycler_view"]/android.widget.LinearLayout['
'1]/android.widget.LinearLayout[1]/android.widget.TextView[2]').all():
text = i.text
print('内存', text)
shuzi = re.findall('\d+', str(text))
# print(shuzi)
result = ""
if "64" in shuzi:
info = d.xpath(
'//*[@resource-id="com.android.settings:id/donut"]').info
result_use = info["contentDescription"]
print(result_use)
# print(result_use)
if '31' not in result_use:
print('存储空间使用不是默认的值31%')
result = "False"
else:
result = "True"
elif "128" in shuzi:
info = d.xpath(
'//*[@resource-id="com.android.settings:id/donut"]').info
# print(info)
result_use = info["contentDescription"]
print(result_use)
if '16' not in result_use:
print('存储空间使用不是默认的值16%')
result = "False"
else:
result = "True"
else:
info = d.xpath(
'//*[@resource-id="com.android.settings:id/donut"]').info
info = dict(info)
result_use = info["contentDescription"]
print(result_use)
# 阿拉伯语无法数字判断 默认都false
# result_use = re.findall('\w[\%]+', str(info))
# result_use = "".join(result_use).encode("gbk", 'ignore').decode("gbk", "ignore")
result = "False"
# print('阿拉博语言存储空间：', info["contentDescription"])
# if '16%' and '31%' not in result_use:
# print(result_use, '存储空间使用不是默认的值16%/31%')
result_use = ''.join(str(s) for s in result_use)
return result_use, result
except:
print('读取storage信息失败')
continue
# 进入工厂模式
def enter_factory():
d = u2.connect(device)
os.system('adb -s %s shell input keyevent 82' % device)
os.system('adb -s %s shell input keyevent 82' % device)
for i in range(5):
try:
d.app_start('com.android.settings')
time.sleep(4)
d.xpath(
'//*[@resource-id="android:id/action_bar"]/android.widget.LinearLayout[1]/android.widget.TextView').click()
time.sleep(2)
d.send_keys('####6030#')
for i in range(5):
if d(text='AE United Arab Emirates').exists:
time.sleep(1)
break
else:
time.sleep(2)
break
except:
print('进入工厂模式失败')
continue
# 切换国家码
def change_code_list(start, end):
d.dump_hierarchy()
if d(className='android.widget.RadioButton', index=start).exists and d(className='android.widget.RadioButton',
index=end).exists:
d(className='android.widget.RadioButton', index=start).click()
country = d(className='android.widget.RadioButton', index=start).info["text"]
time.sleep(1)
if d(resourceId='com.lenovo.EngineeringCode:id/ok').exists:
d(resourceId='com.lenovo.EngineeringCode:id/ok').click()
time.sleep(1)
if d(resourceId='android:id/button1').wait(timeout=2):
d(resourceId='android:id/button1').click()
time.sleep(260)
for i in range(5):
try:
# 等待恢复出厂设置
os.system("adb -s %s root" % device)
os.system("adb -s %s remount" % device)
os.system("adb -s %s shell settings put secure user_setup_complete 1" % device)
os.system("adb -s %s shell settings put global device_provisioned 1" % device)
os.system("adb -s %s reboot" % device)
time.sleep(85) # 重启
os.system('adb -s %s shell input keyevent 82' % device)
os.system('adb -s %s shell input keyevent 82' % device)
os.system('adb -s %s shell input keyevent 82' % device)
os.system('adb -s %s shell input keyevent 82' % device)
os.system('adb -s %s shell settings put system screen_off_timeout 600000' % device)
os.system('adb -s %s shell settings put system screen_off_timeout 600000' % device)
os.system('adb -s %s shell input keyevent 3' % device)
os.system('adb -s %s shell input keyevent 3' % device)
set_up()
break
except:
print('开机失败重新再次尝试')
continue
print(country)
main()
try:
write_excel_xls_append(path=result_path, country=country, camera_info=camera_info,
storage_used=storage_used, result=result)
except:
print('请勿在执行过程中打开结果文件')
def main():
global camera_info, storage_used, result
open_to_the_camera()
time.sleep(2)
result_1 = camera_text()
time.sleep(2)
open_to_the_storage()
time.sleep(4)
result_2 = storage_text()
result_3 = result_1 + result_2
# print(result_3)
# print(result_3)
# 结果默认为True 有一个错误即为错
result = "True"
for i in range(len(result_3)):
if "False" in result_3[i]:
result = "False"
elif "/" in result_3[i]:
camera_info = result_3[i]
elif "%" in result_3[i]:
storage_used = result_3[i]
# 排除没有百分比的特殊情况
if "%" not in result_2:
storage_used = result_2
# print(storage_used)
for m in range(len(storage_used)):
if storage_used[m] != 'True' and storage_used[m] != 'False':
storage_used = storage_used[m]
break
# 排除没有/ 的情况
if "/" not in result_1:
camera_info = result_1
# print(camera_info)
for m in range(len(camera_info)):
if camera_info[m] != 'True' and camera_info[m] != 'False':
camera_info = camera_info[m]
# print(camera_info)
break
return camera_info, storage_used, result
# 选择设备的id
def device_input():
device_id = input('请输入需要进行操作的设备id:')
return device_id
if __name__ == '__main__':
result_path = create_xls("CheckInDifferentCountry.xls")
device = device_input()
d = u2.connect(device)
time.sleep(1)
code_1 = 1
code_2 = 15
code_3 = 30
code_4 = 45
code_5 = 59
# AR -> EC Ecuador
for i in range(14):
enter_factory()
while True:
if d(text='AE United Arab Emirates').exists and d(text='EC Ecuador').exists:
if code_1 < 15:
change_code_list(code_1, 15)
code_1 += 1
break
else:
os.system('adb -s %s shell input swipe 798 327 781 248' % device)
print('第一页国家码结束')
# EE -> IT Italy
for i in range(14):
enter_factory()
while True:
if d(text='EC Ecuador').exists and d(text='IT Italy').exists:
if code_2 < 30:
change_code_list(code_2, 30)
code_2 += 1
break
else:
os.system('adb -s %s shell input swipe 798 327 781 248' % device)
print('第二页国家码结束')
# Japan -> PT Portugal
for i in range(14):
enter_factory()
while True:
if d(text='IT Italy').exists and d(text='PT Portugal').exists:
if code_3 < 45:
change_code_list(code_3, 45)
code_3 += 1
break
else:
os.system('adb -s %s shell input swipe 798 327 781 248' % device)
print('第三页国家码结束')
# RO Romanial-> US United States of America
for i in range(14):
enter_factory()
while True: # ZA South Africa
if d(text='PT Portugal').exists and d(text='US United States of America').exists:
if code_4 < 60:
change_code_list(code_4, 60)
code_4 += 1
break
else:
os.system('adb -s %s shell input swipe 798 327 781 248' % device)
# 点击最后二个国家码
for i in range(2):
enter_factory()
while True:
if d(text='ZA South Africa').exists and d(text='VN Vietnam').exists:
change_code_list(code_5, 61)
code_5 += 1
break
else:
os.system('adb -s %s shell input swipe 798 327 781 248' % device)
# 点击第一个阿拉伯
enter_factory()
if d(text='AE United Arab Emirates').exists:
change_code_list(0, 60)
print('第四页国家码结束')
