# Timeouts

import signal

class TimeoutException(Exception): pass

def run(func, timeout):

    '''@param timeout sec'''

    def handle_timeout(*args): raise TimeoutException

    prev_handler = signal.signal(signal.SIGALRM, handle_timeout)

    signal.alarm(timeout)

    try: return func()

    finally:

        signal.alarm(0)

        signal.signal(signal.SIGALRM, prev_handler)