Staging
v0.5.1
https://github.com/python/cpython
Revision b365a06a8444c90e4c79d4bff1c58cd9dd793569 authored by Benjamin Peterson on 24 March 2015, 16:12:44 UTC, committed by Benjamin Peterson on 24 March 2015, 16:12:44 UTC
https://twitter.com/ncoghlan_dev/status/579173053793353728
1 parent 32af754
Raw File
Tip revision: b365a06a8444c90e4c79d4bff1c58cd9dd793569 authored by Benjamin Peterson on 24 March 2015, 16:12:44 UTC
change Σ to ν for obscure joke reasons
Tip revision: b365a06
memory_watchdog.py
"""Memory watchdog: periodically read the memory usage of the main test process
and print it out, until terminated."""
# stdin should refer to the process' /proc/<PID>/statm: we don't pass the
# process' PID to avoid a race condition in case of - unlikely - PID recycling.
# If the process crashes, reading from the /proc entry will fail with ESRCH.


import os
import sys
import time


try:
    page_size = os.sysconf('SC_PAGESIZE')
except (ValueError, AttributeError):
    try:
        page_size = os.sysconf('SC_PAGE_SIZE')
    except (ValueError, AttributeError):
        page_size = 4096

while True:
    sys.stdin.seek(0)
    statm = sys.stdin.read()
    data = int(statm.split()[5])
    sys.stdout.write(" ... process data size: {data:.1f}G\n"
                     .format(data=data * page_size / (1024 ** 3)))
    sys.stdout.flush()
    time.sleep(1)
back to top