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(\'第四页国家码结束\')