Merge pull request #68 from yhuang43/master

add optional 'pid' parameter to system.vmpeak()

Author: jnolan14

surfa/system.py

@@ -93,15 +93,18 @@ def hostname(short=True):
     return node
 
 
-def vmpeak():
+def vmpeak(pid=None):
     """
     Return the peak memory usage of the process in kilobytes.
 
     Note: This only works on linux machines because it requires `/proc/self/status`.
     """
     # TODO: switch to this (portable across platforms)
     # return resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
-    procstatus = '/proc/self/status'
+    if (pid is None):
+        procstatus = '/proc/self/status'
+    else:
+        procstatus = os.path.join('/proc', str(pid), 'status')
     if os.path.exists(procstatus):
         with open(procstatus, 'r') as file:
             for line in file: