wait_or_kill.py

wait_or_kill.py

import sh
import signal


class TooLongToWait(Exception):
    pass


def alarm_handler(signum, frame):
    raise TooLongToWait


try:
    process = sh.ls("/", _bg=True)
    signal.signal(signal.SIGALRM, alarm_handler)
    signal.alarm(2)
    process.wait()
    signal.alarm(0)
except TooLongToWait:
    process.kill()
    process.wait()
    print "I don't have all day..."

print "I'm alive!"