python统计某一个进程名所占用的内存

设计思路：

通过python，执行cmd中tasklist命令，获取要统计的进程的相关信息；通过正则表达式，查找出进程名称、进程pid、内存使用，然后打印出来。

作为pythoner，有时候需要统计python.exe进程所占用的内存，因此在设计时，通过pid，过滤掉统计内存这个脚本所启用的内存的信息，只统计其他的python.exe的信息。

具体实现：

#!/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
    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 = string.atoi(ori_mem)
        print 'ProcessName:'+ m.group(1) + '\tPID:' + m.group(2) + '\tmemory size:%.2f'% (memEach * 1.0 /1024), 'M'

    print "*" * 58

if __name__ == '__main__':
    #进程名
    ProcessName = 'python.exe'
    ProcessName2 = "chrome.exe"

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

运行结果：