背景: 自动化测试中遇到经常重复某个应用场景时,会导致内存持续上升,为了实时监控内存的使用情况。
#!/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表格的内容,并保存到本地。