Python获取android设备cpu和内存占用情况

1. 功能：获取android设备中某一个app的cpu和内存
2. 环境：python和adb
3. 使用方法：使用adb连接android设备，打开将要测试的app，执行cpu/内存代码
4. cpu获取代码如下：（输入参数为脚本执行时间）

   # coding:utf-8
   '''
   获取系统total cpu
   '''
   import os, csv
   import time
   import csv
   import numpy as np
   from matplotlib import pyplot as plt
   
   cpu_list = []
   time_list = []
   app_list = []
   lines = []
   package_name = []
   
   
   # 读取进程名称（包名）
   def get_applist():
       global package_name
       with open('config/director.txt', encoding='utf-8', mode='r') as f:
           lines_all = f.readlines()
           for appname in lines_all:
               package_name1 = appname
               appname_new = appname[0:15]
               package_name.append(package_name1)
               lines.append(appname_new)
           for line in lines:
               app_list.append(line.strip())
   
   
   # 获取cpu数值
   def get_cpu():
       global filename
       with open(filename, encoding="utf-8", mode="r") as f:
           lines = f.readlines()
           for appname in app_list:
               for lis in lines:
                   # 适配低版本手机
                   if appname in lis and '%' in lis:
                       now = time.strftime("%H:%M:%S", time.localtime())
                       time_list.append(now)
                       cpu_1 = lis.split('%')[0]
                       cpu_2 = cpu_1.split(' ')
                       # print(cpu_2)
                       cpu = cpu_2[len(cpu_2) - 1]
                       print(cpu, now)
                       cpu_list.append(cpu)
                       break
                   # 适配高版本手机
                   elif appname in lis:
                       now = time.strftime("%H:%M:%S", time.localtime())
                       time_list.append(now)
                       cpu1 = lis.split(' ')
                       # print(cpu1)
                       cpu2 = list(set(cpu1))
                       cpu2.sort(key=cpu1.index)
                       cpu_h = cpu2[len(cpu2) - 4]
                       print(cpu_h, now)
                       cpu_list.append(cpu_h)
                       break
                   else:
                       pass
   
   
   # csv头部
   def write_head():
       headers = ['name:']
       headers.append(app_list[0])
       headers.append('init_cpu')
       with open('log_su/cpuinfo.csv', 'w+', newline='') as csvfile:
           writer = csv.DictWriter(csvfile, fieldnames=headers)
           writer.writeheader()
   
   
   # 将数值写入csv，用于绘图时读取
   def write_report():
       # headers = ['name', 'aaa', 'init_cpu']
       with open('log_su/cpuinfo.csv', 'a+', newline='') as csvfile:
           writer = csv.writer(csvfile)
           for key in cpu_list:
               writer.writerow([' ', ' ', key])
   
   
   # 绘制折线图，生成测试报告
   def mapping():
       filename = 'log_su/cpuinfo.csv'
       with open(filename) as f:
           reader = csv.reader(f)
           header_row = next(reader)
           highs = []
           for row in reader:
               high = row[2]
               highs.append(high)
           # print(highs)
   
       wights = time_list
       highs_float = list(map(float, highs))
       # print(f"****{highs}")
       print(f"CPU值：{highs_float}")
       # 输出平均值
       total = 0
       for value in highs_float:
           total += value
       average = round(total/len(highs_float), 2)
       print(f"CPU平均值：{average}")
   
       #输出最低值和最高值
       highs_hl = sorted(highs_float)
       print(f"CPU最低值：{highs_hl[0]}")
       print(f"CPU最高值：{highs_hl[len(highs_hl)-1]}")
   
       # 根据数据绘制图形
       plt.figure(figsize=(11, 4), dpi=600)
       # 生成网格
       # plt.grid()
       plt.grid(axis="y")
       # 折线图
       if package_name[0] == 'com.oneapp.max.security.pro.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="PPP")
       elif package_name[0] == 'com.oneapp.max.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Opt1.6.1")
       elif package_name[0] == 'com.boost.clean.coin.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Fastclear")
       elif package_name[0] == 'com.walk.sports.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Walk")
       elif package_name[0] == 'com.diamond.coin.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Amber")
       elif package_name[0] == 'com.oneapp.max.cleaner.booster.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Space")
       else:
           plt.plot(wights, highs_float, "c-", linewidth=1, label=package_name[0])
       # 坐标轴范围
       # plt.ylim(300, 400)
       # plt.xlim(0, 10)
   
       plt.xlabel('time(H:Min:S)', fontsize=16)
       plt.ylabel("cpu_realtime(%)", fontsize=16)
       plt.title("cpu real time line chart", fontsize=24)
       plt.legend()
   
       # 横坐标显示间隔
       if len(wights) <= 15:
           pass
       else:
           t = int(len(wights) / 15)
           plt.xticks(range(0, len(wights), t))
   
       # 纵坐标显示间隔
       # plt.yticks(range(100, 300, 10))
   
       # 旋转日期
       plt.gcf().autofmt_xdate()
   
       # 展示每个坐标
       # for a, b in zip(wights, highs_float):
       #     plt.text(a, b, (a, b), ha='center', va='bottom', fontsize=8)
   
       # plt.show()
   
       time_now = time.strftime("%m%d-%H:%M:%S", time.localtime())
       path = "report/" + time_now
       plt.savefig(path)
   
   
   # 自动识别当前需检测的
   def name_app():
       cmd = 'adb shell dumpsys window | grep mCurrentFocus > log_su/name_info.csv'
       os.system(cmd)
       with open('log_su/name_info.csv', encoding='utf-8', mode='r') as f:
           lines = f.readlines()
           for line in lines:
               if 'mCurrentFocus' in line:
                   name1 = line.split('/')[0].split(' ')
                   name = name1[len(name1) - 1]
   
       with open('config/director.txt', encoding='utf-8', mode='w') as f_name:
           text = name
           f_name.write(text)
       print(f"将要监测的包名为：{text}")
   
   #控制监测时间
   def time_control():
       global filename
       while True:
           end_time = time.time()
           if (end_time - start_time)/60 >= tol_time:   #分钟
           # if end_time - start_time >= tol_time:  # 秒
               break
   
           time.sleep(1)
           adb = "adb shell top -n 1 > log_su/adb_info.csv"
           d = os.system(adb)
           filename = "log_su/adb_info.csv"
           get_cpu()
   
   
   if __name__ == "__main__":
       name_app()
       tol_time = int(input("请输入脚本执行时间(分钟)："))
       start_time = time.time()
       get_applist()
       write_head()
       time_control()
       write_report()
       mapping()

   会在.py文件同级目录下生成3个文件夹，config、log_su、report，其中运行结果在report中

5. 结果以是生成折线图，看起来直观，如下：这里我解释下，cpu占比是adb获取的实时占比，但是满值并不一定是100%，比如这张图，用的是一个八核的手机，所以CPU满值是800%

6. 内存获取代码如下：（输入参数为脚本执行时间）

   # coding:utf-8
   '''
   获取系统total memory
   '''
   import os, csv
   import time
   import csv
   import numpy as np
   from matplotlib import pyplot as plt
   
   mem_dict = {}
   time_list = []
   app_list = []
   package_name = []
   t = 0
   def get_applist():
       global package_name
       with open('config/director.txt', encoding='utf-8', mode='r') as f:
           lines = f.readlines()
           for line in lines:
               package_name1 = line
               package_name.append(package_name1)
               app_list.append(line.strip())
   
   
   def get_mem():
       global filename
       with open(filename, encoding="utf-8", mode="r") as f:
           lines = f.readlines()
           start_flag = False
           for appname in app_list:
               for line in lines:
                   if "Total PSS by OOM adjustment" in line:
                      break
                   if appname in line and 'pid' in line and 'kB' in line:
                       mem_v = line.strip().split(':')[0].replace('kB', '').replace(',', '')
                       line_name = line.split(':')[1].split('(')[0].strip()
                       if line_name in appname:
                           mem_v = round(float(mem_v) / 1024, 2)
                           mem_dict[appname] = mem_v
                           now_v = time.strftime("%H:%M:%S", time.localtime())
                           # now_int = int(now_v)
                           time_list.append(now_v)
                           print(mem_v, now_v)
                           break
                   elif appname in line and 'pid' in line and 'K' in line:
                       mem_v = line.strip().split(':')[0].replace('K', '').replace(',', '')
                       line_name = line.split(':')[1].split('(')[0].strip()
                       if line_name in appname:
                           mem_v = round(float(mem_v) / 1024, 2)
                           mem_dict[appname] = mem_v
                           now_v = time.strftime("%H:%M:%S", time.localtime())
                           # now_int = int(now_v)
                           time_list.append(now_v)
                           print(mem_v, now_v)
                           break
   
   def write_head():
       headers = ['name:']
       headers.append(app_list[0])
       headers.append('init_mem')
       with open('log_su/meminfo.csv', 'w+', newline='') as csvfile:
           writer = csv.DictWriter(csvfile, fieldnames=headers)
           writer.writeheader()
   
   def write_report():
       headers = ['name','aaa', 'init_mem']
       with open('log_su/meminfo.csv', 'a+', newline='') as csvfile:
           writer = csv.DictWriter(csvfile, fieldnames=headers)
           for key in mem_dict:
               writer.writerow({'init_mem': mem_dict[key]})
   
   
   def mapping():
       filename = 'log_su/meminfo.csv'
       with open(filename) as f:
           reader = csv.reader(f)
           header_row = next(reader)
           highs = []
           for row in reader:
               high = row[2]
               highs.append(high)
           # print(highs)
   
       wights = time_list
       highs_float = list(map(float, highs))
   
       print(f"内存值：{highs_float}")
   
       # 输出平均值
       total = 0
       for value in highs_float:
           total += value
       average = round(total / len(highs_float), 2)
       print(f"内存平均值：{average}")
   
       # 输出最低值和最高值
       highs_hl = sorted(highs_float)
       print(f"内存最低值：{highs_hl[0]}")
       print(f"内存最高值：{highs_hl[len(highs_hl) - 1]}")
   
       # 根据数据绘制图形
   
       plt.figure(figsize=(11, 4), dpi=600)
   
       # 生成网格
       # plt.grid()
       plt.grid(axis="y")
   
       if package_name[0] == 'com.oneapp.max.security.pro.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="PPP")
       elif package_name[0] == 'com.oneapp.max.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Opt")
       elif package_name[0] == 'com.boost.clean.coin.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="fastclear")
       elif package_name[0] == 'com.walk.sports.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Walk")
       elif package_name[0] == 'com.diamond.coin.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Amber")
       elif package_name[0] == 'com.oneapp.max.cleaner.booster.cn':
           plt.plot(wights, highs_float, "c-", linewidth=1, label="Space")
       else:
           plt.plot(wights, highs_float, "c-", linewidth=1, label=package_name[0])
       # 坐标轴范围
       # plt.ylim(300, 400)
       # plt.xlim(0, 10)
   
       plt.xlabel('time(H:Min:S)', fontsize=16)
       plt.ylabel("Number (Mb)", fontsize=16)
       plt.title("meminfo", fontsize=24)
       plt.legend()
   
       # 横坐标显示间隔
       if len(wights) <= 15:
           pass
       else:
           t = int(len(wights) / 15)
           plt.xticks(range(0, len(wights), t))
   
       # 坐标刻度
       # my_y_ticks = np.arange(300, 400, 10)
       # my_x_ticks = np.arange(1, 10, 1)
       # plt.xticks(my_x_ticks)
       # plt.yticks(my_y_ticks)
       # plt.yticks(range(100, 300, 10))
   
       #旋转日期
       plt.gcf().autofmt_xdate()
   
       # 展示每个坐标
       # for a, b in zip(wights, highs_float):
       #     plt.text(a, b, (a, b), ha='center', va='bottom', fontsize=8)
   
       # plt.show()
   
       time_now = time.strftime("%m%d-%H:%M:%S", time.localtime())
       path = "report/" + time_now
       plt.savefig(path)
   
   def name_app():
       cmd = 'adb shell dumpsys window | grep mCurrentFocus > log_su/name_info.csv'
       os.system(cmd)
       with open('log_su/name_info.csv', encoding='utf-8', mode='r') as f:
           lines = f.readlines()
           for line in lines:
               if 'mCurrentFocus' in line:
                   name1 = line.split('/')[0].split(' ')
                   name = name1[len(name1) - 1]
   
       with open('config/director.txt', encoding='utf-8', mode='w') as f_name:
           text = name
           f_name.write(text)
       print(f"将要监测的包名为：{text}")
   
   def time_control():
       global filename
       while True:
           end_time = time.time()
           if (end_time - start_time)/60 >= tol_time:   #分钟
           # if end_time - start_time >= tol_time:    #秒
               break
           # time.sleep(2)
           # filename = str(input("请输入文件名："))
           adb = "adb shell dumpsys meminfo > log_su/adb_info.csv"
           d = os.system(adb)
           filename = "log_su/adb_info.csv"
           get_mem()
           write_report()
   
   if __name__ == "__main__":
       name_app()
       tol_time = int(input("请输入脚本执行时间(分钟)："))
       start_time = time.time()
       get_applist()
       write_head()
       time_control()
       mapping()

   会在.py文件同级目录下生成3个文件夹，config、log_su、report，其中运行结果在report中

7. 生成的内存结果图如下：

8. 若是有帮助请帮忙点个赞吧～

本文地址：https://blog.csdn.net/su869__/article/details/109646582