mercurial-scm/hg: contrib/memory.py annotate

annotate contrib/memory.py @ 52656:3e84e001b6c1

pyupgrade: drop redundant `open()` modes On one hand, this gets rid of some py2 cruft (like `U` and `t` modes). OTOH, I like being explict. But there's no way to disable individual transformations (and this change is part of `_fix_tokens()`, which is even harder to selectively use locally), and I'd rather have less noise when using `pyupgrade`. Warning: `contrib/synthrepo.py` appears not to have been updated to py3, as it's passing str to `error.Abort`. But since it was opening the file in unicode on py2, this change is a no-op.

author	Matt Harbison <matt_harbison@yahoo.com>
date	Tue, 07 Jan 2025 17:28:46 -0500
parents	6000f5b25c9b
children

rev	line source
10017 253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	1 # memory.py - track memory usage
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	2 #
46819 d4ba4d51f85f contributor: change mentions of mpm to olivia Rapha?l Gom?s <rgomes@octobus.net> parents: 43076 diff changeset	3 # Copyright 2009 Olivia Mackall <olivia@selenic.com> and others
10017 253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	4 #
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	5 # This software may be used and distributed according to the terms of the
10264 d6512b3e9ac0 Merge with stable Matt Mackall <mpm@selenic.com> parents: 10017 diff changeset	6 # GNU General Public License version 2 or any later version.
10017 253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	7
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	8 '''helper extension to measure memory usage
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	9
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	10 Reads current and peak memory usage from ``/proc/self/status`` and
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	11 prints it to ``stderr`` on exit.
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	12 '''
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	13
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	14
10017 253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	15 def memusage(ui):
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	16 """Report memory usage of the current process."""
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	17 result = {'peak': 0, 'rss': 0}
52656 3e84e001b6c1 pyupgrade: drop redundant `open()` modes Matt Harbison <matt_harbison@yahoo.com> parents: 48875 diff changeset	18 with open('/proc/self/status') as status:
10017 253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	19 # This will only work on systems with a /proc file system
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	20 # (like Linux).
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	21 for line in status:
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	22 parts = line.split()
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	23 key = parts[0][2:-1].lower()
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	24 if key in result:
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	25 result[key] = int(parts[1])
43076 2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	26 ui.write_err(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	27 ", ".join(
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	28 ["%s: %.1f MiB" % (k, v / 1024.0) for k, v in result.iteritems()]
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	29 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	30 + "\n"
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	31 )
2372284d9457 formatting: blacken the codebase Augie Fackler <augie@google.com> parents: 31958 diff changeset	32
10017 253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	33
253d0da256b2 contrib: helper extension to track memory usage Martin Geisler <mg@lazybytes.net> parents: diff changeset	34 def extsetup(ui):
31958 de5c9d0e02ea atexit: switch to home-grown implementation Bryan O'Sullivan <bryano@fb.com> parents: 30386 diff changeset	35 ui.atexit(memusage, ui)

Mercurial > public > mercurial-scm > hg

annotate contrib/memory.py @ 52656:3e84e001b6c1