Changeset 336

Timestamp:

06/21/08 00:17:17 (4 years ago)

Author:

simon

Message:

Add memory recipe. Make 2.5/2.6/3.0 compatible.

Files:

• hodgestar/Talks/PythonObjects/util.py (modified) (1 diff)

hodgestar/Talks/PythonObjects/util.py

@@ -25 +25 @@
     answer, err = egrep.communicate()
 
-    print "Which module does the keyword %s appear in?" % question
+    print ("Which module does the keyword " + question + " appear in?")
     for i in reversed(range(5)):
-        sys.stdout.write("%s ... " % i)
+        sys.stdout.write(str(i) + " ... ")
         sys.stdout.flush()
         time.sleep(1)
-    print ""
-    print "-- Answers --"
-    print answer
+    print ("")
+    print ("-- Answers --")
+    print (answer)
 
+
+# From:
+# http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/286222
+#
+
+_proc_status = '/proc/' + str(os.getpid()) + '/status'
+
+_scale = {'kB': 1024.0, 'mB': 1024.0*1024.0,
+          'KB': 1024.0, 'MB': 1024.0*1024.0}
+
+def _VmB(VmKey):
+    '''Private.
+    '''
+    global _proc_status, _scale
+    # get pseudo file  /proc/<pid>/status
+    try:
+        t = open(_proc_status)
+        v = t.read()
+        t.close()
+    except:
+        return 0.0  # non-Linux?
+    # get VmKey line e.g. 'VmRSS:  9999  kB\n ...'
+    i = v.index(VmKey)
+    v = v[i:].split(None, 3)  # whitespace
+    if len(v) < 3:
+        return 0.0  # invalid format?
+    # convert Vm value to bytes
+    return float(v[1]) * _scale[v[2]]
+
+def memory(since=0.0):
+    '''Return memory usage in bytes.
+    '''
+    return _VmB('VmSize:') - since
+
+
+def resident(since=0.0):
+    '''Return resident memory usage in bytes.
+    '''
+    return _VmB('VmRSS:') - since
+
+
+def stacksize(since=0.0):
+    '''Return stack size in bytes.
+    '''
+    return _VmB('VmStk:') - since
+
+# END Recipe