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

功能：获取android设备中某一个app的cpu和内存

环境：python和adb

使用方法：使用adb连接android设备，打开将要测试的app，执行cpu/内存代码

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中

结果以是生成折线图，看起来直观，如下：

这里我解释下，cpu占比是adb获取的实时占比，但是满值并不一定是100%，比如这张图，用的是一个八核的手机，所以CPU满值是800%

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

# 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中

生成的内存结果图如下：