timemem

timemem

#!/usr/bin/env python3

import subprocess
import resource
import sys
import time


USAGE = '''usage: timemem <command-line>...

Run the command line and show total time usage (sum of the process, its 
children, its grand children, etc) and the max resident memory usage (the 
largest of these processes, not total).
'''


if len(sys.argv) == 1 or sys.argv[1] in ('-h', '--help'):
    print(USAGE)
    sys.exit(0)

start = time.time()
retcode = subprocess.run(sys.argv[1:]).returncode
end = time.time()

ru = resource.getrusage(resource.RUSAGE_CHILDREN)

print('real %g' % (end - start))
print('user %g' % ru.ru_utime)
print('sys %g' % ru.ru_stime)
print('maxrss %g' % ru.ru_maxrss)

sys.exit(retcode)