import os import re import sys import string class ProcInfo: dir_path = "/proc/" # Our cute Proc File System status_file = "status" stat_file = "stat" proc_list = [] # Our PID list :D proc_info = [] # def __init__(self): self.proc_list = self.Get_PID_List() # Returns Process List def Get_PID_List(self): list = [] # Exists our procfs ? if os.path.isdir(self.dir_path): # around dir entries for f in os.listdir(self.dir_path): if os.path.isdir(self.dir_path+f) & str.isdigit(f): list.append(int(f)) return list def MemoryInfo(self, pid): # Path pidfile = self.dir_path + str(pid) + "/stat" try: infile = open(pidfile, "r") except: print "Error trying " + pidfile return None # Parsing data , check 'man 5 proc' for details stat_data = infile.read() infile.close() process_name = self._get_process_name(stat_data) data = self._get_safe_split(stat_data) state_dic = { 'R': 'Running', 'S': 'Sleeping', 'D': 'Disk sleep', 'Z': 'Zombie', 'T': 'Traced/Stopped', 'W': 'Paging' } # user and group owners pidstat = os.stat(pidfile) info = { 'pid': int(data[0]), # Process ID 'name': process_name, 'state': data[2], # Process State, ex: R|S|D|Z|T|W 'state_name': state_dic[data[2]], # Process State name, ex: Running, sleeping, Zombie, etc 'ppid': int(data[3]), # Parent process ID 'utime': int(data[13]), # Used jiffies in user mode 'stime': int(data[14]), # Used jiffies in kernel mode 'start_time': int(data[21]), # Process time from system boot (jiffies) 'vsize': int(data[22]), # Virtual memory size used (bytes) 'rss': int(data[23])*4, # Resident Set Size (bytes) 'user_id': pidstat.st_uid, # process owner 'group_id': pidstat.st_gid # owner group } return info # Return the process name def _get_process_name(self, data): m = re.search("\(.*\)", data) return m.string[m.start()+1:m.end()-1] def _get_safe_split(self, data): new_data = re.sub("\(.*\)", '()', data) return new_data.split() # Returns the CPU usage expressed in Jiffies def get_CPU_usage(self, cpu_hz, used_jiffies, start_time): # Uptime info uptime_file = self.dir_path + "/uptime" try: infile = file(uptime_file, "r") except: print "Error trying uptime file" return None uptime_line = infile.readline() uptime = string.split(uptime_line, " ",2) infile.close() # System uptime, from /proc/uptime uptime = float(uptime[0]) # Jiffies avail_jiffies = (uptime * cpu_hz) - start_time cpu_usage = {'used_jiffies': used_jiffies, 'avail_jiffies': avail_jiffies} return cpu_usage