Python 2.3 Compatible Memory Profiler

MemoryProfiler

class MemoryProfiler:

    def mem(self, size="rss"):
        """Generalization; memory sizes: rss, rsz, vsz."""
        return os.popen('ps -p %d -o %s | tail -1' % (os.getpid(), size)).read()

    def rss(self):
        """Return ps -o rss (resident) memory in kB."""
        return self.convert_kilobytes(self.mem("rss"))

    def rsz(self):
        """Return ps -o rsz (resident + text) memory in kB."""
        return self.convert_kilobytes(self.mem("rsz"))

    def vsz(self):
        """Return ps -o vsz (virtual) memory in kB."""
        return self.convert_kilobytes(self.mem("vsz"))
        
    def convert_kilobytes(self, kilobytes):
        kilobytes = float(kilobytes)
        return self.convert_bytes(kilobytes * 1024)
   
    def convert_bytes(self, bytes):
        bytes = float(bytes)
        if bytes >= 1099511627776:
            terabytes = bytes / 1099511627776
            size = '%.2fT' % terabytes
        elif bytes >= 1073741824:
            gigabytes = bytes / 1073741824
            size = '%.2fG' % gigabytes
        elif bytes >= 1048576:
            megabytes = bytes / 1048576
            size = '%.2fM' % megabytes
        elif bytes >= 1024:
            kilobytes = bytes / 1024
            size = '%.2fK' % kilobytes
        else:
            size = '%.2fb' % bytes
        return size
        
    def print_usage(label=""):
        memory_profiler = MemoryProfiler()
        
        output = ""
        
        if label:
            output += label + ": "
            
        output += "RSS: %s; RSZ: %s; VSZ: %s" %\
                  (memory_profiler.rss(), memory_profiler.rsz(), memory_profiler.vsz())
                  
        print output
    
    print_usage = staticmethod(print_usage)

To use, call this anywhere:

MemoryProfiler.print_usage("label")

Note, this class derived from answers found on StackOverflow