A very non-portable way of tracking memory consumption in Python

track_mem_linux.py

import os

PAGESIZE = os.sysconf("SC_PAGE_SIZE")  # BYTES
STATM_FIELDS = ["size", "resident", "share", "text", "lib", "data", "dt"]


def meminfo(pid=None):
    if pid is None:
        pid = os.getpid()
    pid = str(pid)
    with open("/proc/%s/statm" % pid) as f:
        info = f.next()
    info = [int(i)*PAGESIZE for i in info.split()]
    return dict(zip(STATM_FIELDS, info))


def get_total_mem(multiple=''):
    v = 1
    mmap = {}
    for k in '', 'K', 'M', 'G':
        mmap[k] = v
        v <<= 10
    mmap['T'] = v
    return meminfo()['size'] / mmap[multiple]