upstream/mercurial-mirror Files · contrib/memory.py

py3kcompat: added a "compatibility layer" for py3k...

py3kcompat: added a "compatibility layer" for py3k This patch adds some ugly constructs. The first of them is bytesformatter, a function that formats strings like when '%' is called. The main motivation for this function is py3k's strange behavior: >>> 'foo %s' % b'bar' "foo b'bar'" >>> b'foo %s' % b'bar' Traceback (most recent call last): File "<stdin>", line 1, in <module> TypeError: unsupported operand type(s) for %: 'bytes' and 'bytes' >>> b'foo %s' % 'bar' Traceback (most recent call last): File "<stdin>", line 1, in <module> TypeError: unsupported operand type(s) for %: 'bytes' and 'str' In other words, if we can't format bytes with bytes, and recall that all mercurial strings will be converted by a fixer, then things will break badly if we don't take a similar approach. The other addition with this patch is that the os.environ dictionary is monkeypatched to have bytes items. Hopefully this won't be needed in the future, as python 3.2 might get a os.environb dictionary that holds bytes items.

Matt Mackall - - Load All Authors

File last commit:

r10282:08a0f04b default


                r11748:37a70a78

default

Download file

             memory.py
        
                    36 lines
            
             | 1.1 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / contrib / memory.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # 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)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# 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)