Python & Ubuntu 获取进程信息

备忘，其实是想知道该进程使用的内存状况。

首先要获取系统的内存信息：

def memory_stat():
'''
return the memory info
'''
mem = {}
stat = {}
f = open('/proc/meminfo')
lines = f.readlines()
for line in lines:
if len(line) < 2 : continue
name = line.split(':')[0]
var = line.split(':')[1].split()[0]
mem[name] = long(var) * 1024.0
stat['MemUsed'] = mem['MemTotal'] - mem['MemFree'] - mem['Buffers'] - mem['Cached']
stat['MemTotal'] = mem['MemTotal']
stat['MemFree'] = mem['MemFree']
stat['Buffers'] = mem['Buffers']
stat['Cached'] = mem['Cached']
return stat

因为我下面使用的方法来获取的进程信息，只能知道该进程占内存的百分比。所以必须先获取内存的总量。

接下来定义一个获取进程信息的函数。

def proc_stat(procid):
import commands
ps_stat = None
try:
#获取进程的内存信息
ps_stat = commands.getoutput('ps -fp %s -u' % procid).split('\n')[2].split()
except:
return {}

procInfo = {}

procInfo['PID'] = procid

#这里调用了刚才定义的 memory_stat 获取内存总量，
#然后乘以该进程所占内存的百分比，即为内存使用量
procInfo['MemoryUsed'] = memory_stat()['MemTotal'] * float(ps_stat[3]) / 100

procInfo['Start'] = ps_stat[8]
procInfo['StartCommand'] = ps_stat[10]

return procInfo

接着使用 ipython ，测试 ，先开一个 python 进程 ，假设进程ID为2168，则：

from MachineInfo import proc_stat

ps = proc_stat(2168)

ps

{'MemoryUsed': 3650289.6639999999, #这里是以 byte 为单位，也可以自己进行转换
'PID': 2168,
'Start': '21:56',
'StartCommand': 'python'}

记录完毕。Over~