统计应用冷热启动时间

#!/usr/bin/python
# -*- coding: utf-8 -*-

import uiautomator2 as u2
import os
import time
import pandas as pd
import numpy as np

d = u2.connect()


dic = {
"相机":"com.oppo.camera",
"抖音":"com.oppo.camera"
}
 

def count_startTime():
    list_cold_startTime = []
    list_hot_startTime = []
    
    for app_name,package_name in dic.items():# cold start time     
        list_cold = []    
        for i in range(10):
            d.app_stop("{}".format(package_name))
            time.sleep(1)
            cold_startTime = os.popen('''adb shell " am start -W -n %s|grep WaitTime|awk -F ':' '{print $2}'" '''%package_name).read()
            time.sleep(6)    
            d.app_stop("{}".format(package_name))
            list_cold.append(cold_startTime)
        ave_time =round(np.mean(list_cold))    
        list_cold_startTime.append([app_name,ave_time]+list_cold)
    print(list_cold_startTime)
    
    for app_name,package_name in dic.items():# hot start time   
        list_hot = []      
        for i in range(10):
            d.app_start("{}".format(value))
            d.press("home")
            time.sleep(1)
            hot_startTime = os.popen('''adb shell " am start -W -n %s|grep WaitTime|awk -F ':' '{print $2}'" '''%package_name).read()
            time.sleep(6)    
            d.app_stop("{}".format(package_name))
            list_hot.append(hot_startTime)
        ave_time =round(np.mean(list_hot))    
        list_hot_startTime.append([app_name,ave_time]+list_hot)
    print(list_hot_startTime)


    df1=pd.DataFrame(list_cold_startTime,columns=['应用名称','平均时间(ms)',"第1次","第2次","第3次","第4次","第5次","第6次","第7次","第8次","第9次","第10次"])
    df2=pd.DataFrame(list_hot_startTime,columns=['应用名称','平均时间(ms)',"第1次","第2次","第3次","第4次","第5次","第6次","第7次","第8次","第9次","第10次"])
    with pd.ExcelWriter('应用冷热启动时间.xlsx') as writer:
        df1.to_excel(writer, sheet_name='冷启动时间', index=False)
        df2.to_excel(writer, sheet_name='热启动时间', index=False)
        
        
if __name__ == "__main__":
    count_startTime()