使用python代码实时监控某个进程的内存使用情况

背景： 自动化测试中遇到经常重复某个应用场景时，会导致内存持续上升，为了实时监控内存的使用情况。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# coding:gbk
import os, re
import time
import string

#统计某一个进程名所占用的内存，同一个进程名，可能有多个进程
def countProcessMemoey(processName):
    pattern = re.compile(r'([^\s]+)\s+(\d+)\s.*\s([^\s]+\sK)')
    cmd = 'tasklist /fi "imagename eq ' + processName + '"' + ' | findstr.exe ' + '"'+ processName + '"'
    # findstr后面的程序名加上引号防止出现程序带有空格
    result = os.popen(cmd).read()
    resultList = result.split("\n")


    for srcLine in resultList:
        srcLine = "".join(srcLine.split('\n'))
        if len(srcLine) == 0:
            break
        m = pattern.search(srcLine)
        if m == None:
            continue
        #由于是查看python进程所占内存，因此通过pid将本程序过滤掉
        if str(os.getpid()) == m.group(2):
            continue
        ori_mem = m.group(3).replace(',','')
        ori_mem = ori_mem.replace(' K','')
        ori_mem = ori_mem.replace(r'\sK','')
        memEach = int(ori_mem)
        print ('ProcessName:'+ m.group(1) + '\tPID:' + m.group(2) + '\tmemory size:%.2f'% (memEach * 1.0 /1024), 'M')
        print(time.ctime())  #打印当前的时间

    print("*" * 58)

if __name__ == '__main__':
    #进程名
    ProcessName = 'pycharm.exe'


    while True:
        countProcessMemoey(ProcessName)
        time.sleep(5)

下一步完善方向：将结果存入excel表格中，制作折线图

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# coding:gbk
import os, re
import time
import string

import xlrd
from xlutils.copy import copy


def countProcessMemoey(processName):
    pattern = re.compile(r'([^\s]+)\s+(\d+)\s.*\s([^\s]+\sK)')
    cmd = 'tasklist /fi "imagename eq ' + processName + '"' + ' | findstr.exe ' + '"'+ processName + '"'
    # findstr后面的程序名加上引号防止出现程序带有空格
    result = os.popen(cmd).read()
    resultList = result.split("\n")
    for srcLine in resultList:
        srcLine = "".join(srcLine.split('\n'))
        if len(srcLine) == 0:
            break
        m = pattern.search(srcLine)
        if m == None:
            continue
        # 由于是查看python进程所占内存，因此通过pid将本程序过滤掉
        if str(os.getpid()) == m.group(2):
            continue
        ori_mem = m.group(3).replace(',','')
        ori_mem = ori_mem.replace(' K','')
        ori_mem = ori_mem.replace(r'\sK','')
        memEach = int(ori_mem)
        # print ('ProcessName:'+ m.group(1) + '\tPID:' + m.group(2) + '\tmemory size:%.2f'% (memEach * 1.0 /1024), 'M')
        cpu = '%.2f'% (memEach * 1.0 /1024) +'M'
        # print(time.ctime())  # 打印当前的时间
        value = [time.ctime(), cpu]
        rb = xlrd.open_workbook(r'F:\Python全课程练习\python_test_project\HTML\data.xls', formatting_info=True)
        wb = copy(rb)
        writeSheet = wb.get_sheet(0)
        writeSheet.write(x, y, value[0])
        writeSheet.write(x, y+1, value[1])
        wb.save(r'F:\Python全课程练习\python_test_project\HTML\data.xls')

    print("*" * 58)


if __name__ == '__main__':
    # 进程名
    ProcessName = input("请输入你想监控的进程名：")
    x = 0
    y = 0

    while True:
        countProcessMemoey(ProcessName)
        x += 1
        time.sleep(5)

展示：

首先尝试用了xlwt去试，出了点问题。应该是多虑了，以为是库的问题，然后换了xlutils，还是推荐使用xlutils吧

附上三者区别：

xlwt模块用于在内存中生成一个xls/xlsx对象，增加表格数据，并把内存中的xls对象保存为本地磁盘xls文件;

xlrd模块用于把本地xls文件加载到内存中，可以读取xls文件的表格数据，查询xls文件的相关信息;

xlwt可以生成xls文件，xlrd可以读取已经存在的xls文件，但是如果要修改本地已经存在的xls文件，就需要用到xlutils模块。

xlutils模块是xlrd和xlwt之间的桥梁，可以使用xlutils模块中的copy模块，拷贝一份通过xlrd读取到内存中的xls对象，就可以在拷贝对象上像xlwt中那样修改xls表格的内容，并保存到本地。