性能测试培训: 监控CPU之python
作为一名测试开发工程师，开发脚本是为了测试服务的，我们在手里没有性能监控工具的情况下，我们会自己来进行开发脚本完成监控任务。下面是python监控cpu
'''
Created on 2015-10-20
@author:laoli
'''
#from sshExpert import sshExpert
import time
import os,sys,platform
class cpuMonitor:
def __init__(self):
return
def getCpu(self):
sysName = platform.system()
if sysName == 'Windows':
return self.getWinCpu()
else:
return self.getLinuxCpu()
return
def getWinCpu(self):
import win32com.client
com=win32com.client.Dispatch("WbemScripting.SWbemRefresher")
obj=win32com.client.GetObject("winmgmts:\rootcimv2")
processorItems=com.AddEnum(obj,"Win32_PerfFormattedData_PerfOS_Processor").objectSet
while(1):
com.Refresh()
for item in processorItems:
if item.Name == '_Total':
#print item.Name,"= ",
#print 'total=',item.PercentProcessorTime,"%"
return item.PercentProcessorTime
#time.sleep(5)
break
def _read_cpu_usage(self):
"""Read the current system cpu usage from /proc/stat."""
try:
fd = open("/proc/stat", 'r')
lines = fd.readlines()
finally:
if fd:
fd.close()
for line in lines:
l = line.split()
if len(l) < 5:
continue
if l[0].startswith('cpu'):
return l
return []
def getLinuxCpu(self):
"""
get cpu avg used by percent
"""
cpustr=self._read_cpu_usage()
if not cpustr:
return 0
#cpu usage=[(user_2 +sys_2+nice_2) - (user_1 + sys_1+nice_1)]/(total_2 - total_1)*100
usni1=long(cpustr[1])+long(cpustr[2])+long(cpustr[3])+long(cpustr[5])+long(cpustr[6])+long(cpustr[7])+long(cpustr[4])
usn1=long(cpustr[1])+long(cpustr[2])+long(cpustr[3])
#usni1=long(cpustr[1])+long(cpustr[2])+long(cpustr[3])+long(cpustr[4])
# self.sleep=2
time.sleep(self.sleep)
cpustr=self._read_cpu_usage()
if not cpustr:
return 0
usni2=long(cpustr[1])+long(cpustr[2])+float(cpustr[3])+long(cpustr[5])+long(cpustr[6])+long(cpustr[7])+long(cpustr[4])
usn2=long(cpustr[1])+long(cpustr[2])+long(cpustr[3])
cpuper=(usn2-usn1)/(usni2-usni1)
return 100*cpuper
def main():
aaa = cpuMonitor()
print aaa.getCpu()
if __name__ == '__main__':
main()