参考链接:MacOS使用top命令查看进程使用内存
import sys
import time
import os
import psutil
# 设置app名称,名称中不允许有空格,否则后面取top结果会错位
if len(sys.argv) < 2:
app_name = "MyDemo"
else:
app_name = str(sys.argv[1])
# 根据app_name查找进程id
pid = None
for proc in psutil.process_iter():
if app_name in proc.name():
pid = proc.pid
print("pid = [%s]" % str(pid))
# 拼接top命令,-pid 制定进程id,-l 查询3次(小于3次不准确),tail -n 获取最后一次的结果, awk print 输出第三列、第八列结果(对应%cpu和mem)
get_cpu_mem = "top -pid " + str(pid) + " -l 3 | tail -n 1 | awk '{print $3,$8}'"
print("execute cmd = [%s]" % str(get_cpu_mem))
# monitor process and write data to file
interval = 30 # polling seconds
with open("process_monitor_" + app_name + '_' + str(pid) + ".csv", "a+") as f:
f.write("time,cpu%,mem%\n") # titles
while True:
current_time = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))
current_cpu_mem = os.popen(get_cpu_mem).read().strip()
cpu = current_cpu_mem.split(" ")[0]
mem = current_cpu_mem.split(" ")[1]
line = current_time + ',' + str(cpu) + ',' + str(mem)
print(line)
f.write(line + "\n")
time.sleep(interval)