# -*- coding:utf-8 -*-
import subprocess
import time
import sys
#支持命令行传参。传入你想测的pid值
pid = sys.argv[1]
# 第一个时间点的pid的进程信息
proc_info_1 = subprocess.check_output("pidstat -r -u -p {}".format(pid), shell=True).decode('utf-8')
time.sleep(3)
# 第二个时间点的pid的进程信息
proc_info_2 = subprocess.check_output("pidstat -r -u -p {}".format(pid), shell=True).decode('utf-8')# 解析进程信息
def parse_process_info(info):lines = info.strip().split('\n')headers1 = lines[2].split()values1 = lines[3].split()headers2 = lines[5].split()values2 = lines[6].split()proc_data1 = dict(zip(headers1, values1))proc_data2 = dict(zip(headers2, values2))#print(proc_data1)return float(proc_data1['%CPU']), int(proc_data2['RSS'])# 解析第一个时间点和第二个时间点的进程信息
cpu_1, mem_1 = parse_process_info(proc_info_1)
cpu_2, mem_2 = parse_process_info(proc_info_2)# 计算差值
cpu_diff = cpu_2 - cpu_1
mem_diff = mem_2 - mem_1# 打印结果
print("CPU利用率差值: {}%".format(cpu_diff))
print("内存占用差值: {}".format(mem_diff))