表格、恢复出厂、正则
作者:互联网
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('第四页国家码结束')
标签:info,表格,text,出厂,正则,result,print,android,id 来源: https://www.cnblogs.com/blackpink/p/14607245.html