cpuhog

#!/usr/bin/env python3
"""
usage: cpuhog [NUM_PROCS]

CPUHog will hog your CPU.
"""

import multiprocessing
import os
import sys


def hog():
    print('hog started: %d' % os.getpid())
    try:
        while True:
            pass
    except KeyboardInterrupt:
        pass


def run_hogs(num_hogs):
    print('Launching %d hogs' % num_hogs)
    procs = [multiprocessing.Process(target=hog) for i in range(num_hogs)]
    for p in procs:
        p.start()
    for p in procs:
        p.join()


if __name__ == '__main__':
    if '-h' in sys.argv or '--help' in sys.argv:
        print(__doc__.strip())
        sys.exit()

    if len(sys.argv) > 1:
        num = int(sys.argv[1])
    else:
        num = multiprocessing.cpu_count()

    try:
        run_hogs(num)
    except KeyboardInterrupt:
        print('')