python查看物理内存_获取进程所占的物理内存

## 概述

通过计算linux系统的/proc文件中物理内存相关内容，获取指定进程所占的物理内存。

这里以获取apache进程的物理内存和所占比例为例。

### 脚本

* 此脚本可以作为模块使用

* apache会开启多个进程，因此这个脚本适合单进程和多进程。

* 在主函数外，对计算过程进程了封装，定义了`getProgTotalMem`函数，简化了主函数中的内容.

* 当进程不存在时，pidof不报错，输出为空，因此这里检测进程是否存在应特别注意

* 可在[github gist][1]中获取脚本，被墙的原因，多数情况网页开。

```python

#!/usr/bin/env python

# coding: utf-8

import os

import sys

from subprocess import Popen, PIPE

def getPidList(prog_name):

"""获取进程的所有PID

输入：进程名

返回：进程PID列表"""

p = Popen(['pidof', '%s' % prog_name], stdout=PIPE, stderr=PIPE)

std_out, std_err = p.communicate()

# 当进程不存在时，pidof也不报错，所以检测std_out如果为空，说明进程不存在。

if std_out:

pids = std_out.split()

return pids

else:

print 'the prog %s is not exist.' % prog_name

sys.exit()

def getProgMemDict(pids):

"""获取进程每个PID所占的物理内存,存入字典

输入：进程PID列表

返回：形如{PID:mem}的字典"""

mem_dict = {}

for pid in pids:

fn = os.path.join('/proc', pid, 'status')

with open(fn,'r') as fd:

for line in fd:

if line.startswith('VmRSS'):

mem = int(line.split()[1])

mem_dict[pid] = mem

break

return mem_dict

def getProgTotalMem(prog_name):

"""获取进程所占的物理内存

输入：进程名

输出：进程所占的物理内存"""

# 获取进程所有PID

pid_list = getPidList(prog_name)

# 获取进程物理内存字典

prog_mem_dict = getProgMemDict(pid_list)

# 计算进程总的的物理内存

prog_total_mem = sum([mem for mem in prog_mem_dict.itervalues()])

return prog_total_mem

def getSysTotalMem():

"""获取系统总的物理内存

输入：无

输出：总的物理内存"""

with open('/proc/meminfo') as fd:

for line in fd:

if line.startswith('MemTotal'):

mem = int(line.split()[1])

return mem

if __name__ == '__main__':

httpd_total_mem = getProgTotalMem('http')

sysmem = getSysTotalMem()

print 'apache memory is: %d kb, and %.2f%%' % (httpd_total_mem, httpd_total_mem/float(sysmem))

```

[1]:https://gist.github.com/PoplarYang/69daae1e2b11a309e0de5a6a52cca5c4