python游戏测试工具自动化遍历游戏中所有关卡

from testcase.tools import *
from testcase.ExcelTool import ExcelReader,ExcelWriter
from testcase.Page import Page
class AutoBattle:
    def __init__(self,level_excel_path):
        self.start_time=0
        self.levelList=ExcelReader(level_excel_path).read_first_sheet_first_col()
        self.excelWriter =  ExcelWriter()
        print(self.levelList)
    def start_battle(self):
        self.start_time=time.time()
        m_btnStartGame = engine.find_element(Page.btn_BeginFight)
        screen_shot_click(m_btnStartGame)
        #auto fight
        m_autoFight = engine.find_element(Page.btn_AutoFight)
        screen_shot_click(m_autoFight)
    def test_each_level(self):
        for index,level in enumerate(self.levelList):
            print("关卡id---->"+level)
            engine.call_registered_handler("GoTo", "n"+level) #这里调用unity里的GM指令
            self.start_battle()
            battleResult = find_element_wait(Page.panel_BattleRes, 65, 5)
            if (battleResult):
                screen_shot_click(battleResult)
                duration = str(int(time.time() - self.start_time))  # 关卡持续时间
                self.excelWriter.write_excel(index, 0, level)  # 第一列写关卡名
                self.excelWriter.write_excel(index, 1, duration)  # 第二列写通关时间
            else:
                duration = str(int(time.time() - self.start_time))  # 关卡持续时间
                self.excelWriter.write_excel(index, 0, level)  # 第一列写关卡名
                self.excelWriter.write_excel(index, 1, duration)  # 第二列写通关时间
                self.restart_game()
                self.default_login()
                find_element_wait(Page.btn_Battle)
        self.excelWriter.save_excel()
    def restart_game(self): #重启游戏
        os.system("adb shell am force-stop  %s"%Page.package_name)
        time.sleep(2)
        os.system("adb shell am start -n %s/.NativeUnityPlayerActivity"%Page.package_name)
    def default_login(self):#登录一次后续直接点击就可以登录
        #m_BtnSart2
        start_btn = find_element_wait(Page.btn_LogIn)
        screen_shot_click(start_btn)
if __name__ == "__main__":
    ab = AutoBattle("level.xls")
    ab.test_each_level()

import xlrd
import time
import xlwt
class ExcelReader:
    def __init__(self,excel_path):
        self.excel_path = excel_path;
    def read_first_sheet_first_col(self):
        data = xlrd.open_workbook(self.excel_path)
        st = data.sheet_by_index(0)
        col  = [str(st.cell_value(i, 0)).replace(".0","") for i in range(0, st.nrows)]
        return col
class ExcelWriter:
    def __init__(self):
        self.wb = xlwt.Workbook()
        self.sh = self.wb.add_sheet("关卡通过时间记录")
        self.cur_col  =0
    def write_excel(self,row ,col,record_str):
        self.sh.write(row, col, record_str)
    def save_excel(self):
        date_string = time.strftime("%Y%m%d%H%M")
        excel_name ="TestResult"+date_string+".xls"
        self.wb.save(excel_name)
if __name__=="__main__":
    ew = ExcelWriter()
    ew.save_excel()