MemoryMonitor.py

Go to the documentation of this file.

import os

def _proc_status(pid):
    return '/proc/%d/status' % pid

_scale = {'kB': 1024.0, 'mB': 1024.0*1024.0,
          'KB': 1024.0, 'MB': 1024.0*1024.0}

def _VmB(pid, VmKey):
    '''Private.
    '''
    global _scale
    proc_status = _proc_status(pid)
     # get pseudo file  /proc/<pid>/status
    try:
        t = open(proc_status)
        v = t.read()
        t.close()
    except:
        return 0.0  # non-Linux?
     # get VmKey line e.g. 'VmRSS:  9999  kB\n ...'
    try:
        i = v.index(VmKey)
    except ValueError:
        return 0.0
    v = v[i:].split(None, 3)  # whitespace
    if len(v) < 3:
        return 0.0  # invalid format?
     # convert Vm value to bytes
    return float(v[1]) * _scale[v[2]]


def memory(pid, since=0.0):
    '''Return memory usage in bytes.
    '''
    return _VmB(pid, 'VmSize:') - since


def resident(pid, since=0.0):
    '''Return resident memory usage in bytes.
    '''
    return _VmB(pid, 'VmRSS:') - since


def stacksize(pid, since=0.0):
    '''Return stack size in bytes.
    '''
    return _VmB(pid, 'VmStk:') - since