view contrib/memory.py @ 14817:70af3b2e3642 stable

i18n-ja: recover template translations from 15200b46165b These translations were removed by 68a697421a57 because the original strings were transfered to the docstrings (264f292a0c6f) without the corresponding sources being added to the Makefile for extraction.
author Wagner Bruna <wbruna@softwareexpress.com.br>
date Thu, 30 Jun 2011 20:05:23 -0300
parents 08a0f04b56bd
children 3e0d27d298b7
line wrap: on
line source

# memory.py - track memory usage
#
# Copyright 2009 Matt Mackall <mpm@selenic.com> and others
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.

'''helper extension to measure memory usage

Reads current and peak memory usage from ``/proc/self/status`` and
prints it to ``stderr`` on exit.
'''

import atexit

def memusage(ui):
    """Report memory usage of the current process."""
    status = None
    result = {'peak': 0, 'rss': 0}
    try:
        # This will only work on systems with a /proc file system
        # (like Linux).
        status = open('/proc/self/status', 'r')
        for line in status:
            parts = line.split()
            key = parts[0][2:-1].lower()
            if key in result:
                result[key] = int(parts[1])
    finally:
        if status is not None:
            status.close()
    ui.write_err(", ".join(["%s: %.1f MiB" % (key, value / 1024.0)
                            for key, value in result.iteritems()]) + "\n")

def extsetup(ui):
    atexit.register(memusage, ui)